Pymon/test.py at master · aenima-x/Pymon · GitHub

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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import unittest
import os
import pymon
from pymon import sender
from pymon.pymonExceptions import ClientMissingInfoError


class MockSender(sender.Sender):
    def __init__(self):
        super(MockSender, self).__init__()

    def send(self, client, debug=False):
        return client.machine


class PymonServerTests(unittest.TestCase):

    def test_server_ip_invalid(self):
        s = pymon.Server()
        with self.assertRaises(ValueError):
            s.address = '256.4.2.1'

    def test_server_hostname_invalid(self):
        s = pymon.Server()
        with self.assertRaises(ValueError):
            s.address = 'aaa.'

    def test_server_port_invalid(self):
        s = pymon.Server()
        with self.assertRaises(ValueError):
            s.port = 'aaa'

    def test_server(self):
        self.assertIsInstance(pymon.Server(address="192.168.1.1", port=1984), pymon.Server)

    def test_server_url(self):
        s = pymon.Server(address="192.168.1.1", port=1984)
        self.assertEqual(s.get_URL(), "192.168.1.1:1984")


class PymonClientTests(unittest.TestCase):

    def setUp(self):
        self.servers = pymon.Server(address="192.168.1.1", port=1984)

    def test_client_missing_info(self):
        with self.assertRaises(ClientMissingInfoError):
            c = pymon.Client("column")

    def test_native_client_ok(self):
        os.environ['XYMSRV'] = '192.168.1.1'
        os.environ['XYMSERVERS'] = '192.168.1.1 192.168.1.2'
        os.environ['XYMONDPORT'] = '1984'
        os.environ['XYMONCLIENTLOGS'] = '/tmp/'
        os.environ['XYMONTMP'] = '/tmp/'
        os.environ['XYMON'] = '/bin/ls'
        os.environ['MACHINE'] = 'kenny'
        os.environ['SERVEROSTYPE'] = 'Darwin'
        os.environ['XYMONCLIENTHOME'] = '/home/xymon'
        client = pymon.Client("column")
        self.assertIsInstance(client, pymon.Client)
        #client.sender = MockSender(client)
        #self.assertEqual(client.sender.send(), "kenny")

    def test_native_client_one_server(self):
        os.environ['XYMSRV'] = '127.0.0.1'
        os.environ['XYMSERVERS'] = ''
        os.environ['XYMONDPORT'] = '1984'
        os.environ['XYMONCLIENTLOGS'] = '/tmp/'
        os.environ['XYMONTMP'] = '/tmp/'
        os.environ['XYMON'] = '/bin/ls'
        os.environ['MACHINE'] = 'kenny'
        os.environ['SERVEROSTYPE'] = 'Darwin'
        os.environ['XYMONCLIENTHOME'] = '/home/xymon'
        client = pymon.Client("column")
        self.assertIsInstance(client, pymon.Client)

    def test_native_client_two_servers(self):
        os.environ['XYMSRV'] = '0.0.0.0'
        os.environ['XYMSERVERS'] = '192.168.1.1 192.168.1.2'
        os.environ['XYMONDPORT'] = '1984'
        os.environ['XYMONCLIENTLOGS'] = '/tmp/'
        os.environ['XYMONTMP'] = '/tmp/'
        os.environ['XYMON'] = '/bin/ls'
        os.environ['MACHINE'] = 'kenny'
        os.environ['SERVEROSTYPE'] = 'Darwin'
        os.environ['XYMONCLIENTHOME'] = '/home/xymon'
        client = pymon.Client("column")
        self.assertIsInstance(client, pymon.Client)

    def test_native_client_one_server_hobbit(self):
        os.environ['BB'] = '127.0.0.1'
        os.environ['BBDISPLAYS'] = ''
        os.environ['BBPORT'] = '1984'
        os.environ['BBCLIENTLOGS'] = '/tmp/'
        os.environ['BBTMP'] = '/tmp/'
        os.environ['BB'] = '/bin/ls'
        os.environ['MACHINE'] = 'kenny'
        os.environ['SERVEROSTYPE'] = 'Darwin'
        os.environ['HOBBITCLIENTHOME'] = '/home/xymon'
        client = pymon.Client("column")
        self.assertIsInstance(client, pymon.Client)

def main():
    unittest.main()

if __name__ == '__main__':
    main()