+package org.argeo.slc.support.deploy.db;\r
+\r
+import java.io.InputStream;\r
+\r
+import org.argeo.slc.core.SlcException;\r
+import org.argeo.slc.core.deploy.DeploymentData;\r
+import org.dbunit.dataset.IDataSet;\r
+import org.dbunit.dataset.xml.FlatXmlDataSet;\r
+import org.springframework.core.io.Resource;\r
+\r
+public class DbUnitDeploymentData implements DeploymentData{\r
+ private Resource dataSetLocation;\r
+\r
+ public IDataSet createDataSet(){\r
+ try {\r
+ InputStream in = dataSetLocation.getInputStream();\r
+ IDataSet dataSet = new FlatXmlDataSet(in);\r
+ in.close();\r
+ return dataSet;\r
+ } catch (Exception e) {\r
+ throw new SlcException("Cannot create data set", e);\r
+ }\r
+\r
+ }\r
+ \r
+ public void setDataSetLocation(Resource dataSetLocation) {\r
+ this.dataSetLocation = dataSetLocation;\r
+ }\r
+ \r
+ \r
+}\r