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"])