From 00f533500ea9bae8fbf3e0fa40b7af129fddebb7 Mon Sep 17 00:00:00 2001 From: Daisie Huang Date: Wed, 2 Oct 2024 14:33:33 -0700 Subject: [PATCH] Update test_uniform_federation.py --- tests/test_uniform_federation.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/tests/test_uniform_federation.py b/tests/test_uniform_federation.py index 46abb40..b6a1cb4 100644 --- a/tests/test_uniform_federation.py +++ b/tests/test_uniform_federation.py @@ -417,7 +417,8 @@ def test_valid_federated_query_one_server_post(mock_requests, mock_session, clie "payload": "", "method": "POST", "service": TestParams["service"], - "unsafe": True}), + "unsafe": True + }), headers=Headers(fedHeader.headers) ): RO, Status = operations.post_search() @@ -453,7 +454,9 @@ def test_valid_federated_query_two_server_get(mock_requests, mock_session, clien data=json.dumps({"path": TestParams["path"], "payload": "", "method": "GET", - "service": TestParams["service"]}), + "service": TestParams["service"], + "unsafe": True + }), headers=Headers(fedHeader.headers) ): RO, Status = operations.post_search() @@ -487,7 +490,9 @@ def test_valid_federated_query_two_server_post(mock_requests, mock_session, clie data=json.dumps({"path": TestParams["path"], "payload": "", "method": "POST", - "service": TestParams["service"]}), + "service": TestParams["service"], + "unsafe": True + }), headers=Headers(fedHeader.headers) ): RO, Status = operations.post_search() @@ -507,8 +512,9 @@ def test_invalid_backslash_endpoint_start(mock_requests, mock_session, client, t data=json.dumps({"path": "/fail/this/path", "payload": "", "method": "GET", - "service": TestParams["service"] - }), + "service": TestParams["service"], + "unsafe": True + }), headers=Headers(fedHeader.headers) ): RO, Status = operations.post_search()