57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
|
"""Tests for PTVSD Debugger."""
|
||
|
|
|
||
|
|
from unittest.mock import patch
|
||
|
|
from asynctest import CoroutineMock
|
||
|
|
from pytest import mark
|
||
|
|
|
||
|
|
import homeassistant.components.ptvsd as ptvsd_component
|
||
|
|
from homeassistant.setup import async_setup_component
|
||
|
|
from homeassistant.bootstrap import _async_set_up_integrations
|
||
|
|
|
||
|
|
|
||
|
|
@mark.skip('causes code cover to fail')
|
||
|
|
async def test_ptvsd(hass):
|
||
|
|
"""Test loading ptvsd component."""
|
||
|
|
with patch('ptvsd.enable_attach') as attach:
|
||
|
|
with patch('ptvsd.wait_for_attach') as wait:
|
||
|
|
assert await async_setup_component(
|
||
|
|
hass, ptvsd_component.DOMAIN, {
|
||
|
|
ptvsd_component.DOMAIN: {}
|
||
|
|
})
|
||
|
|
|
||
|
|
attach.assert_called_once_with(('0.0.0.0', 5678))
|
||
|
|
assert wait.call_count == 0
|
||
|
|
|
||
|
|
|
||
|
|
@mark.skip('causes code cover to fail')
|
||
|
|
async def test_ptvsd_wait(hass):
|
||
|
|
"""Test loading ptvsd component with wait."""
|
||
|
|
with patch('ptvsd.enable_attach') as attach:
|
||
|
|
with patch('ptvsd.wait_for_attach') as wait:
|
||
|
|
assert await async_setup_component(
|
||
|
|
hass, ptvsd_component.DOMAIN, {
|
||
|
|
ptvsd_component.DOMAIN: {
|
||
|
|
ptvsd_component.CONF_WAIT: True
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
attach.assert_called_once_with(('0.0.0.0', 5678))
|
||
|
|
assert wait.call_count == 1
|
||
|
|
|
||
|
|
|
||
|
|
async def test_ptvsd_bootstrap(hass):
|
||
|
|
"""Test loading ptvsd component with wait."""
|
||
|
|
config = {
|
||
|
|
ptvsd_component.DOMAIN: {
|
||
|
|
ptvsd_component.CONF_WAIT: True
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
with patch(
|
||
|
|
'homeassistant.components.ptvsd.async_setup',
|
||
|
|
CoroutineMock()) as setup_mock:
|
||
|
|
setup_mock.return_value = True
|
||
|
|
await _async_set_up_integrations(hass, config)
|
||
|
|
|
||
|
|
assert setup_mock.call_count == 1
|