11"""Tests for info card format definitions."""
22
3- from adcp import FormatId
3+ from adcp import FormatId , get_required_assets
44
55from creative_agent .data .format_types import AssetType , Type
66from creative_agent .data .standard_formats import AGENT_URL , INFO_CARD_FORMATS , filter_formats
@@ -95,12 +95,13 @@ def test_requires_product_assets(self):
9595 assert fmt .assets
9696 assert len (fmt .assets ) == 9 # 8 original + impression_tracker
9797
98- # Check required assets are in assets_required (backward compatibility)
99- assert fmt .assets_required
100- assert len (fmt .assets_required ) == 3 # Only required=True assets
98+ # Check required assets
99+ required_assets = get_required_assets (fmt )
100+ assert required_assets
101+ assert len (required_assets ) == 3 # Only required=True assets
101102
102- # Check required asset ids in assets_required
103- required_asset_ids = {get_asset_attr (asset , "asset_id" ) for asset in fmt . assets_required }
103+ # Check required asset ids
104+ required_asset_ids = {get_asset_attr (asset , "asset_id" ) for asset in required_assets }
104105 assert "product_image" in required_asset_ids
105106 assert "product_name" in required_asset_ids
106107 assert "product_description" in required_asset_ids
@@ -114,7 +115,7 @@ def test_requires_product_assets(self):
114115 assert asset_type_map ["product_description" ] == "text"
115116 assert asset_type_map ["impression_tracker" ] == "url"
116117
117- # Check optional assets are in assets but not in assets_required
118+ # Check optional assets are in assets but not required
118119 optional_asset_ids = {
119120 get_asset_attr (asset , "asset_id" ) for asset in fmt .assets if not get_asset_attr (asset , "required" )
120121 }
@@ -149,12 +150,13 @@ def test_requires_product_assets(self):
149150 assert fmt .assets
150151 assert len (fmt .assets ) == 9 # 8 original + impression_tracker
151152
152- # Check required assets are in assets_required (backward compatibility)
153- assert fmt .assets_required
154- assert len (fmt .assets_required ) == 3 # Only required=True assets
153+ # Check required assets
154+ required_assets = get_required_assets (fmt )
155+ assert required_assets
156+ assert len (required_assets ) == 3 # Only required=True assets
155157
156- # Check required asset ids in assets_required
157- required_asset_ids = {get_asset_attr (asset , "asset_id" ) for asset in fmt . assets_required }
158+ # Check required asset ids
159+ required_asset_ids = {get_asset_attr (asset , "asset_id" ) for asset in required_assets }
158160 assert "product_image" in required_asset_ids
159161 assert "product_name" in required_asset_ids
160162 assert "product_description" in required_asset_ids
@@ -168,7 +170,7 @@ def test_requires_product_assets(self):
168170 assert asset_type_map ["product_description" ] == "text"
169171 assert asset_type_map ["impression_tracker" ] == "url"
170172
171- # Check optional assets are in assets but not in assets_required
173+ # Check optional assets are in assets but not required
172174 optional_asset_ids = {
173175 get_asset_attr (asset , "asset_id" ) for asset in fmt .assets if not get_asset_attr (asset , "required" )
174176 }
@@ -198,9 +200,10 @@ def test_requires_format_asset(self):
198200 format_id = FormatId (agent_url = AGENT_URL , id = "format_card_standard" )
199201 results = filter_formats (format_ids = [format_id ])
200202 fmt = results [0 ]
201- assert fmt .assets_required
202- assert len (fmt .assets_required ) == 1
203- asset = fmt .assets_required [0 ]
203+ required_assets = get_required_assets (fmt )
204+ assert required_assets
205+ assert len (required_assets ) == 1
206+ asset = required_assets [0 ]
204207 assert get_asset_attr (asset , "asset_id" ) == "format"
205208 assert get_asset_attr (asset , "asset_type" ) == "text"
206209 assert get_asset_attr (asset , "required" ) is True
@@ -228,9 +231,10 @@ def test_requires_format_asset(self):
228231 format_id = FormatId (agent_url = AGENT_URL , id = "format_card_detailed" )
229232 results = filter_formats (format_ids = [format_id ])
230233 fmt = results [0 ]
231- assert fmt .assets_required
232- assert len (fmt .assets_required ) == 1
233- asset = fmt .assets_required [0 ]
234+ required_assets = get_required_assets (fmt )
235+ assert required_assets
236+ assert len (required_assets ) == 1
237+ asset = required_assets [0 ]
234238 assert get_asset_attr (asset , "asset_id" ) == "format"
235239 assert get_asset_attr (asset , "asset_type" ) == "text"
236240 assert get_asset_attr (asset , "required" ) is True
0 commit comments