def test_run(self):
"""
Test server runner.
:return:
"""
with patch("salt.utils.ssdp.SSDPFactory", MagicMock()):
config = {
"answer": {
"master": "10.10.10.10"
},
ssdp.SSDPBase.LISTEN_IP: "10.10.10.10",
ssdp.SSDPBase.PORT: 12345,
}
srv = self.get_ssdp_discovery_server(**config)
srv.create_datagram_endpoint = MagicMock()
srv.log = MagicMock()
trnsp = MagicMock()
proto = MagicMock()
loop = MagicMock()
loop.run_until_complete = MagicMock(return_value=(trnsp, proto))
io = MagicMock()
io.ported = False
io.get_event_loop = MagicMock(return_value=loop)
with patch("salt.utils.ssdp.asyncio", io):
srv.run()
cde_args = io.get_event_loop(
).create_datagram_endpoint.call_args[1]
cfg_ip_addr, cfg_port = cde_args["local_addr"]
assert io.get_event_loop.called
assert io.get_event_loop().run_until_complete.called
assert io.get_event_loop().create_datagram_endpoint.called
assert io.get_event_loop().run_forever.called
assert trnsp.close.called
assert loop.close.called
assert srv.log.info.called
assert (srv.log.info.call_args[0][0] ==
"Stopping service discovery listener.")
assert "allow_broadcast" in cde_args
assert cde_args["allow_broadcast"]
assert "local_addr" in cde_args
assert (not cfg_ip_addr
== ssdp.SSDPBase.DEFAULTS[ssdp.SSDPBase.LISTEN_IP]
and cfg_ip_addr == "10.10.10.10")
assert (
not cfg_port == ssdp.SSDPBase.DEFAULTS[ssdp.SSDPBase.PORT]
and cfg_port == 12345)
def test_run(self):
'''
Test server runner.
:return:
'''
with patch('salt.utils.ssdp.SSDPFactory', MagicMock()):
config = {
'answer': {
'master': '10.10.10.10'
},
ssdp.SSDPBase.LISTEN_IP: '10.10.10.10',
ssdp.SSDPBase.PORT: 12345
}
srv = self.get_ssdp_discovery_server(**config)
srv.create_datagram_endpoint = MagicMock()
srv.log = MagicMock()
trnsp = MagicMock()
proto = MagicMock()
loop = MagicMock()
loop.run_until_complete = MagicMock(return_value=(trnsp, proto))
io = MagicMock()
io.ported = False
io.get_event_loop = MagicMock(return_value=loop)
with patch('salt.utils.ssdp.asyncio', io):
srv.run()
cde_args = io.get_event_loop(
).create_datagram_endpoint.call_args[1]
cfg_ip_addr, cfg_port = cde_args['local_addr']
assert io.get_event_loop.called
assert io.get_event_loop().run_until_complete.called
assert io.get_event_loop().create_datagram_endpoint.called
assert io.get_event_loop().run_forever.called
assert trnsp.close.called
assert loop.close.called
assert srv.log.info.called
assert srv.log.info.call_args[0][
0] == 'Stopping service discovery listener.'
assert 'allow_broadcast' in cde_args
assert cde_args['allow_broadcast']
assert 'local_addr' in cde_args
assert not cfg_ip_addr == ssdp.SSDPBase.DEFAULTS[
ssdp.SSDPBase.LISTEN_IP] and cfg_ip_addr == '10.10.10.10'
assert not cfg_port == ssdp.SSDPBase.DEFAULTS[
ssdp.SSDPBase.PORT] and cfg_port == 12345
