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)
|