package fr.gael.dhus.datastore.processing; import java.io.File; import java.io.IOException; import java.io.InputStream; import org.apache.commons.compress.archivers.ArchiveException; import org.apache.commons.compress.compressors.CompressorException; import org.apache.commons.io.FileUtils; import org.testng.Assert; import org.testng.annotations.AfterClass; import org.testng.annotations.BeforeClass; import org.testng.annotations.Test; import com.google.common.io.Files; import fr.gael.dhus.util.UnZip; import fr.gael.drb.impl.DrbFactoryResolver; import fr.gael.drbx.cortex.DrbCortexMetadataResolver; import fr.gael.drbx.cortex.DrbCortexModel; public class ProcessingManagerTest { File sample=null; // Same size for drb/system sizes. private static long SIZE=1650599; @BeforeClass void init() throws IOException { InputStream is=ProcessingManager.class.getResourceAsStream("size-test.zip"); File tmp_folder = Files.createTempDir(); File output = new File (tmp_folder, "size-test.zip"); FileUtils.copyInputStreamToFile(is, output); is.close(); sample = output; } @AfterClass void exit() { FileUtils.deleteQuietly(sample.getParentFile()); } @Test public void drb_size() throws IOException { DrbFactoryResolver.setMetadataResolver (new DrbCortexMetadataResolver ( DrbCortexModel.getDefaultModel ())); ProcessingManager mgr = new ProcessingManager(); long size = mgr.drb_size(sample); Assert.assertEquals(size, SIZE); } @Test public void size() { ProcessingManager mgr = new ProcessingManager(); long size = mgr.size(sample); Assert.assertEquals(size, SIZE); } @Test public void system_size() throws IOException, CompressorException, ArchiveException { ProcessingManager mgr = new ProcessingManager(); long size = mgr.system_size(sample); Assert.assertEquals(size, 494928); File folder=sample.getParentFile(); File extaction_folder=new File(folder, "unzip"); extaction_folder.mkdirs(); UnZip.unCompress(sample.getAbsolutePath(), extaction_folder.getAbsolutePath()); File tocheck = extaction_folder.listFiles()[0]; size = mgr.system_size(tocheck); Assert.assertEquals(size, SIZE, tocheck.getAbsolutePath()); } }