Fixed tests

ecthros · Nov 22, 2016 · 51903e5 · 51903e5
1 parent 5ce691a
commit 51903e5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests.py b/tests.py
@@ -1,4 +1,11 @@
 import os
+import pexpect
+from colorama import *
+import re
+
+prompt = Fore.BLUE + ">> " + Fore.RESET
 
 def base_test():
-    os.system("python cli.py")
+    cli = pexpect.spawn("sudo python cli.py")
+    cli.expect(re.escape(prompt))
+