com.tinkerpop.gremlin.process.graph.step.util.MemoryTest Maven / Gradle / Ivy

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

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

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

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

    @Test
    @LoadGraphWith(CLASSIC)
    public void g_V_memory() {
        final Traversal.Memory memory = get_g_V_memory();
        assertFalse(memory.get("a").isPresent());
        assertTrue(memory.get(Graph.Key.hide("g")).isPresent());
        assertFalse(memory.get("g").isPresent());
        assertTrue(Graph.class.isAssignableFrom(memory.get(Graph.Key.hide("g")).get().getClass()));
    }

    public static class JavaMemoryTest extends MemoryTest {
        public JavaMemoryTest() {
            requiresGraphComputer = false;
        }

        public Traversal.Memory get_g_V_memory() {
            return g.V().memory();
        }
    }
}