From e70974a4f1d397be7c39c46ba710caa731e8e354 Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Fri, 16 Dec 2022 09:49:26 -0800 Subject: [PATCH] Ran Black, refs #1959 --- tests/test_internal_db.py | 36 +++++++++++++++++++++++------------- tests/test_permissions.py | 2 +- 2 files changed, 24 insertions(+), 14 deletions(-) diff --git a/tests/test_internal_db.py b/tests/test_internal_db.py index 3f8d6904..569b9c5e 100644 --- a/tests/test_internal_db.py +++ b/tests/test_internal_db.py @@ -6,16 +6,20 @@ import pytest async def test_internal_only_available_to_root(ds_client): cookie = ds_client.actor_cookie({"id": "root"}) assert (await ds_client.get("/_internal")).status_code == 403 - assert (await ds_client.get("/_internal", cookies={"ds_actor": cookie})).status_code == 200 + assert ( + await ds_client.get("/_internal", cookies={"ds_actor": cookie}) + ).status_code == 200 @pytest.mark.ds_client @pytest.mark.asyncio async def test_internal_databases(ds_client): cookie = ds_client.actor_cookie({"id": "root"}) - databases = (await ds_client.get( - "/_internal/databases.json?_shape=array", cookies={"ds_actor": cookie} - )).json() + databases = ( + await ds_client.get( + "/_internal/databases.json?_shape=array", cookies={"ds_actor": cookie} + ) + ).json() assert len(databases) == 2 assert databases[0]["database_name"] == "_internal" assert databases[1]["database_name"] == "fixtures" @@ -25,9 +29,11 @@ async def test_internal_databases(ds_client): @pytest.mark.asyncio async def test_internal_tables(ds_client): cookie = ds_client.actor_cookie({"id": "root"}) - tables = (await ds_client.get( - "/_internal/tables.json?_shape=array", cookies={"ds_actor": cookie} - )).json() + tables = ( + await ds_client.get( + "/_internal/tables.json?_shape=array", cookies={"ds_actor": cookie} + ) + ).json() assert len(tables) > 5 table = tables[0] assert set(table.keys()) == {"rootpage", "table_name", "database_name", "sql"} @@ -37,9 +43,11 @@ async def test_internal_tables(ds_client): @pytest.mark.asyncio async def test_internal_indexes(ds_client): cookie = ds_client.actor_cookie({"id": "root"}) - indexes = (await ds_client.get( - "/_internal/indexes.json?_shape=array", cookies={"ds_actor": cookie} - )).json() + indexes = ( + await ds_client.get( + "/_internal/indexes.json?_shape=array", cookies={"ds_actor": cookie} + ) + ).json() assert len(indexes) > 5 index = indexes[0] assert set(index.keys()) == { @@ -57,9 +65,11 @@ async def test_internal_indexes(ds_client): @pytest.mark.asyncio async def test_internal_foreign_keys(ds_client): cookie = ds_client.actor_cookie({"id": "root"}) - foreign_keys = (await ds_client.get( - "/_internal/foreign_keys.json?_shape=array", cookies={"ds_actor": cookie} - )).json() + foreign_keys = ( + await ds_client.get( + "/_internal/foreign_keys.json?_shape=array", cookies={"ds_actor": cookie} + ) + ).json() assert len(foreign_keys) > 5 foreign_key = foreign_keys[0] assert set(foreign_key.keys()) == { diff --git a/tests/test_permissions.py b/tests/test_permissions.py index 730f672a..493cfbc0 100644 --- a/tests/test_permissions.py +++ b/tests/test_permissions.py @@ -407,7 +407,7 @@ async def test_permissions_debug(ds_client): ], ) async def test_allow_debug(ds_client, actor, allow, expected_fragment): - response = await ds_client.get( + response = await ds_client.get( "/-/allow-debug?" + urllib.parse.urlencode({"actor": actor, "allow": allow}) ) assert response.status_code == 200