wayang-commons/wayang-utils/wayang-profile-db/src/test/java/profiledb/ProfileDBTest.java

package profiledb;

import org.junit.Assert;
import org.junit.Test;
import profiledb.measurement.TestMemoryMeasurement;
import profiledb.measurement.TestTimeMeasurement;
import profiledb.model.Experiment;
import profiledb.model.Measurement;
import profiledb.model.Subject;
import profiledb.storage.FileStorage;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.util.*;

public class ProfileDBTest {

    @Test
    public void testPolymorphSaveAndLoad() throws IOException {

        try {
            URI uri = new URI("file:///Users/rodrigopardomeza/Desktop/random/myfile.txt");
            FileStorage store = new FileStorage(uri);

            ProfileDB profileDB = new ProfileDB(store)
                    .registerMeasurementClass(TestMemoryMeasurement.class)
                    .registerMeasurementClass(TestTimeMeasurement.class);

            /**
             * Esto es lo que se espera del codigo del cliente
             * Tiene que usar la API para registrar medidas
             */
            // crea un experimento falso
            final Experiment experiment = new Experiment("test-xp", new Subject("PageRank", "1.0"), "test experiment");

            // Agrega medidas falsas hardcoded
            Measurement timeMeasurement = new TestTimeMeasurement("exec-time", 12345L);
            Measurement memoryMeasurement = new TestMemoryMeasurement("exec-time", System.currentTimeMillis(), 54321L);

            /*Agrega las medidas al experimento*/
            experiment.addMeasurement(timeMeasurement);
            experiment.addMeasurement(memoryMeasurement);

            // Save the experiment.
            /**
             * Guarda el experimento en memoria
             */
            byte[] buffer;
            ByteArrayOutputStream bos = new ByteArrayOutputStream();
            profileDB.getStorage().save(Collections.singleton(experiment), bos);
            bos.close();
            buffer = bos.toByteArray();
            System.out.println("Buffer contents: " + new String(buffer, "UTF-8"));

            // Load the experiment.
            /**
             * Lee el experimento desde el buffer en memoria
             */
            ByteArrayInputStream bis = new ByteArrayInputStream(buffer);
            Collection<Experiment> loadedExperiments = profileDB.getStorage().load(bis);

            // Compare the experiments.
            Assert.assertEquals(1, loadedExperiments.size());
            Experiment loadedExperiment = loadedExperiments.iterator().next();
            Assert.assertEquals(experiment, loadedExperiment);

            // Compare the measurements.
            Assert.assertEquals(2, loadedExperiment.getMeasurements().size());
            Set<Measurement> expectedMeasurements = new HashSet<>(2);
            expectedMeasurements.add(timeMeasurement);
            expectedMeasurements.add(memoryMeasurement);
            Set<Measurement> loadedMeasurements = new HashSet<>(loadedExperiment.getMeasurements());
            Assert.assertEquals(expectedMeasurements, loadedMeasurements);

        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
    }

    @Test
    public void testRecursiveSaveAndLoad() throws IOException {
        try {
            URI uri = new URI("file:///Users/rodrigopardomeza/Desktop/random/myfile.txt");
            FileStorage store = new FileStorage(uri);

            ProfileDB profileDB = new ProfileDB(store)
                    .registerMeasurementClass(TestMemoryMeasurement.class)
                    .registerMeasurementClass(TestTimeMeasurement.class);

            // Create an example experiment.
            final Experiment experiment = new Experiment("test-xp", new Subject("PageRank", "1.0"), "test experiment");
            TestTimeMeasurement topLevelMeasurement = new TestTimeMeasurement("exec-time", 12345L);
            TestTimeMeasurement childMeasurement = new TestTimeMeasurement("sub-exec-time", 2345L);
            topLevelMeasurement.addSubmeasurements(childMeasurement);
            experiment.addMeasurement(topLevelMeasurement);

            // Save the experiment.
            byte[] buffer;
            ByteArrayOutputStream bos = new ByteArrayOutputStream();
            profileDB.getStorage().save(Collections.singleton(experiment), bos);
            bos.close();
            buffer = bos.toByteArray();
            System.out.println("Buffer contents: " + new String(buffer, "UTF-8"));

            // Load the experiment.
            ByteArrayInputStream bis = new ByteArrayInputStream(buffer);
            Collection<Experiment> loadedExperiments = profileDB.getStorage().load(bis);

            // Compare the experiments.
            Assert.assertEquals(1, loadedExperiments.size());
            Experiment loadedExperiment = loadedExperiments.iterator().next();
            Assert.assertEquals(experiment, loadedExperiment);

            // Compare the measurements.
            Assert.assertEquals(1, loadedExperiment.getMeasurements().size());
            final Measurement loadedMeasurement = loadedExperiment.getMeasurements().iterator().next();
            Assert.assertEquals(topLevelMeasurement, loadedMeasurement);
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
    }

    @Test
    public void testFileOperations() throws IOException {

        try {
            URI uri = new URI("file:///Users/rodrigopardomeza/Desktop/random/myfile.txt");
            FileStorage store = new FileStorage(uri);

            ProfileDB profileDB = new ProfileDB(store)
                    .registerMeasurementClass(TestMemoryMeasurement.class)
                    .registerMeasurementClass(TestTimeMeasurement.class);

            // Create example experiments.
            final Experiment experiment1 = new Experiment("xp1", new Subject("PageRank", "1.0"), "test experiment 1");
            experiment1.addMeasurement(new TestTimeMeasurement("exec-time", 1L));
            final Experiment experiment2 = new Experiment("xp2", new Subject("KMeans", "1.1"), "test experiment 2");
            experiment2.addMeasurement(new TestTimeMeasurement("exec-time", 2L));
            final Experiment experiment3 = new Experiment("xp3", new Subject("Apriori", "2.0"), "test experiment 3");
            experiment3.addMeasurement(new TestMemoryMeasurement("ram", System.currentTimeMillis(), 3L));

            // Save the experiments.
            File tempDir = Files.createTempDirectory("profiledb").toFile();
            File file = new File(tempDir, "profiledb.json");
            profileDB.getStorage().save(experiment1);
            profileDB.getStorage().append(experiment2, experiment3);

            Files.lines(file.toPath()).forEach(System.out::println);

            // Load and compare.
            final Set<Experiment> loadedExperiments = new HashSet<>(profileDB.getStorage().load());
            final List<Experiment> expectedExperiments = Arrays.asList(experiment1, experiment2, experiment3);
            Assert.assertEquals(expectedExperiments.size(), loadedExperiments.size());
            Assert.assertEquals(new HashSet<>(expectedExperiments), new HashSet<>(loadedExperiments));
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
    }

    @Test
    public void testAppendOnNonExistentFile() throws IOException {

        try {
            URI uri = new URI("file:///Users/rodrigopardomeza/Desktop/random/myfile.txt");
            FileStorage store = new FileStorage(uri);

            // This seems to be an issue on Linux.
            ProfileDB profileDB = new ProfileDB(store)
                    .registerMeasurementClass(TestMemoryMeasurement.class)
                    .registerMeasurementClass(TestTimeMeasurement.class);

            // Create example experiments.
            final Experiment experiment1 = new Experiment("xp1", new Subject("PageRank", "1.0"), "test experiment 1");
            experiment1.addMeasurement(new TestTimeMeasurement("exec-time", 1L));

            // Save the experiments.
            File tempDir = Files.createTempDirectory("profiledb").toFile();
            File file = new File(tempDir, "new-profiledb.json");
            Assert.assertTrue(!file.exists() || file.delete());
            profileDB.getStorage().append(experiment1);

            Files.lines(file.toPath()).forEach(System.out::println);

            // Load and compare.
            final Set<Experiment> loadedExperiments = new HashSet<>(profileDB.getStorage().load());
            final List<Experiment> expectedExperiments = Collections.singletonList(experiment1);
            Assert.assertEquals(expectedExperiments.size(), loadedExperiments.size());
            Assert.assertEquals(new HashSet<>(expectedExperiments), new HashSet<>(loadedExperiments));
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
    }

}