summaryrefslogtreecommitdiff
path: root/tests/unit/vpn
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/vpn')
-rw-r--r--tests/unit/vpn/test_management.py14
1 files changed, 10 insertions, 4 deletions
diff --git a/tests/unit/vpn/test_management.py b/tests/unit/vpn/test_management.py
index 4066ab6..2bcb0b9 100644
--- a/tests/unit/vpn/test_management.py
+++ b/tests/unit/vpn/test_management.py
@@ -19,14 +19,20 @@
Tests for the VPN Management Interface
"""
+import os
import StringIO
from twisted.trial import unittest
from leap.bitmask.vpn.management import ManagementProtocol
-session1 = open('session1.data').readlines()
-session2 = open('session2.data').readlines()
+def get_data(name):
+ return os.path.join(
+ os.path.abspath(os.path.split(__file__)[0]), '%s.data' % name)
+
+
+session1 = open(get_data('session1')).readlines()
+session2 = open(get_data('session2')).readlines()
def feed_the_protocol(protocol, data):
@@ -92,7 +98,7 @@ class ManagementTestCase(unittest.TestCase):
def test_get_pid(self):
proto = ManagementProtocol()
proto.transport = StringIO.StringIO()
- assert proto.pid == None
+ assert proto.pid is None
proto.getPid()
pid_lines = ['SUCCESS: pid=99999']
feed_the_protocol(proto, pid_lines)
@@ -126,4 +132,4 @@ class ManagementTestCase(unittest.TestCase):
assert info['state_simple'] == 'ON'
assert info['state_legend'] == 'Initialization Sequence Completed'
assert info['openvpn_version'].startswith('OpenVPN 2.4.0')
- assert info['pid'] == 23783
+ assert info['pid'] == 23783