add xlsDataSetLoader

.gitignore

@@ -7,3 +7,9 @@ bin
 .DS_Store
 .idea
 *.iml
+*.ipr
+*.iws
+*/*.iml
+*/*.iws
+*/*.ipr
+

spring-test-dbunit-sample/src/test/java/com/github/springtestdbunit/sample/service/PersonServiceTest.java

@@ -18,7 +18,7 @@
 import com.github.springtestdbunit.sample.entity.Person;
 
 @RunWith(SpringJUnit4ClassRunner.class)
-@ContextConfiguration
+@ContextConfiguration({"classpath:com/github/springtestdbunit/sample/service/applicationContext.xml"})
 @TestExecutionListeners({ DependencyInjectionTestExecutionListener.class, DbUnitTestExecutionListener.class })
 public class PersonServiceTest {
 

spring-test-dbunit-sample/src/test/java/com/github/springtestdbunit/sample/service/XlsDataLoaderTest.java

@@ -0,0 +1,37 @@
+package com.github.springtestdbunit.sample.service;
+
+import com.github.springtestdbunit.DbUnitTestExecutionListener;
+import com.github.springtestdbunit.annotation.DatabaseSetup;
+import com.github.springtestdbunit.annotation.DbUnitConfiguration;
+import com.github.springtestdbunit.dataset.XlsDataSetLoader;
+import com.github.springtestdbunit.sample.entity.Person;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.TestExecutionListeners;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.support.DependencyInjectionTestExecutionListener;
+
+import java.util.List;
+
+import static org.junit.Assert.assertEquals;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration({"classpath:com/github/springtestdbunit/sample/service/applicationContext.xml"})
+@DbUnitConfiguration(dataSetLoader= XlsDataSetLoader.class)
+@TestExecutionListeners({ DependencyInjectionTestExecutionListener.class, DbUnitTestExecutionListener.class })
+public class XlsDataLoaderTest {
+
+	@Autowired
+	private PersonService personService;
+
+	@Test
+	@DatabaseSetup("sampleData.xls")
+	public void testFind() throws Exception {
+		List<Person> personList = this.personService.find("wang");
+		assertEquals(1, personList.size());
+		assertEquals("wu", personList.get(0).getLastName());
+	}
+
+}

spring-test-dbunit/src/main/java/com/github/springtestdbunit/dataset/XlsDataSetLoader.java

@@ -0,0 +1,23 @@
+package com.github.springtestdbunit.dataset;
+
+import org.dbunit.dataset.IDataSet;
+import org.dbunit.dataset.excel.XlsDataSet;
+import org.springframework.core.io.Resource;
+
+import java.io.InputStream;
+
+/**
+ * Created by yangjianzhou on 17-11-18.
+ */
+public class XlsDataSetLoader extends AbstractDataSetLoader {
+
+    @Override
+    protected IDataSet createDataSet(Resource resource) throws Exception {
+        InputStream inputStream = resource.getInputStream();
+        try {
+            return new XlsDataSet(inputStream);
+        } finally {
+            inputStream.close();
+        }
+    }
+}