diff options
| author | LV-426 <lv-426@taproot.org.il> | 2015-01-06 16:43:04 +0200 |
|---|---|---|
| committer | LV-426 <lv-426@taproot.org.il> | 2015-01-06 17:30:14 +0200 |
| commit | 85dfe16db3b03d60f3e11f720c2646adb79f6e19 (patch) | |
| tree | 3955d63d37b8f91acfbf8eb3d1660527b385c374 /src/server-tests/test_db_controller.py | |
| parent | c75c98c4ca8aae246592eff7c1fba4bac520eb33 (diff) | |
test_diff_commit__topo(): adapt to namedtuple return type
Diffstat (limited to 'src/server-tests/test_db_controller.py')
| -rw-r--r-- | src/server-tests/test_db_controller.py | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/src/server-tests/test_db_controller.py b/src/server-tests/test_db_controller.py index 449aa318..45ab9586 100644 --- a/src/server-tests/test_db_controller.py +++ b/src/server-tests/test_db_controller.py @@ -150,13 +150,21 @@ class TestDBController(unittest.TestCase): topo_diff = Topo_Diff(node_set_add=n_set, link_set_add=l_set) - # test return set lengths + # commit diff op = DBO_diff_commit__topo(topo_diff) ret_topo_diff = self.db_ctl.exec_op(op) - self.assertEqual(len(ret_topo_diff.node_set_add), len(n_set)) - self.assertEqual(len(ret_topo_diff.link_set_add), len(l_set)) - self.assertEqual(len(ret_topo_diff.node_set_rm), 0) - self.assertEqual(len(ret_topo_diff.link_set_rm), 0) + + # test return type + self.assertTrue(hasattr(ret_topo_diff, 'node_id_set_add')) + self.assertTrue(hasattr(ret_topo_diff, 'link_id_set_add')) + self.assertTrue(hasattr(ret_topo_diff, 'node_id_set_rm')) + self.assertTrue(hasattr(ret_topo_diff, 'link_id_set_rm')) + + # test return set lengths + self.assertEqual(len(ret_topo_diff.node_id_set_add), len(n_set)) + self.assertEqual(len(ret_topo_diff.link_id_set_add), len(l_set)) + self.assertEqual(len(ret_topo_diff.node_id_set_rm), 0) + self.assertEqual(len(ret_topo_diff.link_id_set_rm), 0) # assert nodes persisted id_set = self.db_ctl.exec_op(DBO_match_node_set_by_id_attribute([n_0_id, n_1_id])) @@ -173,7 +181,7 @@ class TestDBController(unittest.TestCase): topo_diff = Topo_Diff(link_set_rm=[l_0_id, l_1_id]) op = DBO_diff_commit__topo(topo_diff) ret_topo_diff = self.db_ctl.exec_op(op) - self.assertEqual(len(ret_topo_diff.link_set_rm), 2) + self.assertEqual(len(ret_topo_diff.link_id_set_rm), 2) # assert links removed op = DBO_load_link_set.init_from_link_ptr_set([l_ptr_0, l_ptr_1]) @@ -184,7 +192,7 @@ class TestDBController(unittest.TestCase): topo_diff = Topo_Diff(node_set_rm=[n_2_id]) op = DBO_diff_commit__topo(topo_diff) ret_topo_diff = self.db_ctl.exec_op(op) - self.assertEqual(len(ret_topo_diff.node_set_rm), 1) + self.assertEqual(len(ret_topo_diff.node_id_set_rm), 1) # assert nodes removed op = DBO_match_node_set_by_id_attribute([n_2_id]) |