mirror of
https://github.com/Mesteriis/hassio-addons-avm.git
synced 2026-01-10 15:31:02 +01:00
60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
|
|
import pytest
|
|
from yarl import URL
|
|
from dev.simulationserver import SimulationServer
|
|
from aiohttp import ClientSession, hdrs
|
|
from backup.config import Config
|
|
from .faketime import FakeTime
|
|
import json
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_known_error(server: SimulationServer, session: ClientSession, config: Config, server_url: URL):
|
|
async with session.post(server_url.with_path("drive/refresh"), json={"blah": "blah"}) as r:
|
|
assert r.status == 503
|
|
assert await r.json() == {
|
|
'error': "Required key 'refresh_token' was missing from the request payload"
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_unknown_error(server: SimulationServer, session: ClientSession, config: Config, server_url: URL):
|
|
async with session.post(server_url.with_path("drive/refresh"), data={}) as r:
|
|
assert r.status == 500
|
|
assert len((await r.json())["error"]) > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_old_auth_method(server: SimulationServer, session: ClientSession, server_url: URL):
|
|
start_auth = server_url.with_path("drive/authorize").with_query({
|
|
"redirectbacktoken": "http://example.com"
|
|
})
|
|
|
|
# Verify the redirect to Drive's oauthv2 endpoint
|
|
async with session.get(start_auth, data={}, allow_redirects=False) as r:
|
|
assert r.status == 303
|
|
redirect = URL(r.headers[hdrs.LOCATION])
|
|
assert redirect.path == "/o/oauth2/v2/auth"
|
|
assert redirect.host == "localhost"
|
|
|
|
# Verify the redirect back to the server's oauth page
|
|
async with session.get(redirect, data={}, allow_redirects=False) as r:
|
|
assert r.status == 303
|
|
redirect = URL(r.headers[hdrs.LOCATION])
|
|
assert redirect.path == "/drive/authorize"
|
|
assert redirect.host == "localhost"
|
|
|
|
# Verify we gte redirected back to the addon (example.com) with creds
|
|
async with session.get(redirect, data={}, allow_redirects=False) as r:
|
|
assert r.status == 303
|
|
redirect = URL(r.headers[hdrs.LOCATION])
|
|
assert redirect.query.get("creds") is not None
|
|
assert redirect.host == "example.com"
|
|
|
|
|
|
async def test_log_to_firestore(time: FakeTime, server: SimulationServer, session: ClientSession, server_url: URL):
|
|
data = {"info": "testing"}
|
|
async with session.post(server_url.with_path("logerror"), data=json.dumps(data)) as r:
|
|
assert r.status == 200
|
|
assert server._authserver.error_store.last_error is not None
|
|
assert server._authserver.error_store.last_error['report'] == data
|