9debbfb1a8
* Add SSDP integration * Fix tests * Sort all the things * Add netdisco to test requirements
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
"""Test the SSDP integration."""
|
|
from unittest.mock import patch, Mock
|
|
|
|
from homeassistant.generated import ssdp as gn_ssdp
|
|
from homeassistant.components import ssdp
|
|
|
|
from tests.common import mock_coro
|
|
|
|
|
|
async def test_scan_match_st(hass):
|
|
"""Test matching based on ST."""
|
|
scanner = ssdp.Scanner(hass)
|
|
|
|
with patch('netdisco.ssdp.scan', return_value=[
|
|
Mock(st="mock-st", location=None)
|
|
]), patch.dict(
|
|
gn_ssdp.SSDP['st'], {'mock-st': ['mock-domain']}
|
|
), patch.object(
|
|
hass.config_entries.flow, 'async_init',
|
|
return_value=mock_coro()
|
|
) as mock_init:
|
|
await scanner.async_scan(None)
|
|
|
|
assert len(mock_init.mock_calls) == 1
|
|
assert mock_init.mock_calls[0][1][0] == 'mock-domain'
|
|
assert mock_init.mock_calls[0][2]['context'] == {'source': 'ssdp'}
|
|
|
|
|
|
async def test_scan_match_manufacturer(hass, aioclient_mock):
|
|
"""Test matching based on ST."""
|
|
aioclient_mock.get('http://1.1.1.1', text="""
|
|
<root>
|
|
<device>
|
|
<manufacturer>Paulus</manufacturer>
|
|
</device>
|
|
</root>
|
|
""")
|
|
scanner = ssdp.Scanner(hass)
|
|
|
|
with patch('netdisco.ssdp.scan', return_value=[
|
|
Mock(st="mock-st", location='http://1.1.1.1')
|
|
]), patch.dict(
|
|
gn_ssdp.SSDP['manufacturer'], {'Paulus': ['mock-domain']}
|
|
), patch.object(
|
|
hass.config_entries.flow, 'async_init',
|
|
return_value=mock_coro()
|
|
) as mock_init:
|
|
await scanner.async_scan(None)
|
|
|
|
assert len(mock_init.mock_calls) == 1
|
|
assert mock_init.mock_calls[0][1][0] == 'mock-domain'
|
|
assert mock_init.mock_calls[0][2]['context'] == {'source': 'ssdp'}
|
|
|
|
|
|
async def test_scan_match_device_type(hass, aioclient_mock):
|
|
"""Test matching based on ST."""
|
|
aioclient_mock.get('http://1.1.1.1', text="""
|
|
<root>
|
|
<device>
|
|
<deviceType>Paulus</deviceType>
|
|
</device>
|
|
</root>
|
|
""")
|
|
scanner = ssdp.Scanner(hass)
|
|
|
|
with patch('netdisco.ssdp.scan', return_value=[
|
|
Mock(st="mock-st", location='http://1.1.1.1')
|
|
]), patch.dict(
|
|
gn_ssdp.SSDP['device_type'], {'Paulus': ['mock-domain']}
|
|
), patch.object(
|
|
hass.config_entries.flow, 'async_init',
|
|
return_value=mock_coro()
|
|
) as mock_init:
|
|
await scanner.async_scan(None)
|
|
|
|
assert len(mock_init.mock_calls) == 1
|
|
assert mock_init.mock_calls[0][1][0] == 'mock-domain'
|
|
assert mock_init.mock_calls[0][2]['context'] == {'source': 'ssdp'}
|