summaryrefslogtreecommitdiff
path: root/tests/e2e/check_ip
blob: 4b903251648dae0508f013931c8cc1d80751b138 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/python3

import urllib.request
import json
import sys

DEMO_ISPS = ('Riseup Networks',
             'Leaseweb Deutschland GmbH')
CI_LEAP_SE_ISPS = ('AS47172 Greenhost BV')

url = 'https://ipapi.co/json'

with urllib.request.urlopen(url) as _url:
    data = json.loads(_url.read().decode())

isp = data['org']
ip = data['ip']

print('ISP >> %s' % isp)
print('IP  >> %s' % ip)


class BadCmd(Exception):
    pass


try:
    cmd = sys.argv[1]

    if cmd == 'vpn_on':
        print('checking VPN ON...')
        assert isp in CI_LEAP_SE_ISPS
    elif cmd == 'vpn_off':
        print('checking VPN OFF...')
        assert isp not in CI_LEAP_SE_ISPS
    else:
        raise BadCmd()

except BadCmd:
    print("Cannot parse that command. Valid commands: vpn_on, vpn_off")
    sys.exit(1)

except Exception as exc:
    print("IP *NOT* as expected")
    print("%r" % exc)
    sys.exit(1)

else:
    print("OK :)")
    sys.exit(0)