def test_get_region_id_or_name_error(self, mock_db_session):
mock_get_session = mock.MagicMock()
mock_get_session.get_session.side_effect = ValueError('test')
mock_db_session.return_value = mock_get_session
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(ValueError, my_data_manager.get_region_by_id_or_name,
"id")
def test_get_all_regions(self, mock_db_session):
all_regions = [return_region]
mock_db_session.return_value = MyFacade(query=all_regions)
my_data_manager = data_manager.DataManager("url", "", "")
result = my_data_manager.get_all_regions()
self.assertEqual(len(result), len(all_regions))
self.assertEqual(result[0].id, all_regions[0].region_id)
def test_add_region_sanity(self, mock_db_session):
"""Test that no exception is raised when calling add_status_record."""
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.add_region("reg1", "region1", "a_state", "a_country",
"a_city", "a_street", "a_zip", "a_status",
"ranger_agent_ver", "os_ver", "design_type",
"loc_type", "vlcp", "clli", end_point_list,
meta_data_dict, "a_desc")
def test_group_functions_sanity(self, mock_db_session):
"""Test that no exception is raised when calling group functions."""
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.delete_group('group')
my_data_manager.get_group('group')
my_data_manager.update_group('id', 'name', 'description', ['region'])
my_data_manager.get_all_groups()
def test_add_region_duplicate_error(self, mock_db_session):
"""Test that duplicate exception is raised when calling add_status_record."""
my_data_manager = data_manager.DataManager("url", "", "")
with self.assertRaises(DuplicateEntryError):
my_data_manager.add_region("reg1", "region1", "a_state",
"a_country", "a_city", "a_street",
"a_zip", "a_status", "ranger_agent_ver",
"os_ver", "design_type", "loc_type",
"vlcp", "clli", [], {}, "a_desc")
def test_update_region_region_not_found(self, mock_engine_facade):
"""Test that NotFoundError is raised when calling update_region."""
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(error_base.NotFoundError,
my_data_manager.update_region, "reg1", "region1",
"region_name", "a_state", "a_country", "a_city",
"a_street", "a_zip", "a_status", "ranger_agent_ver",
"os_ver", "design_type", "loc_type", "vlcp", "clli",
end_point_list, meta_data_dict, "a_desc")
def test_update_region_other_error(self, mock_engine_facade):
"""Test that ValueError is raised when calling update_region."""
mock_session = mock.MagicMock()
mock_session.get_session.side_effect = ValueError('test')
mock_engine_facade.return_value = mock_session
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(ValueError, my_data_manager.update_region, "reg1",
"region1", "region_name", "a_state", "a_country",
"a_city", "a_street", "a_zip", "a_status",
"ranger_agent_ver", "os_ver", "design_type",
"loc_type", "vlcp", "clli", end_point_list,
meta_data_dict, "a_desc")
def test_add_meta_data_to_region_error(self, mock_db_session):
mock_begin = mock.MagicMock()
mock_begin.begin.side_effect = data_manager.oslo_db.exception.DBDuplicateEntry(
'test')
mock_get_session = mock.MagicMock()
mock_get_session.get_session.return_value = mock_begin
mock_db_session.return_value = mock_get_session
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(data_manager.error_base.ConflictError,
my_data_manager.add_meta_data_to_region, 'region',
{'meta': 'data'})
def test_get_regions(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
result = my_data_manager.get_regions(
region_filters_dict={"meta_data_key": "1"},
meta_data_dict=None,
end_point_dict=None)
self.assertEqual(result[0].id, return_region.region_id)
# Test that no exception is raised on the other successful flow
mock_db_session.return_value = mock.MagicMock()
my_data_manager.get_regions(region_filters_dict=None,
meta_data_dict={
"meta_data_keys": ["1"],
"meta_data_pairs": []
},
end_point_dict=end_point_list[0])
def test_delete_region_sanity(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.delete_region("region")
def test_update_region_status_region_not_found(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(error_base.NotFoundError,
my_data_manager.update_region_status, 'region',
'status')
def test_update_region_status_sanity(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.update_region_status('region', 'status')
def test_delete_region_metadata_region_not_found(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
self.assertRaises(error_base.NotFoundError,
my_data_manager.delete_region_metadata, 'region',
{'meta': 'data'})
def test_delete_region_metadata_sanity(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.delete_region_metadata('region', {'meta': 'data'})
def test_get_region_id_or_name_success(self, mock_db_session):
my_data_manager = data_manager.DataManager('url', "", "")
result = my_data_manager.get_region_by_id_or_name("id")
self.assertEqual(result.id, 'SNA1')
def test_add_group_sanity(self, mock_db_session):
"""Test that no exception is raised when calling add_group."""
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.add_group("group1", "group 1", "bla bla", ["region1"])
def test_get_region_id_or_name_None(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
result = my_data_manager.get_region_by_id_or_name("id")
self.assertEqual(result, None)
def test_add_meta_data_to_region_sanity(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
my_data_manager.add_meta_data_to_region('region', {'meta': 'data'})
def test_add_meta_data_to_region_region_not_found(self, mock_db_session):
my_data_manager = data_manager.DataManager("url", "", "")
result = my_data_manager.add_meta_data_to_region(
'region', {'meta': 'data'})
self.assertIsNone(result)
def test_add_group_duplicate_error(self, mock_db_session):
"""Test that ConflictError is raised when calling add_group."""
my_data_manager = data_manager.DataManager("url", "", "")
with self.assertRaises(error_base.ConflictError):
my_data_manager.add_group("group1", "group 1", "bla bla",
["region1"])
def test_add_group_reference_error(self, mock_db_session):
"""Test that reference exception is raised when calling add_group."""
my_data_manager = data_manager.DataManager("url", "", "")
with self.assertRaises(error_base.InputValueError):
my_data_manager.add_group("group1", "group 1", "bla bla",
["region1"])
