Add Xls format support, close #20

takimafr · Dec 4, 2012 · e1265df · e1265df
1 parent 254808a
commit e1265df
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/spring-dbunit-core/src/main/java/com/excilys/ebi/spring/dbunit/config/DataSetFormat.java b/spring-dbunit-core/src/main/java/com/excilys/ebi/spring/dbunit/config/DataSetFormat.java
@@ -23,6 +23,7 @@
 import org.dbunit.dataset.DataSetException;
 import org.dbunit.dataset.IDataSet;
 import org.dbunit.dataset.csv.CsvDataSet;
+import org.dbunit.dataset.excel.XlsDataSet;
 import org.dbunit.dataset.stream.StreamingDataSet;
 import org.dbunit.dataset.xml.FlatDtdDataSet;
 import org.dbunit.dataset.xml.FlatXmlDataSet;
@@ -84,6 +85,7 @@ protected IDataSet fromResource(Resource resource, DataSetFormatOptions options)
 			return builder.build(resource.getInputStream());
 		}
 	},
+
 	/**
 	 * @see {@link XmlDataSet}.
 	 */
@@ -96,6 +98,7 @@ protected IDataSet fromResource(Resource resource, DataSetFormatOptions options)
 			return new XmlDataSet(resource.getInputStream());
 		}
 	},
+
 	/**
 	 * @see {@link StreamingXmlDataSet}.
 	 */
@@ -108,6 +111,7 @@ protected IDataSet fromResource(Resource resource, DataSetFormatOptions options)
 			return new StreamingDataSet(new XmlProducer(new InputSource(resource.getInputStream())));
 		}
 	},
+
 	/**
 	 * @see {@link FlatDtdDataSet}.
 	 */
@@ -118,6 +122,7 @@ protected IDataSet fromResource(Resource resource, DataSetFormatOptions options)
 			return new FlatDtdDataSet(new LinkedHashMapFlatDtdProducer(new InputSource(resource.getInputStream())));
 		}
 	},
+
 	/**
 	 * @see {@link CsvDataSet}.
 	 */
@@ -126,6 +131,16 @@ protected IDataSet fromResource(Resource resource, DataSetFormatOptions options)
 		protected IDataSet fromResource(Resource resource, DataSetFormatOptions options) throws DataSetException, IOException {
 			return new CsvDataSet(resource.getFile());
 		}
+	},
+
+	/**
+	 * @see {@link CsvDataSet}.
+	 */
+	XLS {
+		@Override
+		protected IDataSet fromResource(Resource resource, DataSetFormatOptions options) throws DataSetException, IOException {
+			return new XlsDataSet(resource.getFile());
+		}
 	};
 
 	private static final ResourcePatternResolver RESOURCE_LOADER = new PathMatchingResourcePatternResolver();