parent
ecba87179f
commit
64306922b1
|
@ -31,10 +31,30 @@ def client(hass, hass_client):
|
||||||
yield hass.loop.run_until_complete(hass_client())
|
yield hass.loop.run_until_complete(hass_client())
|
||||||
|
|
||||||
|
|
||||||
|
@HANDLERS.register('comp1')
|
||||||
|
class Comp1ConfigFlow:
|
||||||
|
"""Config flow with options flow."""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
@callback
|
||||||
|
def async_get_options_flow(config, options):
|
||||||
|
"""Get options flow."""
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@HANDLERS.register('comp2')
|
||||||
|
class Comp2ConfigFlow:
|
||||||
|
"""Config flow without options flow."""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
"""Init."""
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
async def test_get_entries(hass, client):
|
async def test_get_entries(hass, client):
|
||||||
"""Test get entries."""
|
"""Test get entries."""
|
||||||
MockConfigEntry(
|
MockConfigEntry(
|
||||||
domain='comp',
|
domain='comp1',
|
||||||
title='Test 1',
|
title='Test 1',
|
||||||
source='bla',
|
source='bla',
|
||||||
connection_class=core_ce.CONN_CLASS_LOCAL_POLL,
|
connection_class=core_ce.CONN_CLASS_LOCAL_POLL,
|
||||||
|
@ -47,18 +67,6 @@ async def test_get_entries(hass, client):
|
||||||
connection_class=core_ce.CONN_CLASS_ASSUMED,
|
connection_class=core_ce.CONN_CLASS_ASSUMED,
|
||||||
).add_to_hass(hass)
|
).add_to_hass(hass)
|
||||||
|
|
||||||
class CompConfigFlow:
|
|
||||||
@staticmethod
|
|
||||||
@callback
|
|
||||||
def async_get_options_flow(config, options):
|
|
||||||
pass
|
|
||||||
HANDLERS['comp'] = CompConfigFlow()
|
|
||||||
|
|
||||||
class Comp2ConfigFlow:
|
|
||||||
def __init__(self):
|
|
||||||
pass
|
|
||||||
HANDLERS['comp2'] = Comp2ConfigFlow()
|
|
||||||
|
|
||||||
resp = await client.get('/api/config/config_entries/entry')
|
resp = await client.get('/api/config/config_entries/entry')
|
||||||
assert resp.status == 200
|
assert resp.status == 200
|
||||||
data = await resp.json()
|
data = await resp.json()
|
||||||
|
@ -66,7 +74,7 @@ async def test_get_entries(hass, client):
|
||||||
entry.pop('entry_id')
|
entry.pop('entry_id')
|
||||||
assert data == [
|
assert data == [
|
||||||
{
|
{
|
||||||
'domain': 'comp',
|
'domain': 'comp1',
|
||||||
'title': 'Test 1',
|
'title': 'Test 1',
|
||||||
'source': 'bla',
|
'source': 'bla',
|
||||||
'state': 'not_loaded',
|
'state': 'not_loaded',
|
||||||
|
|
Loading…
Reference in New Issue