diff --git a/dataproc_jupyter_plugin/controllers/bigquery.py b/dataproc_jupyter_plugin/controllers/bigquery.py index d611814b..157a832c 100644 --- a/dataproc_jupyter_plugin/controllers/bigquery.py +++ b/dataproc_jupyter_plugin/controllers/bigquery.py @@ -141,7 +141,7 @@ async def get(self): class SearchController(APIHandler): @tornado.web.authenticated - async def get(self): + async def post(self): try: search_string = self.get_argument("search_string") type = self.get_argument("type") diff --git a/dataproc_jupyter_plugin/controllers/executor.py b/dataproc_jupyter_plugin/controllers/executor.py index 45fc9612..73f6163e 100644 --- a/dataproc_jupyter_plugin/controllers/executor.py +++ b/dataproc_jupyter_plugin/controllers/executor.py @@ -51,7 +51,7 @@ async def post(self): class DownloadOutputController(APIHandler): @tornado.web.authenticated - async def get(self): + async def post(self): try: composer_name = self.get_argument("composer") bucket_name = self.get_argument("bucket_name") diff --git a/dataproc_jupyter_plugin/tests/test_bigquery.py b/dataproc_jupyter_plugin/tests/test_bigquery.py index 05d5d287..828a3dee 100644 --- a/dataproc_jupyter_plugin/tests/test_bigquery.py +++ b/dataproc_jupyter_plugin/tests/test_bigquery.py @@ -183,6 +183,8 @@ async def mock_config(config_field): "system": mock_system, "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 index 8544d3ba..67638e40 100644 --- a/dataproc_jupyter_plugin/tests/test_executor.py +++ b/dataproc_jupyter_plugin/tests/test_executor.py @@ -129,6 +129,8 @@ async def mock_list_dag_run_task(*args, **kwargs): "dag_id": mock_dag_id, "dag_run_id": mock_dag_run_id, }, + method="POST", + allow_nonstandard_methods=True, ) assert response.code == 200 payload = json.loads(response.body) @@ -149,6 +151,8 @@ async def test_invalid_composer_name(monkeypatch, jp_fetch): "dag_id": mock_dag_id, "dag_run_id": mock_dag_run_id, }, + method="POST", + allow_nonstandard_methods=True, ) assert response.code == 200 payload = json.loads(response.body) @@ -171,6 +175,8 @@ async def test_invalid_bucket_name(monkeypatch, jp_fetch): "dag_id": mock_dag_id, "dag_run_id": mock_dag_run_id, }, + method="POST", + allow_nonstandard_methods=True, ) assert response.code == 200 payload = json.loads(response.body) @@ -193,6 +199,8 @@ async def test_invalid_dag_id(monkeypatch, jp_fetch): "dag_id": mock_dag_id, "dag_run_id": mock_dag_run_id, }, + method="POST", + allow_nonstandard_methods=True, ) assert response.code == 200 payload = json.loads(response.body) @@ -215,6 +223,8 @@ async def test_invalid_dag_run_id(monkeypatch, jp_fetch): "dag_id": mock_dag_id, "dag_run_id": mock_dag_run_id, }, + method="POST", + allow_nonstandard_methods=True, ) assert response.code == 200 payload = json.loads(response.body) diff --git a/src/bigQuery/bigQueryService.tsx b/src/bigQuery/bigQueryService.tsx index 3cd3ff53..e45c27f5 100644 --- a/src/bigQuery/bigQueryService.tsx +++ b/src/bigQuery/bigQueryService.tsx @@ -407,7 +407,10 @@ export class BigQueryService { setSearchLoading(true); try { const data: any = await requestAPI( - `bigQuerySearch?search_string=${searchTerm}&type=(table|dataset)&system=bigquery` + `bigQuerySearch?search_string=${searchTerm}&type=(table|dataset)&system=bigquery`, + { + method: 'POST' + } ); setSearchResponse(data); } catch (reason) { diff --git a/src/scheduler/schedulerServices.tsx b/src/scheduler/schedulerServices.tsx index 2116edc8..b675538a 100644 --- a/src/scheduler/schedulerServices.tsx +++ b/src/scheduler/schedulerServices.tsx @@ -702,7 +702,9 @@ export class SchedulerService { try { dagRunId = encodeURIComponent(dagRunId); const serviceURL = `downloadOutput?composer=${composerName}&bucket_name=${bucketName}&dag_id=${dagId}&dag_run_id=${dagRunId}`; - const formattedResponse: any = await requestAPI(serviceURL); + const formattedResponse: any = await requestAPI(serviceURL, { + method: 'POST' + }); dagRunId = decodeURIComponent(dagRunId); if (formattedResponse.status === 0) { toast.success(