async def test_flow_works_multiple_sites(hass, aioclient_mock):
    'Test config flow works when finding multiple sites.'
    result = (await hass.config_entries.flow.async_init(UNIFI_DOMAIN, context={'source': 'user'}))
    assert (result['type'] == data_entry_flow.RESULT_TYPE_FORM)
    assert (result['step_id'] == 'user')
    aioclient_mock.get('https://1.2.3.4:1234', status=302)
    aioclient_mock.post('https://1.2.3.4:1234/api/login', json={'data': 'login successful', 'meta': {'rc': 'ok'}}, headers={'content-type': CONTENT_TYPE_JSON})
    aioclient_mock.get('https://1.2.3.4:1234/api/self/sites', json={'data': [{'name': 'default', 'role': 'admin', 'desc': 'site name'}, {'name': 'site2', 'role': 'admin', 'desc': 'site2 name'}], 'meta': {'rc': 'ok'}}, headers={'content-type': CONTENT_TYPE_JSON})
    result = (await hass.config_entries.flow.async_configure(result['flow_id'], user_input={CONF_HOST: '1.2.3.4', CONF_USERNAME: 'username', CONF_PASSWORD: 'password', CONF_PORT: 1234, CONF_VERIFY_SSL: True}))
    assert (result['type'] == data_entry_flow.RESULT_TYPE_FORM)
    assert (result['step_id'] == 'site')
    assert result['data_schema']({'site': 'default'})
    assert result['data_schema']({'site': 'site2'})
Test config flow works when finding multiple sites.
async def test_flow_works_multiple_sites(hass, aioclient_mock):
    
    result = (await hass.config_entries.flow.async_init(UNIFI_DOMAIN, context={'source': 'user'}))
    assert (result['type'] == data_entry_flow.RESULT_TYPE_FORM)
    assert (result['step_id'] == 'user')
    aioclient_mock.get('https://1.2.3.4:1234', status=302)
    aioclient_mock.post('https://1.2.3.4:1234/api/login', json={'data': 'login successful', 'meta': {'rc': 'ok'}}, headers={'content-type': CONTENT_TYPE_JSON})
    aioclient_mock.get('https://1.2.3.4:1234/api/self/sites', json={'data': [{'name': 'default', 'role': 'admin', 'desc': 'site name'}, {'name': 'site2', 'role': 'admin', 'desc': 'site2 name'}], 'meta': {'rc': 'ok'}}, headers={'content-type': CONTENT_TYPE_JSON})
    result = (await hass.config_entries.flow.async_configure(result['flow_id'], user_input={CONF_HOST: '1.2.3.4', CONF_USERNAME: 'username', CONF_PASSWORD: 'password', CONF_PORT: 1234, CONF_VERIFY_SSL: True}))
    assert (result['type'] == data_entry_flow.RESULT_TYPE_FORM)
    assert (result['step_id'] == 'site')
    assert result['data_schema']({'site': 'default'})
    assert result['data_schema']({'site': 'site2'})