test file review comment changes

GoogleCloudDataproc · Sep 24, 2024 · 23d0d08 · 23d0d08
1 parent f9a668b
commit 23d0d08
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/dataproc_jupyter_plugin/tests/test_bigquery.py b/dataproc_jupyter_plugin/tests/test_bigquery.py
@@ -184,6 +184,7 @@ async def mock_config(config_field):
             "type": mock_type,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)["results"][0]

diff --git a/dataproc_jupyter_plugin/tests/test_executor.py b/dataproc_jupyter_plugin/tests/test_executor.py
@@ -130,6 +130,7 @@ async def mock_list_dag_run_task(*args, **kwargs):
             "dag_run_id": mock_dag_run_id,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)
@@ -151,6 +152,7 @@ async def test_invalid_composer_name(monkeypatch, jp_fetch):
             "dag_run_id": mock_dag_run_id,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)
@@ -174,6 +176,7 @@ async def test_invalid_bucket_name(monkeypatch, jp_fetch):
             "dag_run_id": mock_dag_run_id,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)
@@ -197,6 +200,7 @@ async def test_invalid_dag_id(monkeypatch, jp_fetch):
             "dag_run_id": mock_dag_run_id,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)
@@ -220,6 +224,7 @@ async def test_invalid_dag_run_id(monkeypatch, jp_fetch):
             "dag_run_id": mock_dag_run_id,
         },
         method="POST",
+        allow_nonstandard_methods=True,
     )
     assert response.code == 200
     payload = json.loads(response.body)