Add some tests

aws-cloudformation · May 25, 2024 · 8ecbe5c · 8ecbe5c
1 parent c5670d4
commit 8ecbe5c
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/test/unit/module/test_template.py b/test/unit/module/test_template.py
@@ -742,6 +742,15 @@ def test_get_object_without_conditions(self):
                                     {"Ref": "LaunchConfiguration"},
                                 ]
                             },
+                            "PropertyA": [
+                                {
+                                    "Fn::If": [
+                                        "useAmiId",
+                                        {"Ref": "AWS::NoValue"},
+                                        {"Ref": "LaunchConfiguration"},
+                                    ]
+                                }
+                            ],
                         },
                     }
                 },
@@ -761,7 +770,10 @@ def test_get_object_without_conditions(self):
             elif result["Scenario"] == {"useAmiId": False}:
                 self.assertDictEqual(
                     result["Object"],
-                    {"LaunchConfiguration": {"Ref": "LaunchConfiguration"}},
+                    {
+                        "LaunchConfiguration": {"Ref": "LaunchConfiguration"},
+                        "PropertyA": [{"Ref": "LaunchConfiguration"}],
+                    },
                 )
 
     def test_get_object_without_conditions_no_value(self):
@@ -797,6 +809,15 @@ def test_get_object_without_conditions_no_value(self):
                                 ]
                             },
                         ],
+                        "PropertyA": [
+                            {
+                                "Fn::If": [
+                                    "useAmiId",
+                                    {"Ref": "AWS::NoValue"},
+                                    {"Ref": "LaunchConfiguration"},
+                                ]
+                            }
+                        ],
                     },
                 }
             },