Minor fixes

Karakatiza666 · Karakatiza666 · commit f8dd308292e4 · 2025-09-20T22:04:51.000Z
Signed-off-by: Karakatiza666 &lt;bulakh.96@gmail.com&gt;
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -23,7 +23,8 @@
 				"ms-playwright.playwright",
 				"svelte.svelte-vscode",
 				"bradlc.vscode-tailwindcss",
-				"esbenp.prettier-vscode"
+				"esbenp.prettier-vscode",
+				"ms-python.python"
 			],
 			"settings": {
 				"editor.formatOnSave": true,
diff --git a/python/feldera/testutils.py b/python/feldera/testutils.py
@@ -21,7 +21,7 @@
 def _get_oidc_token():
     """Get OIDC token if environment is configured, otherwise return None"""
     try:
-        from tests.platform.oidc_test_helper import get_oidc_test_helper
+        from feldera.testutils_oidc import get_oidc_test_helper
 
         oidc_helper = get_oidc_test_helper()
         if oidc_helper is not None:
diff --git a/python/feldera/testutils_oidc.py b/python/feldera/testutils_oidc.py
diff --git a/python/tests/platform/TEST_AUTH.md b/python/tests/platform/TEST_AUTH.md
@@ -158,5 +158,5 @@ OIDC_TEST_PASSWORD=test-password
 
 - `helper.py` - Main test utilities with unified authentication support
 - `test_oidc_auth_simple.py` - OIDC-specific authentication tests
-- `oidc_test_helper.py` - OIDC token management utilities
+- `testutils_oidc.py` - OIDC token management utilities
 - All other `test_*.py` files - Platform tests that use unified authentication
diff --git a/python/tests/platform/conftest.py b/python/tests/platform/conftest.py
@@ -19,7 +19,7 @@ def is_master(config) -> bool:
 
 def _fetch_oidc_token():
     """Fetch OIDC token using Resource Owner Password Grant flow."""
-    from .oidc_test_helper import get_oidc_test_helper
+    from feldera.testutils_oidc import get_oidc_test_helper
 
     oidc_helper = get_oidc_test_helper()
     if oidc_helper is None:
@@ -118,13 +118,13 @@ def oidc_token_fixture(request):
             token_json = base64.b64decode(env_token.encode()).decode()
             token_data = json.loads(token_json)
 
+            access_token = token_data.get("access_token")
             if (
-                token_data.get("access_token")
-                and time.time() < token_data.get("expires_at", 0) - 30
+                access_token is None
+                or time.time() >= token_data.get("expires_at", 0) - 30
             ):
-                return token_data["access_token"]
-            else:
                 raise RuntimeError("OIDC token expired before test execution")
+            return access_token
         except Exception as e:
             raise RuntimeError(f"Failed to parse OIDC token from environment: {e}")
 
diff --git a/python/tests/platform/helper.py b/python/tests/platform/helper.py
@@ -23,7 +23,7 @@
 from urllib.parse import quote, quote_plus
 
 from tests import FELDERA_TLS_INSECURE, API_KEY, BASE_URL, unique_pipeline_name
-from .oidc_test_helper import get_oidc_test_helper
+from feldera.testutils_oidc import get_oidc_test_helper
 
 API_PREFIX = "/v0"
 
