summaryrefslogtreecommitdiffstats
path: root/tests/test_discovery.py
blob: a211167494a9bfeb43944a5fd9ea1836b47bbad4 (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
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
#!/usr/bin/python
# Copyright 2014 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#

from plugins import discovery
from testrunner import PluginsTestCase as TestCaseBase


class TestDiscovery(TestCaseBase):
    def _mockExcecNRPECommand(self, host, command, arguments=None,
                              jsonOutput=True):
        if command == "discover_volume_list":
            return self._getVolumeNames()
        elif command == "discover_volume_info":
            return self._getVolumeInfo(arguments[0])
        elif command == "discoverpeers":
            return self._getPeers()
        elif command == "discoverhostparams":
            return self._getHostParams(host)

    def _getLogicalComponents(self):
        result = {}
        result['volumes'] = []
        result['volumes'].append(self._getVolumeInfo("V1")['V1'])
        result['volumes'].append(self._getVolumeInfo("V2")['V2'])
        return result

    def _getVolumeInfo(self, volumeName):
        result = {}
        if volumeName == "V1":
            volume = {"bricks": [{"brickpath": "/bricks/v1-1",
                                  "hostUuid": "0000-1111",
                                  "hostip": "172.16.53.1"}],
                      "type": "DISTRIBUTE", "name": "V1"}

        elif volumeName == "V2":
            volume = {"bricks": [{"brickpath": "/bricks/v2-1",
                                  "hostUuid": "0000-1112",
                                  "hostip": "172.16.53.2"}],
                      "type": "DISTRIBUTE", "name": "V2"}
        result[volume['name']] = volume
        return result

    def _getVolumeNames(self):
        result = {}
        result['V1'] = {"type": "DISTRIBUTE", "name": "V1"}
        result['V2'] = {"type": "DISTRIBUTE", "name": "V2"}
        return result

    def _getPeers(self):
        result = []
        result.append({"hostip": "lo", "uuid": "0000-1111"})
        result.append({"hostip": "172.16.53.2", "uuid": "0000-1112"})
        return result

    def _getHostParams(self, hostip):
        if hostip == "172.16.53.1":
            return {"hostname": "node-1"}
        elif hostip == "172.16.53.2":
            return {"hostname": "node-2"}

    def _verifyClusterData(self, clusterdata, clusterName, host):

        self.assertEqual(clusterdata['name'], clusterName)
        self.assertEqual(clusterdata['hosts'][0]['hostip'], host)
        for host in clusterdata['hosts']:
            hostDetails = self._getHostParams(host['hostip'])
            self.assertEqual(host['hostname'], hostDetails['hostname'])
            self.assertEqual(len(host['bricks']), 1)
        volumes = self._getLogicalComponents()['volumes']
        self.assertEqual(len(clusterdata['volumes']), len(volumes))

        for i in range(len(volumes)):
            self._verifyVolume(volumes[i], clusterdata['volumes'][i])

    def _verifyVolume(self, expected, actual):
        self.assertEqual(expected['name'], actual['name'])
        self.assertEqual(expected['type'], actual['type'])

    # Method to test the discoverCluster() method
    def testDiscoverCluster(self):
        discovery.excecNRPECommand = self._mockExcecNRPECommand
        clusterName = "test-cluster"
        host = "172.16.53.1"
        clusterdata = discovery.discoverCluster(host, clusterName)
        self._verifyClusterData(clusterdata, clusterName, host)