|
35 | 35 | TEST_WASB_CONN_ID = "wasb_default"
|
36 | 36 | POKE_INTERVAL = 5.0
|
37 | 37 |
|
| 38 | +pytestmark = pytest.mark.db_test |
| 39 | + |
38 | 40 |
|
39 | 41 | class TestWasbBlobSensorTrigger:
|
40 | 42 | TRIGGER = WasbBlobSensorTrigger(
|
@@ -78,7 +80,6 @@ async def test_running(self, mock_check_for_blob, blob_exists):
|
78 | 80 | assert task.done() is False
|
79 | 81 | asyncio.get_event_loop().stop()
|
80 | 82 |
|
81 |
| - @pytest.mark.db_test |
82 | 83 | @pytest.mark.asyncio
|
83 | 84 | @mock.patch("airflow.providers.microsoft.azure.hooks.wasb.WasbAsyncHook.check_for_blob_async")
|
84 | 85 | async def test_success(self, mock_check_for_blob):
|
@@ -115,7 +116,6 @@ async def test_waiting_for_blob(self, mock_check_for_blob, caplog):
|
115 | 116 | assert message in caplog.text
|
116 | 117 | asyncio.get_event_loop().stop()
|
117 | 118 |
|
118 |
| - @pytest.mark.db_test |
119 | 119 | @pytest.mark.asyncio
|
120 | 120 | @mock.patch("airflow.providers.microsoft.azure.hooks.wasb.WasbAsyncHook.check_for_blob_async")
|
121 | 121 | async def test_trigger_exception(self, mock_check_for_blob):
|
@@ -170,7 +170,6 @@ async def test_running(self, mock_check_for_prefix, prefix_exists):
|
170 | 170 | assert task.done() is False
|
171 | 171 | asyncio.get_event_loop().stop()
|
172 | 172 |
|
173 |
| - @pytest.mark.db_test |
174 | 173 | @pytest.mark.asyncio
|
175 | 174 | @mock.patch("airflow.providers.microsoft.azure.hooks.wasb.WasbAsyncHook.check_for_prefix_async")
|
176 | 175 | async def test_success(self, mock_check_for_prefix):
|
@@ -208,7 +207,6 @@ async def test_waiting_for_blob(self, mock_check_for_prefix):
|
208 | 207 | mock_log_info.assert_called_once_with(message)
|
209 | 208 | asyncio.get_event_loop().stop()
|
210 | 209 |
|
211 |
| - @pytest.mark.db_test |
212 | 210 | @pytest.mark.asyncio
|
213 | 211 | @mock.patch("airflow.providers.microsoft.azure.hooks.wasb.WasbAsyncHook.check_for_prefix_async")
|
214 | 212 | async def test_trigger_exception(self, mock_check_for_prefix):
|
|
0 commit comments