diff --git a/providers/microsoft/azure/tests/unit/microsoft/azure/hooks/test_msgraph.py b/providers/microsoft/azure/tests/unit/microsoft/azure/hooks/test_msgraph.py index 1f5e1f8d10ae6..a81e4df2e5a95 100644 --- a/providers/microsoft/azure/tests/unit/microsoft/azure/hooks/test_msgraph.py +++ b/providers/microsoft/azure/tests/unit/microsoft/azure/hooks/test_msgraph.py @@ -401,7 +401,6 @@ def test_handle_response_async_when_internal_server_error(self): # TODO: Elad: review this after merging the bump 2.10 PR # We should not have specific provider test block the release @pytest.mark.xfail(reason="TODO: Remove") - @pytest.mark.db_test def test_when_provider_min_airflow_version_is_2_10_or_higher_remove_obsolete_code(self): """ Once this test starts failing due to the fact that the minimum Airflow version is now 2.10.0 or higher diff --git a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_adx.py b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_adx.py index ec83ed1863f6c..0161338f86384 100644 --- a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_adx.py +++ b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_adx.py @@ -19,7 +19,6 @@ from unittest import mock -import pytest from azure.kusto.data._models import KustoResultTable from airflow.models import DAG @@ -82,7 +81,6 @@ def test_run_query(self, mock_conn, mock_run_query): ) -@pytest.mark.db_test @mock.patch.object(AzureDataExplorerHook, "run_query", return_value=MockResponse()) @mock.patch.object(AzureDataExplorerHook, "get_conn") def test_azure_data_explorer_query_operator_xcom_push_and_pull( diff --git a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_msgraph.py b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_msgraph.py index 6ae597ba0e31c..d6a9fae4bdcaf 100644 --- a/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_msgraph.py +++ b/providers/microsoft/azure/tests/unit/microsoft/azure/operators/test_msgraph.py @@ -44,7 +44,6 @@ class TestMSGraphAsyncOperator(Base): - @pytest.mark.db_test def test_execute_with_old_result_processor_signature(self): users = load_json_from_resources(dirname(__file__), "..", "resources", "users.json") next_users = load_json_from_resources(dirname(__file__), "..", "resources", "next_users.json") @@ -76,7 +75,6 @@ def test_execute_with_old_result_processor_signature(self): assert events[1].payload["type"] == "builtins.dict" assert events[1].payload["response"] == json.dumps(next_users) - @pytest.mark.db_test def test_execute_with_new_result_processor_signature(self): users = load_json_from_resources(dirname(__file__), "..", "resources", "users.json") next_users = load_json_from_resources(dirname(__file__), "..", "resources", "next_users.json") @@ -104,7 +102,6 @@ def test_execute_with_new_result_processor_signature(self): assert events[1].payload["type"] == "builtins.dict" assert events[1].payload["response"] == json.dumps(next_users) - @pytest.mark.db_test def test_execute_with_old_paginate_function_signature(self): users = load_json_from_resources(dirname(__file__), "..", "resources", "users.json") next_users = load_json_from_resources(dirname(__file__), "..", "resources", "next_users.json") @@ -139,7 +136,6 @@ def test_execute_with_old_paginate_function_signature(self): assert events[1].payload["type"] == "builtins.dict" assert events[1].payload["response"] == json.dumps(next_users) - @pytest.mark.db_test def test_execute_when_do_xcom_push_is_false(self): users = load_json_from_resources(dirname(__file__), "..", "resources", "users.json") users.pop("@odata.nextLink") @@ -162,7 +158,6 @@ def test_execute_when_do_xcom_push_is_false(self): assert events[0].payload["type"] == "builtins.dict" assert events[0].payload["response"] == json.dumps(users) - @pytest.mark.db_test def test_execute_when_an_exception_occurs(self): with self.patch_hook_and_request_adapter(AirflowException()): operator = MSGraphAsyncOperator( @@ -175,7 +170,6 @@ def test_execute_when_an_exception_occurs(self): with pytest.raises(AirflowException): execute_operator(operator) - @pytest.mark.db_test def test_execute_when_an_exception_occurs_on_custom_event_handler_with_old_signature(self): with self.patch_hook_and_request_adapter(AirflowException("An error occurred")): @@ -205,7 +199,6 @@ def custom_event_handler(context: Context, event: dict[Any, Any] | None = None): assert events[0].payload["status"] == "failure" assert events[0].payload["message"] == "An error occurred" - @pytest.mark.db_test def test_execute_when_an_exception_occurs_on_custom_event_handler_with_new_signature(self): with self.patch_hook_and_request_adapter(AirflowException("An error occurred")): @@ -231,7 +224,6 @@ def custom_event_handler(event: dict[Any, Any] | None = None, **context): assert events[0].payload["status"] == "failure" assert events[0].payload["message"] == "An error occurred" - @pytest.mark.db_test def test_execute_when_response_is_bytes(self): content = load_file_from_resources( dirname(__file__), "..", "resources", "dummy.pdf", mode="rb", encoding=None @@ -259,7 +251,6 @@ def test_execute_when_response_is_bytes(self): assert events[0].payload["type"] == "builtins.bytes" assert events[0].payload["response"] == base64_encoded_content - @pytest.mark.db_test def test_execute_with_lambda_parameter_when_response_is_bytes(self): content = load_file_from_resources( dirname(__file__), "..", "resources", "dummy.pdf", mode="rb", encoding=None