summaryrefslogtreecommitdiff
path: root/tests/test_vpn_management.py
blob: d8a0314b9eafc5cda88c9a845f7c12ba6b611848 (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
import unittest
import sys
import time

from eip_client.mocks.manager import get_openvpn_manager_mocks


class VPNManagerTests(unittest.TestCase):

    def setUp(self):
        self.manager = get_openvpn_manager_mocks()

    #
    # tests
    #

    def test_status_command(self):
        ret = self.manager.status()
        #print ret

    def test_connection_state(self):
        ts, status, ok, ip, remote = self.manager.get_connection_state()
        self.assertTrue(status in ('CONNECTED', 'DISCONNECTED'))
        self.assertTrue(isinstance(ts, time.struct_time))

    def test_status_io(self):
        when_ts, counters  = self.manager.get_status_io()
        self.assertTrue(isinstance(when_ts, time.struct_time))
        self.assertEqual(len(counters), 5)
        self.assertTrue(all(map(lambda x: x.isdigit(), counters)))


def test():
    suite = unittest.TestSuite()
    for cls in (VPNManagerTests,):
        suite.addTest(unittest.makeSuite(cls))
    runner = unittest.TextTestRunner(sys.stdout, verbosity=2,
                                         failfast=False)
    result = runner.run(suite)

if __name__ == "__main__":
    test()