mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 09:23:56 +00:00
Merge pull request #22594 from saltstack/revert-22583-win_network-unit-test
Revert "adding win_network unit test case."
This commit is contained in:
Nicole Thomas
commit
a0105d9f26
@ -1,285 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
'''
|
||||
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
||||
'''
|
||||
|
||||
# Import Python Libs
|
||||
from __future__ import absolute_import
|
||||
|
||||
# Import Salt Testing Libs
|
||||
from salttesting import TestCase, skipIf
|
||||
from salttesting.mock import (
|
||||
MagicMock,
|
||||
patch,
|
||||
NO_MOCK,
|
||||
NO_MOCK_REASON
|
||||
)
|
||||
|
||||
from salttesting.helpers import ensure_in_syspath
|
||||
|
||||
ensure_in_syspath('../../')
|
||||
|
||||
# Import Salt Libs
|
||||
from salt.modules import win_network
|
||||
|
||||
win_network.__salt__ = {}
|
||||
|
||||
|
||||
class Mockwmi(object):
|
||||
'''
|
||||
Mock wmi class
|
||||
'''
|
||||
NetConnectionID = 'Ethernet'
|
||||
|
||||
def __init__(self):
|
||||
self.netenabled = None
|
||||
|
||||
@staticmethod
|
||||
def WMI():
|
||||
'''
|
||||
Mock WMI method
|
||||
'''
|
||||
return Mockwmi()
|
||||
|
||||
def Win32_NetworkAdapter(self, NetEnabled=True):
|
||||
'''
|
||||
Mock Win32_NetworkAdapter method
|
||||
'''
|
||||
self.netenabled = NetEnabled
|
||||
return [Mockwmi()]
|
||||
|
||||
|
||||
class Mockwinapi(object):
|
||||
'''
|
||||
Mock winapi class
|
||||
'''
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
class network(object):
|
||||
'''
|
||||
Mock network class
|
||||
'''
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@staticmethod
|
||||
def sanitize_host(host):
|
||||
'''
|
||||
Mock sanitize_host method
|
||||
'''
|
||||
return host
|
||||
|
||||
@staticmethod
|
||||
def interfaces():
|
||||
'''
|
||||
Mock interfaces method
|
||||
'''
|
||||
return True
|
||||
|
||||
@staticmethod
|
||||
def hw_addr(iface):
|
||||
'''
|
||||
Mock hw_addr method
|
||||
'''
|
||||
return iface
|
||||
|
||||
@staticmethod
|
||||
def subnets():
|
||||
'''
|
||||
Mock subnets method
|
||||
'''
|
||||
return '10.1.1.0/24'
|
||||
|
||||
@staticmethod
|
||||
def in_subnet(cidr):
|
||||
'''
|
||||
Mock in_subnet method
|
||||
'''
|
||||
return cidr
|
||||
|
||||
@staticmethod
|
||||
def ip_addrs(interface=None, include_loopback=False):
|
||||
'''
|
||||
Mock ip_addrs method
|
||||
'''
|
||||
return (interface, include_loopback)
|
||||
|
||||
@staticmethod
|
||||
def ip_addrs6(interface=None, include_loopback=False):
|
||||
'''
|
||||
Mock ip_addrs6 method
|
||||
'''
|
||||
return (interface, include_loopback)
|
||||
|
||||
class winapi(object):
|
||||
'''
|
||||
Mock winapi class
|
||||
'''
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@staticmethod
|
||||
def Com():
|
||||
'''
|
||||
Mock Com method
|
||||
'''
|
||||
pass
|
||||
|
||||
win_network.salt.utils = Mockwinapi
|
||||
win_network.wmi = Mockwmi
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class WinNetworkTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.modules.win_network
|
||||
'''
|
||||
# 'ping' function tests: 1
|
||||
|
||||
def test_ping(self):
|
||||
'''
|
||||
Test if it performs a ping to a host.
|
||||
'''
|
||||
mock = MagicMock(return_value=True)
|
||||
with patch.dict(win_network.__salt__, {'cmd.run': mock}):
|
||||
self.assertTrue(win_network.ping('127.0.0.1'))
|
||||
|
||||
# 'netstat' function tests: 1
|
||||
|
||||
def test_netstat(self):
|
||||
'''
|
||||
Test if it return information on open ports and states
|
||||
'''
|
||||
ret = (' Proto Local Address Foreign Address State PID\n'
|
||||
' TCP 127.0.0.1:1434 0.0.0.0:0 LISTENING 1728\n'
|
||||
' UDP 127.0.0.1:1900 *:* 4240')
|
||||
mock = MagicMock(return_value=ret)
|
||||
with patch.dict(win_network.__salt__, {'cmd.run': mock}):
|
||||
self.assertListEqual(win_network.netstat(),
|
||||
[{'local-address': '127.0.0.1:1434',
|
||||
'program': '1728', 'proto': 'TCP',
|
||||
'remote-address': '0.0.0.0:0',
|
||||
'state': 'LISTENING'},
|
||||
{'local-address': '127.0.0.1:1900',
|
||||
'program': '4240', 'proto': 'UDP',
|
||||
'remote-address': '*:*', 'state': None}])
|
||||
|
||||
# 'traceroute' function tests: 1
|
||||
|
||||
def test_traceroute(self):
|
||||
'''
|
||||
Test if it performs a traceroute to a 3rd party host
|
||||
'''
|
||||
ret = (' 1 1 ms <1 ms <1 ms 172.27.104.1\n'
|
||||
' 2 1 ms <1 ms 1 ms 121.242.35.1.static-chennai.vsnl.net.in [121.242.35.1]\n'
|
||||
' 3 3 ms 2 ms 2 ms 121.242.4.53.static-pune.vsnl.net.in [121.242.4.53]\n')
|
||||
mock = MagicMock(return_value=ret)
|
||||
with patch.dict(win_network.__salt__, {'cmd.run': mock}):
|
||||
self.assertListEqual(win_network.traceroute('google.com'),
|
||||
[{'count': '1', 'hostname': None,
|
||||
'ip': '172.27.104.1', 'ms1': '1',
|
||||
'ms2': '<1', 'ms3': '<1'},
|
||||
{'count': '2',
|
||||
'hostname': '121.242.35.1.static-chennai.vsnl.net.in',
|
||||
'ip': '[121.242.35.1]', 'ms1': '1',
|
||||
'ms2': '<1', 'ms3': '1'},
|
||||
{'count': '3',
|
||||
'hostname': '121.242.4.53.static-pune.vsnl.net.in',
|
||||
'ip': '[121.242.4.53]', 'ms1': '3',
|
||||
'ms2': '2', 'ms3': '2'}])
|
||||
|
||||
# 'nslookup' function tests: 1
|
||||
|
||||
def test_nslookup(self):
|
||||
'''
|
||||
Test if it query DNS for information about a domain or ip address
|
||||
'''
|
||||
ret = ('Server: ct-dc-3-2.cybage.com\n'
|
||||
'Address: 172.27.172.12\n'
|
||||
'Non-authoritative answer:\n'
|
||||
'Name: google.com\n'
|
||||
'Addresses: 2404:6800:4007:806::200e\n'
|
||||
'216.58.196.110\n')
|
||||
mock = MagicMock(return_value=ret)
|
||||
with patch.dict(win_network.__salt__, {'cmd.run': mock}):
|
||||
self.assertListEqual(win_network.nslookup('google.com'),
|
||||
[{'Server': 'ct-dc-3-2.cybage.com'},
|
||||
{'Address': '172.27.172.12'},
|
||||
{'Name': 'google.com'},
|
||||
{'Addresses': ['2404:6800:4007:806::200e',
|
||||
'216.58.196.110']}])
|
||||
|
||||
# 'dig' function tests: 1
|
||||
|
||||
def test_dig(self):
|
||||
'''
|
||||
Test if it performs a DNS lookup with dig
|
||||
'''
|
||||
mock = MagicMock(return_value=True)
|
||||
with patch.dict(win_network.__salt__, {'cmd.run': mock}):
|
||||
self.assertTrue(win_network.dig('google.com'))
|
||||
|
||||
# 'interfaces_names' function tests: 1
|
||||
|
||||
def test_interfaces_names(self):
|
||||
'''
|
||||
Test if it return a list of all the interfaces names
|
||||
'''
|
||||
with patch('salt.utils.winapi.Com', MagicMock()):
|
||||
self.assertListEqual(win_network.interfaces_names(), ['Ethernet'])
|
||||
|
||||
# 'interfaces' function tests: 1
|
||||
|
||||
def test_interfaces(self):
|
||||
'''
|
||||
Test if it return information about all the interfaces on the minion
|
||||
'''
|
||||
self.assertTrue(win_network.interfaces())
|
||||
|
||||
# 'hw_addr' function tests: 1
|
||||
|
||||
def test_hw_addr(self):
|
||||
'''
|
||||
Test if it return the hardware address (a.k.a. MAC address)
|
||||
for a given interface
|
||||
'''
|
||||
self.assertEqual(win_network.hw_addr('Ethernet'), 'Ethernet')
|
||||
|
||||
# 'subnets' function tests: 1
|
||||
|
||||
def test_subnets(self):
|
||||
'''
|
||||
Test if it returns a list of subnets to which the host belongs
|
||||
'''
|
||||
self.assertEqual(win_network.subnets(), '10.1.1.0/24')
|
||||
|
||||
# 'in_subnet' function tests: 1
|
||||
|
||||
def test_in_subnet(self):
|
||||
'''
|
||||
Test if it returns True if host is within specified subnet,
|
||||
otherwise False
|
||||
'''
|
||||
self.assertTrue(win_network.in_subnet('10.1.1.0/16'))
|
||||
|
||||
# 'ip_addrs' function tests: 1
|
||||
|
||||
def test_ip_addrs(self):
|
||||
'''
|
||||
Test if it returns a list of IPv4 addresses assigned to the host.
|
||||
'''
|
||||
self.assertTrue(win_network.ip_addrs())
|
||||
|
||||
# 'ip_addrs6' function tests: 1
|
||||
|
||||
def test_ip_addrs6(self):
|
||||
'''
|
||||
Test if it returns a list of IPv6 addresses assigned to the host.
|
||||
'''
|
||||
self.assertTrue(win_network.ip_addrs6())
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(WinNetworkTestCase, needs_daemon=False)
|
||||
Loading…
Reference in New Issue
Block a user