def test_get_dbstatus_snapshot_failure(self):
os.environ["Region"] = "us-west-2"
factory_patch = patch('get_dbstatus_function.boto3.client')
mock_factory_boto_client = factory_patch.start()
mock_response = Mock(name='response')
mock_factory_boto_client.return_value = mock_response
mock_waiter = Mock()
mock_response.get_waiter.return_value = mock_waiter
waiter_error = Exception("Waiter encountered a terminal failure state")
mock_waiter.wait.side_effect = waiter_error
event = {
"output": {
"taskname": "SnapshotCreation",
"identifier": "database-1"
}
}
context = ContextManager(45)
data = get_dbstatus_function.lambda_get_dbinstance_status(
event, context)
self.assertEqual(data.get("taskname"), "SnapshotCreation")
self.assertEqual(
data.get("status"),
"Identifier:database-1 \nWaiter encountered a terminal failure state"
)
self.assertEqual(data.get("identifier"), "database-1")
self.assertEqual(data.get("task"), "TASK_FAILED")
def test_get_dbstatus_restore_max_attempts_exceeded_failure(self):
os.environ["Region"] = "us-west-2"
factory_patch = patch('get_dbstatus_function.boto3.client')
mock_factory_boto_client = factory_patch.start()
mock_response = Mock(name='response')
mock_waiter = Mock()
mock_response.get_waiter.return_value = mock_waiter
waiter_error = Exception("Waiter database-1 Max attempts exceeded")
mock_waiter.wait.side_effect = waiter_error
event = {"output": {"taskname": "Restore", "identifier": "database-1"}}
context = ContextManager(40)
try:
get_dbstatus_function.lambda_get_dbinstance_status(event, context)
except Exception as ex:
self.assertEqual(
str(ex),
"Identifier:database-1 \nWaiter database-1 Max attempts exceeded"
)
def test_get_dbstatus_rename_success(self):
os.environ["Region"] = "us-west-2"
factory_patch = patch('get_dbstatus_function.boto3.client')
mock_factory_boto_client = factory_patch.start()
mock_response = Mock(name='response')
mock_factory_boto_client.return_value = mock_response
mock_waiter = Mock()
mock_response.get_waiter.return_value = mock_waiter
mock_waiter.wait.side_effect = mock_waiter
event = {"output": {"taskname": "Rename", "identifier": "database-1"}}
context = ContextManager(45)
data = get_dbstatus_function.lambda_get_dbinstance_status(
event, context)
self.assertEqual(data.get("taskname"), "Rename")
self.assertEqual(data.get("identifier"), "database-1")
self.assertEqual(data.get("task"), "TASK_COMPLETE")