Fix: test for deferable operation

apache · May 2, 2024 · 67c7eda · 67c7eda
1 parent bac4ede
commit 67c7eda
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/providers/amazon/aws/operators/test_emr_containers.py b/tests/providers/amazon/aws/operators/test_emr_containers.py
@@ -152,8 +152,7 @@ def test_operator_defer_with_timeout(self, mock_submit_job, mock_check_query_sta
  self.emr_container.deferrable = True
  self.emr_container.max_polling_attempts = 1000
 
- error_match = "Waiter error: max attempts reached"
- with pytest.raises(AirflowException, match=error_match):
+ with pytest.raises(TaskDeferred):
  self.emr_container.execute(context=None)
 
  assert mock_check_query_status.call_count == 1000

diff --git a/tests/providers/amazon/aws/sensors/test_emr_containers.py b/tests/providers/amazon/aws/sensors/test_emr_containers.py
@@ -92,8 +92,7 @@ def test_sensor_defer_with_timeout(self, mock_check_query_status):
  self.sensor.deferrable = True
  self.sensor.max_polling_attempts = 1000
 
- error_match = "Waiter error: max attempts reached"
- with pytest.raises(TaskDeferred, match=error_match):
+ with pytest.raises(TaskDeferred):
  self.sensor.execute(context=None)
 
  assert mock_check_query_status.call_count == 1000