com.tinkerpop.gremlin.process.graph.step.sideEffect.StoreTest Maven / Gradle / Ivy

Go to download
package com.tinkerpop.gremlin.process.graph.step.sideEffect;

import com.tinkerpop.gremlin.LoadGraphWith;
import com.tinkerpop.gremlin.process.AbstractGremlinProcessTest;
import com.tinkerpop.gremlin.process.Traversal;
import com.tinkerpop.gremlin.structure.Vertex;
import org.junit.Test;

import java.util.Collection;

import static com.tinkerpop.gremlin.LoadGraphWith.GraphData.CLASSIC;
import static org.junit.Assert.*;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public abstract class StoreTest extends AbstractGremlinProcessTest {
    public abstract Traversal get_g_V_storeXnameX_asXaX_out_capXaX();

    @Test
    @LoadGraphWith(CLASSIC)
    public void g_V_storeXa_nameX_out_capXaX() {
        final Traversal traversal = get_g_V_storeXnameX_asXaX_out_capXaX();
        printTraversalForm(traversal);
        Collection names = traversal.next();
        assertEquals(6, names.size());
        assertTrue(names.contains("marko"));
        assertTrue(names.contains("josh"));
        assertTrue(names.contains("peter"));
        assertTrue(names.contains("lop"));
        assertTrue(names.contains("ripple"));
        assertTrue(names.contains("vadas"));
        assertFalse(traversal.hasNext());
    }

    public static class JavaStoreTest extends StoreTest {
        public JavaStoreTest() {
            requiresGraphComputer = false;
        }

        public Traversal get_g_V_storeXnameX_asXaX_out_capXaX() {
            return g.V().store(v -> v.value("name")).as("a").out().cap("a");
        }
    }

    public static class JavaComputerStoreTest extends StoreTest {
        public JavaComputerStoreTest() {
            requiresGraphComputer = true;
        }

        public Traversal get_g_V_storeXnameX_asXaX_out_capXaX() {
            return g.V().store(v -> v.value("name")).as("a").out().cap("a").submit(g.compute());
        }
    }

}