Python ZhaFlowHandler 예제들

예제 #1

파일: test_config_flow.py 프로젝트: crazyfish1111/home

async def test_user_flow(hass):
    """Test that config flow works."""
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    with patch(
            'homeassistant.components.zha.config_flow'
            '.check_zigpy_connection',
            return_value=False):
        result = await flow.async_step_user(user_input={
            'usb_path': '/dev/ttyUSB1',
            'radio_type': 'ezsp'
        })

    assert result['errors'] == {'base': 'cannot_connect'}

    with patch(
            'homeassistant.components.zha.config_flow'
            '.check_zigpy_connection',
            return_value=True):
        result = await flow.async_step_user(user_input={
            'usb_path': '/dev/ttyUSB1',
            'radio_type': 'ezsp'
        })

    assert result['type'] == 'create_entry'
    assert result['title'] == '/dev/ttyUSB1'
    assert result['data'] == {'usb_path': '/dev/ttyUSB1', 'radio_type': 'ezsp'}

예제 #2

async def test_user_flow(hass):
    """Test that config flow works."""
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    with patch(
            "homeassistant.components.zha.config_flow"
            ".check_zigpy_connection",
            return_value=False,
    ):
        result = await flow.async_step_user(user_input={
            "usb_path": "/dev/ttyUSB1",
            "radio_type": "ezsp"
        })

    assert result["errors"] == {"base": "cannot_connect"}

    with patch(
            "homeassistant.components.zha.config_flow"
            ".check_zigpy_connection",
            return_value=True,
    ):
        result = await flow.async_step_user(user_input={
            "usb_path": "/dev/ttyUSB1",
            "radio_type": "ezsp"
        })

    assert result["type"] == "create_entry"
    assert result["title"] == "/dev/ttyUSB1"
    assert result["data"] == {"usb_path": "/dev/ttyUSB1", "radio_type": "ezsp"}

예제 #3

async def test_user_flow_existing_config_entry(hass):
    """Test if config entry already exists."""
    MockConfigEntry(domain=DOMAIN, data={
        "usb_path": "/dev/ttyUSB1"
    }).add_to_hass(hass)
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    result = await flow.async_step_user()

    assert result["type"] == "abort"

예제 #4

async def test_import_flow(hass):
    """Test import from configuration.yaml ."""
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    result = await flow.async_step_import({
        "usb_path": "/dev/ttyUSB1",
        "radio_type": "xbee"
    })

    assert result["type"] == "create_entry"
    assert result["title"] == "/dev/ttyUSB1"
    assert result["data"] == {"usb_path": "/dev/ttyUSB1", "radio_type": "xbee"}

예제 #5

파일: test_config_flow.py 프로젝트: crazyfish1111/home

async def test_import_flow(hass):
    """Test import from configuration.yaml ."""
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    result = await flow.async_step_import({
        'usb_path': '/dev/ttyUSB1',
        'radio_type': 'xbee',
    })

    assert result['type'] == 'create_entry'
    assert result['title'] == '/dev/ttyUSB1'
    assert result['data'] == {'usb_path': '/dev/ttyUSB1', 'radio_type': 'xbee'}

예제 #6

async def test_import_flow_existing_config_entry(hass):
    """Test import from configuration.yaml ."""
    MockConfigEntry(domain=DOMAIN, data={
        "usb_path": "/dev/ttyUSB1"
    }).add_to_hass(hass)
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    result = await flow.async_step_import({
        "usb_path": "/dev/ttyUSB1",
        "radio_type": "xbee"
    })

    assert result["type"] == "abort"

예제 #7

파일: test_config_flow.py 프로젝트: crazyfish1111/home

async def test_import_flow_existing_config_entry(hass):
    """Test import from configuration.yaml ."""
    MockConfigEntry(domain=DOMAIN, data={
        'usb_path': '/dev/ttyUSB1'
    }).add_to_hass(hass)
    flow = config_flow.ZhaFlowHandler()
    flow.hass = hass

    result = await flow.async_step_import({
        'usb_path': '/dev/ttyUSB1',
        'radio_type': 'xbee',
    })

    assert result['type'] == 'abort'