示例#1
0
async def test_already_configured(opp, mock_hub_discover):
    """Test that flow aborts when all hubs are configured."""

    dummy_hub_1 = aiopulse.Hub(DUMMY_HOST1)
    dummy_hub_1.id = "ABC123"

    mock_hub_discover.return_value = async_generator([dummy_hub_1])

    MockConfigEntry(domain=DOMAIN, unique_id=dummy_hub_1.id,
                    data=CONFIG).add_to_opp(opp)

    result = await opp.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER})

    assert result["type"] == "abort"
    assert result["reason"] == "no_devices_found"
示例#2
0
async def test_show_form_one_hub(hass, mock_hub_discover, mock_hub_run):
    """Test that a config is created when one hub discovered."""

    dummy_hub_1 = aiopulse.Hub(DUMMY_HOST1)
    dummy_hub_1.id = "ABC123"

    mock_hub_discover.return_value = async_generator([dummy_hub_1])

    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER})

    assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
    assert result["title"] == dummy_hub_1.id
    assert result["result"].data == {
        "host": DUMMY_HOST1,
    }

    # Check we performed the discovery
    assert len(mock_hub_discover.mock_calls) == 1