DBO_load_link_id_set & test

1 files changed, 9 insertions, 0 deletions

diff --git a/src-py/test_db_controller.py b/src-py/test_db_controller.py
index 446d9288..176fce4b 100644
--- a/src-py/test_db_controller.py
+++ b/src-py/test_db_controller.py
@@ -94,6 +94,15 @@ class TestDBController(unittest.TestCase):
         n_set = self.db_ctl.exec_op(dbc.DBO_load_node_set_by_id_attribute(['skill_00', 'person_01']))
         self.assertEqual(len(n_set), 2)
 
+    def test_load_link_set_by_type(self):
+        op = dbc.DBO_load_link_id_set(filter_type='Knows')
+        id_set = self.db_ctl.exec_op(op)
+        self.assertEqual(len(id_set), 2)
+
+        op = dbc.DBO_load_link_id_set(filter_type='Nan_Type')
+        id_set = self.db_ctl.exec_op(op)
+        self.assertEqual(len(id_set), 0)
+
     def test_load_link_set_by_src_or_dst_id_attributes(self):
         op = dbc.DBO_load_link_set_by_src_or_dst_id_attributes(src_id='person_00', dst_id='skill_00')
         n_set = self.db_ctl.exec_op(op)