summaryrefslogtreecommitdiff
path: root/src/server-tests
diff options
context:
space:
mode:
authorLV-426 <lv-426@taproot.org.il>2015-01-05 19:51:59 +0200
committerLV-426 <lv-426@taproot.org.il>2015-01-06 17:30:14 +0200
commitad5d358b20b2db90fc60b86ab4156f2a7552a2ca (patch)
treee524518506b69c99862658971e31668bfd50da7d /src/server-tests
parent7925b292faec4b1a52f89816b307d1201e3e91d8 (diff)
test_add_link_set(): update: use rand_label(), etc
Diffstat (limited to 'src/server-tests')
-rw-r--r--src/server-tests/test_db_controller.py26
1 files changed, 13 insertions, 13 deletions
diff --git a/src/server-tests/test_db_controller.py b/src/server-tests/test_db_controller.py
index 6d608dd0..895ad2bd 100644
--- a/src/server-tests/test_db_controller.py
+++ b/src/server-tests/test_db_controller.py
@@ -84,10 +84,10 @@ class TestDBController(unittest.TestCase):
i = i + 1
def test_add_node_set(self):
- n_label = neo4j_test_util.rand_label()
- n_0 = test_util.generate_random_node_dict(n_label)
- n_1 = test_util.generate_random_node_dict(n_label)
- n_map = { n_label: [n_0, n_1] }
+ test_label = neo4j_test_util.rand_label()
+ n_0 = test_util.generate_random_node_dict(test_label)
+ n_1 = test_util.generate_random_node_dict(test_label)
+ n_map = { test_label: [n_0, n_1] }
op = DBO_add_node_set(n_map)
self.assertEqual(len(op.statement_set), 1) # assert a single statement is issued
@@ -96,17 +96,17 @@ class TestDBController(unittest.TestCase):
self.assertEqual(len(id_set), 2)
def test_add_link_set(self):
- src_id = rand_id()
- dst_id_0 = rand_id()
- dst_id_1 = rand_id()
- n_map = { 'T_test_add_node_set': [{'id': src_id },
- {'id': dst_id_0 },
- {'id': dst_id_1 }] }
- self.db_ctl.exec_op(DBO_add_node_set(n_map))
+ test_label = neo4j_test_util.rand_label()
+ n_0, n_0_id = test_util.generate_random_node_dict(test_label)
+ n_1, n_1_id = test_util.generate_random_node_dict(test_label)
+ n_2, n_2_id = test_util.generate_random_node_dict(test_label)
- l_map = { 'T_test_add_link_set' : [{'__src': src_id, '__dst': dst_id_0},
- {'__src': src_id, '__dst': dst_id_1}] }
+ n_map = { test_label: [n_0, n_1, n_2] }
+ op = DBO_add_node_set(n_map)
+ self.db_ctl.exec_op(op)
+ l_map = { test_label : [Link.link_ptr(n_0_id, n_1_id),
+ Link.link_ptr(n_0_id, n_2_id)]}
op = DBO_add_link_set(l_map)
self.assertEqual(len(op.statement_set), 2) # no support yet for parameterized statements for link creation