Ejemplo n.º 1
0
 def setUp(self):
     self.client = GremlinClient('http://172.17.0.248:8182')
Ejemplo n.º 2
0
class TestClient(unittest.TestCase):
    def setUp(self):
        self.client = GremlinClient('http://172.17.0.248:8182')

    def tearDown(self):
        for v in self.client.V().run():
            self.client.delete_vertex(vertex_id = v.vertex_id)

    def test_add_vertex(self):
        init_cnt = len(self.client.V().run())
        v = self.client.addVertex(label = "a", a = 123,  b = 'qweq').first()
        print repr(v)
        vs = self.client.V().run()
        print repr(vs)
        self.assertEqual(len(vs) - init_cnt, 1)
        self.client.delete_vertex(vertex_id = v.vertex_id)
        self.assertEqual(len(self.client.V().run()), init_cnt)

    def test_query_vertices(self):
        print 'before', repr(self.client.V().run())
        v1 = self.client.addVertex(a = 123,  b = 'qwe').first()
        v2 = self.client.addVertex(a = 1234,  b = 'qwe').first()
        print 'all', repr(self.client.V().run())
        print '12', repr(self.client.V().has('a', 12).run())
        print '123', repr(self.client.V().has('a', 123).run())
        print '1234', repr(self.client.V().has('a', 1234).run())
        print 'qwe', repr(self.client.V().has('b', 'qwe').run())
        self.assertEqual(len(self.client.V().has('a', 12).run()), 0)
        self.assertEqual(len(self.client.V().has('a', 123).run()), 1)
        self.assertEqual(len(self.client.V().has('a', 1234).run()), 1)
        self.assertEqual(len(self.client.V().has('b', 'qwe').run()), 2)
        self.client.delete_vertex(vertex_id = v1.vertex_id)
        self.client.delete_vertex(vertex_id = v2.vertex_id)