class TestRefCanResolveResourceIdRefs(TestCase):
def setUp(self):
self.supported_resource_id_refs_mock = Mock()
self.ref = RefAction()
def test_must_replace_refs(self):
resolved_value = "NewLogicalId"
input = {
"Ref": "LogicalId"
}
expected = {
"Ref": resolved_value
}
self.supported_resource_id_refs_mock.get.return_value = resolved_value
output = self.ref.resolve_resource_id_refs(input, self.supported_resource_id_refs_mock)
self.assertEqual(expected, output)
self.supported_resource_id_refs_mock.get.assert_called_once_with("LogicalId")
def test_handle_unsupported_references(self):
input = {
"Ref": "OtherLogicalId.Property"
}
expected = {
"Ref": "OtherLogicalId.Property"
}
self.supported_resource_id_refs_mock.get.return_value = None
output = self.ref.resolve_resource_id_refs(input, self.supported_resource_id_refs_mock)
self.assertEqual(expected, output)
self.supported_resource_id_refs_mock.get.assert_not_called()
@patch.object(RefAction, "can_handle")
def test_return_value_if_cannot_handle(self, can_handle_mock):
input = {
"Ref": "key"
}
expected = {
"Ref": "key"
}
ref = RefAction()
can_handle_mock.return_value = False # Simulate failure to handle the input. Result should be same as input
self.assertEqual(expected, ref.resolve_resource_id_refs(input, self.supported_resource_id_refs_mock))
def test_return_value_if_cannot_handle(self, can_handle_mock):
input = {"Ref": "key"}
expected = {"Ref": "key"}
ref = RefAction()
can_handle_mock.return_value = False # Simulate failure to handle the input. Result should be same as input
self.assertEquals(
expected,
ref.resolve_resource_id_refs(input,
self.supported_resource_id_refs_mock))
def test_return_value_if_cannot_handle(self, can_handle_mock):
input = {
"Ref": "key"
}
expected = {
"Ref": "key"
}
ref = RefAction()
can_handle_mock.return_value = False # Simulate failure to handle the input. Result should be same as input
self.assertEqual(expected, ref.resolve_resource_id_refs(input, self.supported_resource_id_refs_mock))
class TestRefCanResolveResourceIdRefs(TestCase):
def setUp(self):
self.supported_resource_id_refs_mock = Mock()
self.ref = RefAction()
def test_must_replace_refs(self):
resolved_value = "NewLogicalId"
input = {"Ref": "LogicalId"}
expected = {"Ref": resolved_value}
self.supported_resource_id_refs_mock.get.return_value = resolved_value
output = self.ref.resolve_resource_id_refs(
input, self.supported_resource_id_refs_mock)
self.assertEqual(expected, output)
self.supported_resource_id_refs_mock.get.assert_called_once_with(
"LogicalId")
def test_handle_unsupported_references(self):
input = {"Ref": "OtherLogicalId.Property"}
expected = {"Ref": "OtherLogicalId.Property"}
self.supported_resource_id_refs_mock.get.return_value = None
output = self.ref.resolve_resource_id_refs(
input, self.supported_resource_id_refs_mock)
self.assertEqual(expected, output)
self.supported_resource_id_refs_mock.get.assert_not_called()
@patch.object(RefAction, "can_handle")
def test_return_value_if_cannot_handle(self, can_handle_mock):
input = {"Ref": "key"}
expected = {"Ref": "key"}
ref = RefAction()
can_handle_mock.return_value = False # Simulate failure to handle the input. Result should be same as input
self.assertEqual(
expected,
ref.resolve_resource_id_refs(input,
self.supported_resource_id_refs_mock))