Updated tests with modifications to Web3 class

easierdata · Sep 25, 2024 · 7c6e3e5 · 7c6e3e5
1 parent 7f596a4
commit 7c6e3e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -17,6 +17,7 @@
 LOCAL_GATEWAY = "127.0.0.1"
 API_PORT = 5001
 STAC_ENDPOINT = "fake_endpoint"
+GATEWAY_PORT = 8081
 
 
 @patch('requests.get')
@@ -50,7 +51,7 @@ def test_to_pd_df(mock_getFromCID, mock_get):
 
 class TestWeb3(SetUp):
     def setUp(self):
-        self.client = Web3(local_gateway=LOCAL_GATEWAY, stac_endpoint=STAC_ENDPOINT)
+        self.client = Web3(local_gateway=LOCAL_GATEWAY, stac_endpoint=STAC_ENDPOINT, gateway_port=GATEWAY_PORT)
 
     def test_init(self):
         self.assertEqual(self.client.local_gateway, LOCAL_GATEWAY)

diff --git a/tests/test_client_integration.py b/tests/test_client_integration.py
@@ -12,10 +12,11 @@
 LOCAL_GATEWAY = "127.0.0.1"
 API_PORT = 5001
 STAC_ENDPOINT = "http://ec2-54-172-212-55.compute-1.amazonaws.com/api/v1/pgstac/"
+GATEWAY_PORT = 8081
 
 class TestIntegration(SetUp):
     def setUp(self):
-        self.client = Web3(local_gateway=LOCAL_GATEWAY, stac_endpoint=STAC_ENDPOINT)
+        self.client = Web3(local_gateway=LOCAL_GATEWAY, stac_endpoint=STAC_ENDPOINT, gateway_port=GATEWAY_PORT)
         # Colosseum, Rome, Italy
         self.bbox = [12.490827, 41.889249, 12.494162, 41.891876]