import commands import os from behave import given, when, then from common import ( click_button, wait_until_button_is_visible, ) from selenium.common.exceptions import TimeoutException # For checking IP import requests resolv = None def apply_dns_workaround(): # we need this until we can get proper iptables execution # because of the dns rewrite hacks in the firewall print("CURRENT UID %s" % os.getuid()) if os.getuid() == 0: global resolv resolv = commands.getoutput('cat /etc/resolv.conf') print("original resolv.conf: %s" % resolv) commands.getoutput('echo nameserver 10.42.0.1 > /etc/ressolv.conf') resolv2 = commands.getoutput('cat /etc/resolv.conf') print("changed resolv.conf: %s" % resolv2) def deapply_dns_workaround(): if os.getuid() == 0: commands.getoutput('echo nameserver 85.214.20.141 > /etc/resolv.conf') @given('An initial network configuration') def record_ip(context): context.initial_ip = _current_ip() def _current_ip(): url = 'https://ipapi.co/json' r = requests.get(url) try: data = r.json() except Exception: print("ERROR: data received was %s" % r.content) raise current_ip = data.get('ip') print("Current IP: %s\n\n" % current_ip) return current_ip @when('I activate VPN') def activate_vpn(context): try: click_button(context, 'Install Helper Files') except TimeoutException: pass click_button(context, 'Turn ON') @then('I should have my ass covered') def assert_vpn(context): wait_until_button_is_visible(context, 'Turn OFF') apply_dns_workaround() assert _current_ip() != context.initial_ip @when('I deactivate VPN') def deactivate_vpn(context): click_button(context, 'Turn OFF') wait_until_button_is_visible(context, 'Turn ON', timeout=60) @then('My network should be configured as before') def network_as_before(context): deapply_dns_workaround() ip = _current_ip() assert ip == context.initial_ip