def test_import_valid_relationship(self):
num_imported, errors = import_relationships(
self.mock_db_map,
[["existing_rc2", ["existing_o2", "existing_o1"]]])
self.mock_db_map._add_wide_relationships.assert_called_once_with({
"name":
"existing_rc2_existing_o2__existing_o1",
"class_id":
2,
"object_id_list": (2, 1)
})
self.assertEqual(len(errors), 0)
def test_import_valid_relationship_with_object_name_in_multiple_classes(
self):
num_imported, errors = import_relationships(
self.mock_db_map,
[["existing_rc1", ["duplicate_name", "existing_o2"]]])
self.mock_db_map._add_wide_relationships.assert_called_once_with({
"name":
"existing_rc1_duplicate_name__existing_o2",
"class_id":
1,
"object_id_list": (3, 2)
})
self.assertEqual(len(errors), 0)
def import_relationships(self, relationships) -> int:
"""Import relationships from a list of relationship name and object name list tuples
Example::
relationships = [('relationship_class_name', ('object_name1', 'object_name2'))]
import_relationships(relationships)
Returns:
n_imported (int): Number of improrted entities
"""
n_imported, errors = import_functions.import_relationships(
self._db_map, relationships)
if errors:
self._handle_errors(errors)
return n_imported
def test_import_existing_relationship(self):
num_imported, errors = import_relationships(
self.mock_db_map,
[["existing_rc1", ["existing_o1", "existing_o2"]]])
self.mock_db_map._add_wide_relationships.assert_called_once_with()
self.assertEqual(len(errors), 0)
def test_import_relationship_with_invalid_object_name(self):
num_imported, errors = import_relationships(
self.mock_db_map,
[["existing_rc1", ["none_existing_object", "existing_o2"]]])
self.mock_db_map._add_wide_relationships.assert_called_once()
self.assertEqual(len(errors), 1)