def test_import_valid_objects(self):
num_imported, errors = import_objects(self.mock_db_map,
[["existing_oc1", "new_object"]])
self.mock_db_map._add_objects.assert_called_once_with({
"name": "new_object",
"class_id": 1
})
self.assertEqual(len(errors), 0)
def import_objects(self, objects) -> int:
"""Add objects of specific class from a list of class name and object name tuples
Returns:
n_imported (int): Number of imported entities
"""
n_imported, errors = import_functions.import_objects(
self._db_map, objects)
if errors:
self._handle_errors(errors)
return n_imported
def test_import_existing_object(self):
num_imported, errors = import_objects(
self.mock_db_map, [["existing_oc1", "existing_o1"]])
self.mock_db_map._add_objects.assert_called_once_with()
self.assertEqual(len(errors), 0)
def test_import_object_with_invalid_object_class_name(self):
num_imported, errors = import_objects(
self.mock_db_map, [["invalid_class_name", "new_object"]])
self.mock_db_map._add_objects.assert_called_once()
self.assertEqual(len(errors), 1)