http://b/277898785 fix more flaky tests

Cesia19 · Apr 13, 2023 · 698e1ce · 698e1ce
1 parent 7d9e8c2
commit 698e1ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -39,7 +39,7 @@ def do_POST(s):
 
 
 class TestDatasets(unittest.TestCase):
-    SERVER_ADDRESS = urlparse(os.getenv(_KAGGLE_URL_BASE_ENV_VAR_NAME, default="http://127.0.0.1:8001"))
+    SERVER_ADDRESS = urlparse(os.getenv(_KAGGLE_URL_BASE_ENV_VAR_NAME, default="http://127.0.0.1:0"))
 
     def _test_client(self, client_func, expected_path, expected_body, is_tpu=True, success=True, iap_token=False):
         _request = {}

diff --git a/tests/test_user_session.py b/tests/test_user_session.py
@@ -30,7 +30,7 @@ def do_POST(s):
         s.wfile.write(json.dumps(s.get_response()).encode("utf-8"))
 
 class TestUserSessionClient(unittest.TestCase):
-    SERVER_ADDRESS = urlparse(os.getenv(_KAGGLE_URL_BASE_ENV_VAR_NAME, default="http://127.0.0.1:8001"))
+    SERVER_ADDRESS = urlparse(os.getenv(_KAGGLE_URL_BASE_ENV_VAR_NAME, default="http://127.0.0.1:0"))
     TEST_JWT = 'test-secrets-key'
 
     def _test_client(self, client_func, expected_path, expected_body, source=None, success=True):