Improve unitary test for client/server by using venv if possible

1 files changed, 6 insertions, 1 deletions

diff --git a/unitest-restful.py b/unitest-restful.py
index df64cf99..362fce61 100755
--- a/unitest-restful.py
+++ b/unitest-restful.py
@@ -15,6 +15,7 @@ import subprocess
 import time
 import numbers
 import unittest
+import os
 
 from glances import __version__
 from glances.compat import text_type
@@ -54,7 +55,11 @@ class TestGlances(unittest.TestCase):
         global pid
 
         print('INFO: [TEST_000] Start the Glances Web Server')
-        cmdline = "python -m glances -B localhost -w -p %s" % SERVER_PORT
+        if os.path.isfile("./venv/bin/python"):
+            cmdline = "./venv/bin/python"
+        else:
+            cmdline = "python"
+        cmdline += " -m glances -B localhost -w -p %s" % SERVER_PORT
         print("Run the Glances Web Server on port %s" % SERVER_PORT)
         args = shlex.split(cmdline)
         pid = subprocess.Popen(args)