Update test_uniform_federation.py

CanDIG · Oct 2, 2024 · 00f5335 · 00f5335
1 parent 57e916a
commit 00f5335
Showing 1 changed file with 11 additions and 5 deletions.
diff --git 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()