runtime/org.argeo.slc.support.simple/src/main/java/org/argeo/slc/support/deploy/db/DbUnitDeploymentData.java

   1 package org.argeo.slc.support.deploy.db;
   2
   3 import java.io.InputStream;
   4
   5 import org.argeo.slc.SlcException;
   6 import org.argeo.slc.deploy.DeploymentData;
   7 import org.dbunit.dataset.IDataSet;
   8 import org.dbunit.dataset.xml.FlatXmlDataSet;
   9 import org.springframework.core.io.Resource;
  10
  11 public class DbUnitDeploymentData implements DeploymentData {
  12         private Resource dataSetLocation;
  13
  14         public IDataSet createDataSet() {
  15                 try {
  16                         InputStream in = dataSetLocation.getInputStream();
  17                         IDataSet dataSet = new FlatXmlDataSet(in);
  18                         in.close();
  19                         return dataSet;
  20                 } catch (Exception e) {
  21                         throw new SlcException("Cannot create data set", e);
  22                 }
  23
  24         }
  25
  26         public void setDataSetLocation(Resource dataSetLocation) {
  27                 this.dataSetLocation = dataSetLocation;
  28         }
  29
  30 }