@@ -25,30 +25,9 @@ def test_create_orchestrator_link_with_valid_data(self):
2525 link = create_orchestrator_link (orchestrator_info )
2626
2727 assert link is not None
28- assert link .text == "View in Airflow"
2928 assert link .url == "https://airflow.example.com/run/12345"
3029 assert link .orchestrator == "airflow"
31- assert link .icon == Icon .INFO
32-
33- def test_create_orchestrator_link_formats_orchestrator_names (self ):
34- test_cases = [
35- ("airflow" , "View in Airflow" ),
36- ("dbt_cloud" , "View in Dbt Cloud" ),
37- ("github_actions" , "View in Github Actions" ),
38- ("custom_orchestrator" , "View in Custom Orchestrator" ),
39- ]
40-
41- for orchestrator , expected_text in test_cases :
42- orchestrator_info = {
43- "orchestrator" : orchestrator ,
44- "run_url" : "https://example.com/run/123" , # noqa: E231
45- }
46-
47- link = create_orchestrator_link (orchestrator_info )
48-
49- assert link is not None
50- assert link .text == expected_text
51- assert link .orchestrator == orchestrator
30+ assert link .icon == Icon .LINK
5231
5332 def test_create_orchestrator_link_without_url_returns_none (self ):
5433 orchestrator_info = {
@@ -290,20 +269,20 @@ class TestOrchestratorLinkDataModel:
290269 def test_orchestrator_link_data_creation (self ):
291270 link = OrchestratorLinkData (
292271 url = "https://airflow.example.com/run/123" ,
293- text = "View in Airflow " ,
272+ text = "View in airflow " ,
294273 orchestrator = "airflow" ,
295- icon = Icon .INFO ,
274+ icon = Icon .LINK ,
296275 )
297276
298277 assert link .url == "https://airflow.example.com/run/123"
299- assert link .text == "View in Airflow "
278+ assert link .text == "View in airflow "
300279 assert link .orchestrator == "airflow"
301- assert link .icon == Icon .INFO
280+ assert link .icon == Icon .LINK
302281
303282 def test_orchestrator_link_data_without_icon (self ):
304283 link = OrchestratorLinkData (
305284 url = "https://airflow.example.com/run/123" ,
306- text = "View in Airflow " ,
285+ text = "View in airflow " ,
307286 orchestrator = "airflow"
308287 # icon is optional
309288 )
@@ -315,5 +294,5 @@ def test_orchestrator_link_data_required_fields(self):
315294 with pytest .raises ((TypeError , ValueError )):
316295 OrchestratorLinkData (
317296 # Missing required fields
318- icon = Icon .INFO
297+ icon = Icon .LINK
319298 )
0 commit comments