New Addon

Google BK

hassio-google-drive-backup/tests/test_server.py

@@ -0,0 +1,50 @@
+
+import pytest
+from yarl import URL
+from dev.simulationserver import SimulationServer
+from aiohttp import ClientSession, hdrs
+from backup.config import Config
+
+
+@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"