fix: pass video_url as list of dicts to entity model

When the `video_url` field is passed to create an
entity model from the local config,
it should be in a list of dict format.

Since the local config provides `video_url` as a
list of strings, convert it before sending it to
the entity.

Author: JessicaJang

awsmp/cli.py

@@ -71,6 +71,7 @@ def entity_list(entity_type, filter_visibility):
 @click.argument("entity-id")
 def entity_show(entity_id):
     details = _driver.get_entity_details(entity_id)
+    breakpoint()
     print(json.dumps(details, indent=2))
 
 

awsmp/models.py

@@ -387,7 +387,7 @@ class PromotionalResourcesModel(BaseModel):
     """
 
     LogoUrl: HttpUrl
-    Videos: List[dict[str, str]]
+    Videos: List[dict[str, Any]]
     AdditionalResources: YamlSupportResources
 
     @field_validator("AdditionalResources")
@@ -825,7 +825,9 @@ def get_entity_from_yaml(yaml_config: dict[str, Any]) -> EntityModel:
             },
             "PromotionalResources": {
                 "LogoUrl": ami_product.description.logourl,
-                "Videos": ami_product.description.video_urls,
+                "Videos": [{"Url": ami_product.description.video_urls[-1]}]
+                if ami_product.description.video_urls
+                else [],
                 "AdditionalResources": ami_product.description.additional_resources,
             },
             "SupportInformation": {

tests/test_models.py

@@ -532,6 +532,22 @@ def test_yaml_to_entity_version(self, mock_boto3):
         entity_model = models.EntityModel.get_entity_from_yaml(local_config)
         assert entity_model.Versions.ReleaseNotes == "test_release_notes\n"
 
+    def test_yaml_to_entity_video_urls(self, mock_boto3):
+        with open("./tests/test_config.yaml", "r") as f:
+            local_config = yaml.safe_load(f)
+        local_config["product"]["description"]["video_urls"] = ["https://test.com"]
+
+        entity_model = models.EntityModel.get_entity_from_yaml(local_config)
+        assert entity_model.PromotionalResources.Videos == [HttpUrl("https://test.com")]
+
+    def test_yaml_to_entity_video_urls_empty(self, mock_boto3):
+        with open("./tests/test_config.yaml", "r") as f:
+            local_config = yaml.safe_load(f)
+        local_config["product"]["description"]["video_urls"] = []
+
+        entity_model = models.EntityModel.get_entity_from_yaml(local_config)
+        assert entity_model.PromotionalResources.Videos == []
+
     def test_yaml_to_entity_term(self, mock_boto3):
         with open("./tests/test_config.yaml", "r") as f:
             local_config = yaml.safe_load(f)