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"
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