def test_get_all_proxies_osx_fails(self):
proxy.__grains__['os'] = 'Darwin'
mock = MagicMock()
with patch.dict(proxy.__salt__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
assert not mock.called
self.assertEqual(out, None)
def test_get_all_proxies_windows(self):
'''
Test to make sure that we correctly get the current proxy info on
Windows
'''
results = [{'vdata': 'http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128'},
{'vdata': 1}]
mock = MagicMock(side_effect=results)
expected = {'enabled': True,
'http': {'server': '192.168.0.1',
'port': '3128'},
'https': {'server': '192.168.0.2',
'port': '3128'},
'ftp': {'server': '192.168.0.3',
'port': '3128'}}
calls = [
call(hive='HKEY_CURRENT_USER',
key='SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyServer'),
call(hive='HKEY_CURRENT_USER',
key='SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyEnable')]
with patch.dict(proxy.__grains__, {'os': 'Windows'}):
with patch.dict(proxy.__salt__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
mock.assert_has_calls(calls)
self.assertEqual(expected, out)
def test_get_all_proxies_windows(self):
'''
Test to make sure that we correctly get the current proxy info
on Windows
'''
proxy.__grains__['os'] = 'Windows'
result = {
'vdata':
'http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128'
}
mock = MagicMock(return_value=result)
expected = {
'http': {
'server': '192.168.0.1',
'port': '3128'
},
'https': {
'server': '192.168.0.2',
'port': '3128'
},
'ftp': {
'server': '192.168.0.3',
'port': '3128'
}
}
with patch.dict(proxy.__salt__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
mock.assert_called_once_with(
'HKEY_CURRENT_USER',
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
'ProxyServer')
self.assertEqual(expected, out)
def test_get_all_proxies_windows():
"""
Test to make sure that we correctly get the current proxy info on
Windows
"""
results = [
{
"vdata":
("http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128"
)
},
{
"vdata": 1
},
]
mock = MagicMock(side_effect=results)
expected = {
"enabled": True,
"http": {
"server": "192.168.0.1",
"port": "3128"
},
"https": {
"server": "192.168.0.2",
"port": "3128"
},
"ftp": {
"server": "192.168.0.3",
"port": "3128"
},
}
calls = [
call(
hive="HKEY_CURRENT_USER",
key=
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
vname="ProxyServer",
),
call(
hive="HKEY_CURRENT_USER",
key=
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
vname="ProxyEnable",
),
]
with patch.dict(proxy.__grains__,
{"os": "Windows"}), patch.dict(proxy.__utils__,
{"reg.read_value": mock}):
out = proxy.get_proxy_win()
mock.assert_has_calls(calls)
assert expected == out
def test_get_all_proxies_windows(self):
"""
Test to make sure that we correctly get the current proxy info on
Windows
"""
results = [
{
"vdata":
"http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128"
},
{
"vdata": 1
},
]
mock = MagicMock(side_effect=results)
expected = {
"enabled": True,
"http": {
"server": "192.168.0.1",
"port": "3128"
},
"https": {
"server": "192.168.0.2",
"port": "3128"
},
"ftp": {
"server": "192.168.0.3",
"port": "3128"
},
}
calls = [
call(
hive='HKEY_CURRENT_USER',
key=
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyServer'),
call(
hive='HKEY_CURRENT_USER',
key=
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyEnable')
]
with patch.dict(proxy.__grains__, {'os': 'Windows'}):
with patch.dict(proxy.__utils__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
mock.assert_has_calls(calls)
self.assertEqual(expected, out)
def test_get_proxy_win(self):
"""
Test to make sure that we correctly get the current proxy info on
Windows
"""
result = [
{
"vdata":
"http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128"
},
{
"vdata": 1
},
]
mock_reg_read = MagicMock(side_effect=result)
expected = {
"enabled": True,
"http": {
"server": "192.168.0.1",
"port": "3128"
},
"https": {
"server": "192.168.0.2",
"port": "3128"
},
"ftp": {
"server": "192.168.0.3",
"port": "3128"
},
}
with patch.dict(proxy.__grains__, {"os": "Windows"}), patch.dict(
proxy.__utils__, {"reg.read_value": mock_reg_read}):
out = proxy.get_proxy_win()
self.assertDictEqual(out, expected)
mock_reg_read.assert_any_call(
hive="HKEY_CURRENT_USER",
key=
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
vname="ProxyServer",
)
mock_reg_read.assert_any_call(
hive="HKEY_CURRENT_USER",
key=
"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
vname="ProxyEnable",
)
def test_get_all_proxies_windows(self):
'''
Test to make sure that we correctly get the current proxy info
on Windows
'''
proxy.__grains__['os'] = 'Windows'
results = [
{
'vdata': 'http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128'
},
{
'vdata': 1
}
]
mock = MagicMock(side_effect=results)
expected = {
'enabled': True,
'http': {
'server': '192.168.0.1',
'port': '3128'
},
'https': {
'server': '192.168.0.2',
'port': '3128'
},
'ftp': {
'server': '192.168.0.3',
'port': '3128'
}
}
with patch.dict(proxy.__salt__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
calls = [
call('HKEY_CURRENT_USER',
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
'ProxyServer'),
call('HKEY_CURRENT_USER',
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
'ProxyEnable'),
]
mock.assert_has_calls(calls)
self.assertEqual(expected, out)
def test_get_proxy_win(self):
'''
Test to make sure that we correctly get the current proxy info on
Windows
'''
result = [{
'vdata':
'http=192.168.0.1:3128;https=192.168.0.2:3128;ftp=192.168.0.3:3128'
}, {
'vdata': 1
}]
mock_reg_read = MagicMock(side_effect=result)
expected = {
'enabled': True,
'http': {
'server': '192.168.0.1',
'port': '3128'
},
'https': {
'server': '192.168.0.2',
'port': '3128'
},
'ftp': {
'server': '192.168.0.3',
'port': '3128'
}
}
with patch.dict(proxy.__grains__, {'os': 'Windows'}), \
patch.dict(proxy.__utils__, {'reg.read_value': mock_reg_read}):
out = proxy.get_proxy_win()
self.assertDictEqual(out, expected)
mock_reg_read.assert_any_call(
hive='HKEY_CURRENT_USER',
key=
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyServer')
mock_reg_read.assert_any_call(
hive='HKEY_CURRENT_USER',
key=
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings',
vname='ProxyEnable')
def test_get_all_proxies_macos_fails(self):
mock = MagicMock()
with patch.dict(proxy.__salt__, {'reg.read_value': mock}):
out = proxy.get_proxy_win()
assert not mock.called
self.assertEqual(out, None)
def test_get_all_proxies_macos_fails():
mock = MagicMock()
with patch.dict(proxy.__utils__, {"reg.read_value": mock}):
out = proxy.get_proxy_win()
assert not mock.called
assert out is None