@@ -286,11 +286,6 @@ def mount(self, prefix, adapter):
286286
287287
288288class TestHttpOpSensor :
289- def setup_method (self ):
290- args = {"owner" : "airflow" , "start_date" : DEFAULT_DATE_ISO }
291- dag = DAG (TEST_DAG_ID , schedule = None , default_args = args )
292- self .dag = dag
293-
294289 @mock .patch ("airflow.providers.http.hooks.http.Session" , FakeSession )
295290 def test_get (self ):
296291 op = HttpOperator (
@@ -299,9 +294,8 @@ def test_get(self):
299294 endpoint = "/search" ,
300295 data = {"client" : "ubuntu" , "q" : "airflow" },
301296 headers = {},
302- dag = self .dag ,
303297 )
304- op .run ( start_date = DEFAULT_DATE , end_date = DEFAULT_DATE , ignore_ti_state = True )
298+ op .execute ({} )
305299
306300 @mock .patch ("airflow.providers.http.hooks.http.Session" , FakeSession )
307301 def test_get_response_check (self ):
@@ -312,9 +306,8 @@ def test_get_response_check(self):
312306 data = {"client" : "ubuntu" , "q" : "airflow" },
313307 response_check = lambda response : ("apache/airflow" in response .text ),
314308 headers = {},
315- dag = self .dag ,
316309 )
317- op .run ( start_date = DEFAULT_DATE , end_date = DEFAULT_DATE , ignore_ti_state = True )
310+ op .execute ({} )
318311
319312 @pytest .mark .skipif (not AIRFLOW_V_3_0_PLUS , reason = "Test only for Airflow 3.0+" )
320313 @mock .patch ("airflow.providers.http.hooks.http.Session" , FakeSession )
@@ -328,13 +321,13 @@ def test_sensor(self, run_task):
328321 response_check = lambda response : f"apache/airflow/{ DEFAULT_DATE :%Y-%m-%d} " in response .text ,
329322 poke_interval = 5 ,
330323 timeout = 15 ,
331- dag = self .dag ,
332324 )
333325 run_task (sensor )
334326
335327 @pytest .mark .skipif (AIRFLOW_V_3_0_PLUS , reason = "Test only for Airflow < 3.0" )
336328 @mock .patch ("airflow.providers.http.hooks.http.Session" , FakeSession )
337329 def test_sensor_af2 (self ):
330+ dag = DAG (TEST_DAG_ID , schedule = None )
338331 sensor = HttpSensor (
339332 task_id = "http_sensor_check" ,
340333 http_conn_id = "http_default" ,
@@ -344,7 +337,7 @@ def test_sensor_af2(self):
344337 response_check = lambda response : f"apache/airflow/{ DEFAULT_DATE :%Y-%m-%d} " in response .text ,
345338 poke_interval = 5 ,
346339 timeout = 15 ,
347- dag = self . dag ,
340+ dag = dag ,
348341 )
349342 sensor .run (start_date = DEFAULT_DATE , end_date = DEFAULT_DATE , ignore_ti_state = True )
350343
0 commit comments