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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gremlin-test Show documentation

The newest version!

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.Map;

import static com.tinkerpop.gremlin.LoadGraphWith.GraphData.MODERN;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;

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

    @Test
    @LoadGraphWith(MODERN)
    public void g_V_hasXageX_groupCountXaX_byXnameX_out_capXaX() {
        final Traversal> traversal = get_g_V_hasXageX_groupCountXaX_byXnameX_out_capXaX();
        printTraversalForm(traversal);
        Map map = traversal.next();
        assertFalse(traversal.hasNext());
        assertEquals(map.get("marko"), new Long(1l));
        assertEquals(map.get("vadas"), new Long(1l));
        assertEquals(map.get("peter"), new Long(1l));
        assertEquals(map.get("josh"), new Long(1l));
        assertEquals(map.size(), 4);
    }

    public static class StandardTest extends SideEffectCapTest {
        public StandardTest() {
            this.requiresGraphComputer = false;
        }

        @Override
        public Traversal> get_g_V_hasXageX_groupCountXaX_byXnameX_out_capXaX() {
            return g.V().has("age").groupCount("a").by("name").out().cap("a");
        }
    }

    public static class ComputerTest extends SideEffectCapTest {

        public ComputerTest() {
            this.requiresGraphComputer = true;
        }


        @Override
        public Traversal> get_g_V_hasXageX_groupCountXaX_byXnameX_out_capXaX() {
            return g.V().has("age").groupCount("a").by("name").out().>cap("a").submit(g.compute());
        }
    }
}