From 06332996fc05d477ae68be2f81b94d9ea1d21f34 Mon Sep 17 00:00:00 2001
From: Aurelien Vaillant <a.vaillant.moderlab@gmail.com>
Date: Tue, 29 Mar 2022 17:47:46 +0200
Subject: [PATCH] Update token call

---
 tests/utils/forge.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/utils/forge.py b/tests/utils/forge.py
index d364ef1..db29572 100644
--- a/tests/utils/forge.py
+++ b/tests/utils/forge.py
@@ -7,8 +7,12 @@ from github import Github
 
 def read_token():
     token_file = Path(os.getcwd(), 'tests', 'token.txt')
-    with open(token_file, 'r') as f:
-        token = f.read()
+    if token_file.exists():
+        with open(token_file, 'r') as f:
+            token = f.read()
+
+    else:
+        token = os.environ.get('token')
 
     return token