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
|
#!/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 network_utils
from testrunner import PluginsTestCase as TestCaseBase
class TestNetworkUtils(TestCaseBase):
def mockGetFlags(self, device):
if device == 'lo':
return 73
else:
return 4163
def mockGetDevices(self):
return ['lo', 'etho', 'eth1']
def mockGetIpAddr(self, device):
if device == 'lo':
return '127.0.0.1'
else:
return '10.70.42.1'
def mockGethostbyname_ex(self, host):
if host == 'test.host.com':
return host, [], ['10.70.42.1']
else:
return host, [], ['82.94.164.162']
def setUpMocks(self):
network_utils.ethtool.get_active_devices = self.mockGetDevices
network_utils.ethtool.get_flags = self.mockGetFlags
network_utils.ethtool.get_ipaddr = self.mockGetIpAddr
network_utils.socket.gethostbyname_ex = self.mockGethostbyname_ex
#Methods to test validateHostAddress(address)
def testValidateHostAddressWithEmptyAddress(self):
self.setUpMocks()
validationMsg = network_utils.validateHostAddress(None)
self.assertEqual('Please specify host Address', validationMsg)
def testValidateHostAddressWithLoopBackAddress(self):
self.setUpMocks()
validationMsg = network_utils.validateHostAddress("127.0.0.1")
self.assertEqual("Address '127.0.0.1' can't be mapped to non loopback "
"devices on this host", validationMsg)
def testValidateHostAddressWithValidAddress(self):
self.setUpMocks()
validationMsg = network_utils.validateHostAddress("10.70.42.1")
self.assertEqual(None, validationMsg)
def testValidateHostAddressWithInvalidFQDN(self):
self.setUpMocks()
validationMsg = network_utils.validateHostAddress("this-is-"
"invalid-fqdn")
self.assertEqual("Host FQDN name 'this-is-invalid-fqdn' has no domain "
"suffix", validationMsg)
validationMsg = network_utils.validateHostAddress("this.is."
"invalid.fqdn.")
self.assertEqual("Host FQDN name 'this.is.invalid.fqdn.' has invalid "
"domain name", validationMsg)
validationMsg = network_utils.validateHostAddress("test.host."
"com.notfound")
self.assertEqual("The following addreses: 'set(['82.94.164.162'])' "
"can't be mapped to non loopback devices on this "
"host", validationMsg)
def testValidateHostAddressWithValidFQDN(self):
self.setUpMocks()
validationMsg = network_utils.validateHostAddress("test.host.com")
self.assertEqual(None, validationMsg)
|