Search in sources :

Example 6 with Knows

use of com.tinkerpop.frames.domain.incidences.Knows in project frames by tinkerpop.

the class FramedGraphQueryImplTest method testDelegation.

@Test
public void testDelegation() {
    GraphQuery mockGraphQuery = mock(GraphQuery.class);
    FramedGraph framedGraph = new FramedGraphFactory().create(null);
    FramedGraphQueryImpl query = new FramedGraphQueryImpl(framedGraph, mockGraphQuery);
    stub(mockGraphQuery.has("")).toReturn(mockGraphQuery);
    query.has("");
    verify(mockGraphQuery).has("");
    stub(mockGraphQuery.has("", "bar")).toReturn(mockGraphQuery);
    query.has("", "bar");
    verify(mockGraphQuery).has("", "bar");
    Predicate predicate = new Predicate() {

        @Override
        public boolean evaluate(Object first, Object second) {
            return false;
        }
    };
    stub(mockGraphQuery.has("", predicate, "bar")).toReturn(mockGraphQuery);
    query.has("", predicate, "bar");
    verify(mockGraphQuery).has(eq(""), same(predicate), eq("bar"));
    stub(mockGraphQuery.has("", 2, Compare.EQUAL)).toReturn(mockGraphQuery);
    query.has("", 2, Compare.EQUAL);
    verify(mockGraphQuery).has(eq(""), same(2), eq(Compare.EQUAL));
    stub(mockGraphQuery.hasNot("")).toReturn(mockGraphQuery);
    query.hasNot("");
    verify(mockGraphQuery).hasNot(eq(""));
    stub(mockGraphQuery.hasNot("", "bar")).toReturn(mockGraphQuery);
    query.hasNot("", "bar");
    verify(mockGraphQuery).hasNot(eq(""), eq("bar"));
    stub(mockGraphQuery.interval("", "bar", "bif")).toReturn(mockGraphQuery);
    query.interval("", "bar", "bif");
    verify(mockGraphQuery).interval(eq(""), eq("bar"), eq("bif"));
    stub(mockGraphQuery.limit(1)).toReturn(mockGraphQuery);
    query.limit(1);
    verify(mockGraphQuery).limit(1);
    List<Vertex> v = new ArrayList<Vertex>();
    stub(mockGraphQuery.vertices()).toReturn(v);
    query.vertices();
    verify(mockGraphQuery).vertices();
    Iterable<Person> people = query.vertices(Person.class);
    verify(mockGraphQuery, times(2)).vertices();
    assertFalse(people.iterator().hasNext());
    List<Edge> e = new ArrayList<Edge>();
    stub(mockGraphQuery.edges()).toReturn(e);
    query.edges();
    verify(mockGraphQuery).edges();
    Iterable<Knows> knows = query.edges(Knows.class);
    verify(mockGraphQuery, times(2)).edges();
    assertFalse(knows.iterator().hasNext());
}
Also used : Vertex(com.tinkerpop.blueprints.Vertex) ArrayList(java.util.ArrayList) Predicate(com.tinkerpop.blueprints.Predicate) FramedGraph(com.tinkerpop.frames.FramedGraph) FramedGraphFactory(com.tinkerpop.frames.FramedGraphFactory) Knows(com.tinkerpop.frames.domain.incidences.Knows) GraphQuery(com.tinkerpop.blueprints.GraphQuery) Person(com.tinkerpop.frames.domain.classes.Person) Edge(com.tinkerpop.blueprints.Edge) Test(org.junit.Test)

Example 7 with Knows

use of com.tinkerpop.frames.domain.incidences.Knows in project frames by tinkerpop.

the class TypeResolverTest method testAdditionalTypes.

@Test
public void testAdditionalTypes() {
    Graph graph = TinkerGraphFactory.createTinkerGraph();
    FramedGraphFactory factory = new FramedGraphFactory(new AbstractModule() {

        @Override
        public void doConfigure(FramedGraphConfiguration config) {
            config.addTypeResolver(new TypeResolver() {

                @Override
                public Class<?>[] resolveTypes(Edge e, Class<?> defaultType) {
                    return new Class[] { AdditionalEdge.class };
                }

                @Override
                public Class<?>[] resolveTypes(Vertex v, Class<?> defaultType) {
                    return new Class[] { AdditionalVertex.class };
                }
            });
        }
    });
    framedGraph = factory.create(graph);
    Person marko = framedGraph.getVertex(1, Person.class);
    Assert.assertTrue(marko instanceof AdditionalVertex);
    Assert.assertFalse(marko instanceof AdditionalEdge);
    Knows knows = marko.getKnows().iterator().next();
    Assert.assertTrue(knows instanceof AdditionalEdge);
    Assert.assertFalse(knows instanceof AdditionalVertex);
}
Also used : Vertex(com.tinkerpop.blueprints.Vertex) TypeResolver(com.tinkerpop.frames.modules.TypeResolver) AbstractModule(com.tinkerpop.frames.modules.AbstractModule) TransactionalGraph(com.tinkerpop.blueprints.TransactionalGraph) FramedGraph(com.tinkerpop.frames.FramedGraph) Graph(com.tinkerpop.blueprints.Graph) FramedGraphFactory(com.tinkerpop.frames.FramedGraphFactory) Knows(com.tinkerpop.frames.domain.incidences.Knows) Edge(com.tinkerpop.blueprints.Edge) Person(com.tinkerpop.frames.domain.classes.Person) FramedGraphConfiguration(com.tinkerpop.frames.FramedGraphConfiguration) Test(org.junit.Test)

Example 8 with Knows

use of com.tinkerpop.frames.domain.incidences.Knows in project frames by tinkerpop.

the class JavaHandlerTest method testJavaHandlerEdges.

@Test
public void testJavaHandlerEdges() {
    Knows knows = g.getEdge(7, Direction.OUT, Knows.class);
    Assert.assertEquals("marko<->vadas", knows.getNames());
}
Also used : Knows(com.tinkerpop.frames.domain.incidences.Knows) Test(org.junit.Test)

Aggregations

Knows (com.tinkerpop.frames.domain.incidences.Knows)8 Test (org.junit.Test)6 Person (com.tinkerpop.frames.domain.classes.Person)5 Edge (com.tinkerpop.blueprints.Edge)3 Graph (com.tinkerpop.blueprints.Graph)3 Vertex (com.tinkerpop.blueprints.Vertex)3 FramedGraph (com.tinkerpop.frames.FramedGraph)2 FramedGraphFactory (com.tinkerpop.frames.FramedGraphFactory)2 GraphQuery (com.tinkerpop.blueprints.GraphQuery)1 Predicate (com.tinkerpop.blueprints.Predicate)1 TransactionalGraph (com.tinkerpop.blueprints.TransactionalGraph)1 TinkerGraph (com.tinkerpop.blueprints.impls.tg.TinkerGraph)1 FramedGraphConfiguration (com.tinkerpop.frames.FramedGraphConfiguration)1 Project (com.tinkerpop.frames.domain.classes.Project)1 CreatedInfo (com.tinkerpop.frames.domain.incidences.CreatedInfo)1 AbstractModule (com.tinkerpop.frames.modules.AbstractModule)1 TypeResolver (com.tinkerpop.frames.modules.TypeResolver)1 ArrayList (java.util.ArrayList)1 Before (org.junit.Before)1