1 files changed, 26 insertions, 0 deletions
diff --git a/tests/functional/features/steps/bitmask.py b/tests/functional/features/steps/bitmask.py
new file mode 100644
index 0000000..b5b4cf1
--- /dev/null
+++ b/tests/functional/features/steps/bitmask.py
@@ -0,0 +1,26 @@
+import commands
+import shutil
+import os
+import time
+from leap.common.config import get_path_prefix
+
+from behave import given
+
+
+@given('I start bitmask for the first time')
+def initial_run(context):
+    commands.getoutput('bitmaskctl stop')
+    # TODO: fix bitmaskctl to only exit once bitmaskd has stopped
+    time.sleep(2)
+    _initialize_home_path()
+    commands.getoutput('bitmaskctl start')
+    tokenpath = os.path.join(get_path_prefix(), 'leap', 'authtoken')
+    token = open(tokenpath).read().strip()
+    context.login_url = "http://localhost:7070/#%s" % token
+
+
+def _initialize_home_path():
+    home_path = '/tmp/bitmask-test'
+    shutil.rmtree(home_path, ignore_errors=True)
+    os.environ['HOME'] = home_path
+    os.makedirs(get_path_prefix())