52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""Test Nest local auth."""
|
|
from homeassistant.components.nest import const, config_flow, local_auth
|
|
from urllib.parse import parse_qsl
|
|
|
|
import pytest
|
|
|
|
import requests_mock as rmock
|
|
|
|
|
|
@pytest.fixture
|
|
def registered_flow(hass):
|
|
"""Mock a registered flow."""
|
|
local_auth.initialize(hass, 'TEST-CLIENT-ID', 'TEST-CLIENT-SECRET')
|
|
return hass.data[config_flow.DATA_FLOW_IMPL][const.DOMAIN]
|
|
|
|
|
|
async def test_generate_auth_url(registered_flow):
|
|
"""Test generating an auth url.
|
|
|
|
Mainly testing that it doesn't blow up.
|
|
"""
|
|
url = await registered_flow['gen_authorize_url']('TEST-FLOW-ID')
|
|
assert url is not None
|
|
|
|
|
|
async def test_convert_code(requests_mock, registered_flow):
|
|
"""Test converting a code."""
|
|
from nest.nest import ACCESS_TOKEN_URL
|
|
|
|
def token_matcher(request):
|
|
"""Match a fetch token request."""
|
|
if request.url != ACCESS_TOKEN_URL:
|
|
return None
|
|
|
|
assert dict(parse_qsl(request.text)) == {
|
|
'client_id': 'TEST-CLIENT-ID',
|
|
'client_secret': 'TEST-CLIENT-SECRET',
|
|
'code': 'TEST-CODE',
|
|
'grant_type': 'authorization_code'
|
|
}
|
|
|
|
return rmock.create_response(request, json={
|
|
'access_token': 'TEST-ACCESS-TOKEN'
|
|
})
|
|
|
|
requests_mock.add_matcher(token_matcher)
|
|
|
|
tokens = await registered_flow['convert_code']('TEST-CODE')
|
|
assert tokens == {
|
|
'access_token': 'TEST-ACCESS-TOKEN'
|
|
}
|