Author: tchemit Date: 2012-10-24 14:18:10 +0200 (Wed, 24 Oct 2012) New Revision: 741 Url: http://forge.codelutin.com/repositories/revision/echobase/741 Log: add nice method to import test db Modified: trunk/echobase-services/src/test/java/fr/ifremer/echobase/services/EchoBaseTestServiceSupport.java Modified: trunk/echobase-services/src/test/java/fr/ifremer/echobase/services/EchoBaseTestServiceSupport.java =================================================================== --- trunk/echobase-services/src/test/java/fr/ifremer/echobase/services/EchoBaseTestServiceSupport.java 2012-10-21 18:36:58 UTC (rev 740) +++ trunk/echobase-services/src/test/java/fr/ifremer/echobase/services/EchoBaseTestServiceSupport.java 2012-10-24 12:18:10 UTC (rev 741) @@ -30,7 +30,14 @@ import fr.ifremer.echobase.entities.data.DataProcessing; import fr.ifremer.echobase.entities.data.Voyage; import fr.ifremer.echobase.io.InputFile; +import fr.ifremer.echobase.services.exportdb.ExportDbConfiguration; +import fr.ifremer.echobase.services.exportdb.ExportDbMode; +import fr.ifremer.echobase.services.exportdb.ExportDbService; import fr.ifremer.echobase.services.importdata.EchoBaseCsvFileImportResult; +import fr.ifremer.echobase.services.importdata.ImportException; +import fr.ifremer.echobase.services.importdb.ImportDbConfiguration; +import fr.ifremer.echobase.services.importdb.ImportDbMode; +import fr.ifremer.echobase.services.importdb.ImportDbService; import org.apache.commons.io.FileUtils; import org.apache.commons.io.IOUtils; import org.apache.commons.logging.Log; @@ -40,6 +47,7 @@ import org.junit.Rule; import org.nuiton.topia.TopiaException; import org.nuiton.topia.persistence.TopiaEntity; +import org.nuiton.util.FileUtil; import org.nuiton.util.StringUtil; import java.io.File; @@ -225,4 +233,40 @@ Preconditions.checkNotNull(dataProcessing); return dataProcessing.getTopiaId(); } + + protected void importdb(ImportDbMode mode, String path) throws IOException, ImportException { + + ImportDbService service = getService(ImportDbService.class); + + ImportDbConfiguration conf = new ImportDbConfiguration(getLocale()); + + File workingDirectory = new File(getTestDir(), "work-dir"); + FileUtil.createDirectoryIfNecessary(workingDirectory); + conf.setWorkingDirectory(workingDirectory); + conf.setComputeSteps(true); + conf.setCommitAfterEachFile(false); + conf.setImportDbMode(mode); + prepareInputFile(conf.getInput(), path); + + EchoBaseUser user = new EchoBaseUserImpl(); + user.setEmail("testUser@fake.fr"); + + service.doImport(conf, user); + } + + protected void exportDb(Voyage voyage, String exportFilename) throws IOException { + + ExportDbService service = getService(ExportDbService.class); + + ExportDbConfiguration conf = new ExportDbConfiguration(); + conf.setVoyageIds(new String[]{voyage.getTopiaId()}); + + File workingDirectory = new File(getTestDir(), "work-dir"); + FileUtil.createDirectoryIfNecessary(workingDirectory); + conf.setWorkingDirectory(workingDirectory); + conf.setFileName(exportFilename); + conf.setComputeSteps(true); + conf.setExportDbMode(ExportDbMode.ALL); + service.doExport(conf); + } }