def setUp(self): self.client = RexsterClient('http://localhost:8182', 'graph')
class TestClient(unittest.TestCase): def setUp(self): self.client = RexsterClient('http://localhost:8182', 'graph') def tearDown(self): for v in self.client.vertices().get(): self.client.delete_vertex(v['_id']).get() def test_add_vertex(self): init_cnt = len(self.client.vertices().get()) v = self.client.create_vertex(a=123, b='qweq').get() print repr(v) vs = self.client.vertices().get() print repr(vs) self.assertEqual(len(vs) - init_cnt, 1) self.client.delete_vertex(v['_id']).get() self.assertEqual(len(self.client.vertices().get()), init_cnt) def test_query_vertices(self): print 'before', repr(self.client.vertices().get()) v1 = self.client.create_vertex(a=123, b='qwe').get() v2 = self.client.create_vertex(a=1234, b='qwe').get() print 'all', repr(self.client.vertices().get()) print '12', repr(self.client.vertices('a', 12).get()) print '123', repr(self.client.vertices('a', 123).get()) print '1234', repr(self.client.vertices('a', 1234).get()) print 'qwe', repr(self.client.vertices('b', 'qwe')) self.assertEqual(len(self.client.vertices('a', 12).get()), 0) self.assertEqual(len(self.client.vertices('a', 123).get()), 1) self.assertEqual(len(self.client.vertices('a', 1234).get()), 1) self.assertEqual(len(self.client.vertices('b', 'qwe').get()), 2) self.client.delete_vertex(v1['_id']).get() self.client.delete_vertex(v2['_id']).get() def test_scripts(self): script = ''' qq = 123; 1 + a + qq''' self.assertEqual( self.client.run_script_on_graph(script, a=22).get()[0], 146) self.assertEqual( self.client.run_script_on_graph('[1, 2, a]', a=22).get(), [1, 2, 22]) def test_lookup(self): v1 = self.client.create_vertex(a=123, b='qwe').get() v2 = self.client.create_vertex(a=1234, b='qwe').get() v3 = self.client.create_vertex(a=1234, b='qweq').get() self.assertEqual(len(self.client.lookup_vertex(a=123).get()), 1) self.assertEqual(len(self.client.lookup_vertex(Q(a=123)).get()), 1) self.assertEqual( len( self.client.lookup_vertex( Q(_properties=[['a', Cmp.GTE, 123]])).get()), 3) self.assertEqual( len( self.client.lookup_vertex( Q(_properties=[['a', Cmp.GT, 123]])).get()), 2) self.assertEqual( len( self.client.lookup_vertex( Q(_properties=[['a', Cmp.GT, 123]]) & Q(_properties=[['b', Cmp.EQ, 'qwe']])).get()), 1) self.assertEqual( len( self.client.lookup_vertex( Q(_properties=[['a', Cmp.GT, 123]]) & Q(_properties=[['b', Cmp.NE, 'qwe']])).get()), 1) self.client.delete_vertex(v1['_id']).get() self.client.delete_vertex(v2['_id']).get() self.client.delete_vertex(v3['_id']).get()
class TestClient(unittest.TestCase): def setUp(self): self.client = RexsterClient('http://localhost:8182', 'graph') def tearDown(self): for v in self.client.vertices().get(): self.client.delete_vertex(v['_id']).get() def test_add_vertex(self): init_cnt = len(self.client.vertices().get()) v = self.client.create_vertex(a = 123, b = 'qweq').get() print repr(v) vs = self.client.vertices().get() print repr(vs) self.assertEqual(len(vs) - init_cnt, 1) self.client.delete_vertex(v['_id']).get() self.assertEqual(len(self.client.vertices().get()), init_cnt) def test_query_vertices(self): print 'before', repr(self.client.vertices().get()) v1 = self.client.create_vertex(a = 123, b = 'qwe').get() v2 = self.client.create_vertex(a = 1234, b = 'qwe').get() print 'all', repr(self.client.vertices().get()) print '12', repr(self.client.vertices('a', 12).get()) print '123', repr(self.client.vertices('a', 123).get()) print '1234', repr(self.client.vertices('a', 1234).get()) print 'qwe', repr(self.client.vertices('b', 'qwe')) self.assertEqual(len(self.client.vertices('a', 12).get()), 0) self.assertEqual(len(self.client.vertices('a', 123).get()), 1) self.assertEqual(len(self.client.vertices('a', 1234).get()), 1) self.assertEqual(len(self.client.vertices('b', 'qwe').get()), 2) self.client.delete_vertex(v1['_id']).get() self.client.delete_vertex(v2['_id']).get() def test_scripts(self): script = ''' qq = 123; 1 + a + qq''' self.assertEqual(self.client.run_script_on_graph(script, a = 22).get()[0], 146) self.assertEqual(self.client.run_script_on_graph('[1, 2, a]', a = 22).get(), [1, 2, 22]) def test_lookup(self): v1 = self.client.create_vertex(a = 123, b = 'qwe').get() v2 = self.client.create_vertex(a = 1234, b = 'qwe').get() v3 = self.client.create_vertex(a = 1234, b = 'qweq').get() self.assertEqual(len(self.client.lookup_vertex(a = 123).get()), 1) self.assertEqual(len(self.client.lookup_vertex(Q(a = 123)).get()), 1) self.assertEqual(len(self.client.lookup_vertex(Q(_properties = [['a', Cmp.GTE, 123]])).get()), 3) self.assertEqual(len(self.client.lookup_vertex(Q(_properties = [['a', Cmp.GT, 123]])).get()), 2) self.assertEqual(len(self.client.lookup_vertex(Q(_properties = [['a', Cmp.GT, 123]]) & Q(_properties = [['b', Cmp.EQ, 'qwe']])).get()), 1) self.assertEqual(len(self.client.lookup_vertex(Q(_properties = [['a', Cmp.GT, 123]]) & Q(_properties = [['b', Cmp.NE, 'qwe']])).get()), 1) self.client.delete_vertex(v1['_id']).get() self.client.delete_vertex(v2['_id']).get() self.client.delete_vertex(v3['_id']).get()