Outline imports

vyperlang · Feb 20, 2024 · 72556bb · 72556bb
1 parent 50ee8e9
commit 72556bb
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/integration/jupyter/conftest.py b/tests/integration/jupyter/conftest.py
@@ -1,8 +1,9 @@
 from multiprocessing.shared_memory import SharedMemory
-from os import urandom
 
 import pytest
 
+from boa.integrations.jupyter.browser import _generate_token
+
 
 @pytest.fixture()
 def shared_memory_length():
@@ -13,9 +14,7 @@ def shared_memory_length():
 
 @pytest.fixture()
 def token():
-    from boa.integrations.jupyter.constants import CALLBACK_TOKEN_BYTES, PLUGIN_NAME
-
-    return f"{PLUGIN_NAME}_{urandom(CALLBACK_TOKEN_BYTES).hex()}"
+    return _generate_token()
 
 
 @pytest.fixture()