1 files changed, 18 insertions, 0 deletions
diff --git a/tests/functional/features/steps/vpn.py b/tests/functional/features/steps/vpn.py
new file mode 100644
index 0000000..f08c2af
--- /dev/null
+++ b/tests/functional/features/steps/vpn.py
@@ -0,0 +1,18 @@
+from behave import when, then
+from common import (
+    click_button,
+    wait_until_button_is_visible,
+    find_element_containing_text
+)
+
+
+@when('I activate VPN')
+def activate_vpn(context):
+    click_button(context, 'Install Helper Files')
+    click_button(context, 'Turn ON')
+
+
+@then('I should have my ass covered')
+def assert_vpn(context):
+    wait_until_button_is_visible(context, 'Turn OFF')
+    assert find_element_containing_text(context, 'Turn OFF', 'button')