def test_update_region_mapping_fail(self, mock_response, mock_accessor):
"""test sucessful update_region_mapping()"""
mock_common_accessor = mock_accessor().__enter__()
mock_response.return_value = MockResponse(self.test_data, 200)
mock_common_accessor.add.side_effect = IntegrityError('fake', 'fake', 'fake')
expected = 'WARNING:masu.util.aws.region_map:Duplicate entry in DB: "us-east-1" - "US East (N. Virginia)"'
logging.disable(logging.NOTSET)
with self.assertLogs('masu.util.aws.region_map', level='WARNING') as logger:
rmap.update_region_mapping()
self.assertIn(expected, logger.output)
def test_update_region_mapping_success(self, mock_response, mock_accessor):
"""test sucessful update_region_mapping()"""
mock_common_accessor = mock_accessor().__enter__()
mock_response.return_value = MockResponse(self.test_data, 200)
response = rmap.update_region_mapping()
self.assertEqual(str(response), 'True')
mock_common_accessor.add.assert_called()
def update_region_map():
"""Return download file async task ID."""
return jsonify(update_region_mapping())
def update_region_map(request):
"""Return download file async task ID."""
return Response(update_region_mapping())
