diff options
Diffstat (limited to 'tests/unit/auth/test_resources.py')
| -rw-r--r-- | tests/unit/auth/test_resources.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tests/unit/auth/test_resources.py b/tests/unit/auth/test_resources.py index b756cd9..e6ebeb9 100644 --- a/tests/unit/auth/test_resources.py +++ b/tests/unit/auth/test_resources.py @@ -3,6 +3,7 @@ import uuid import pytest +from gn3.auth import db from gn3.auth.authorisation.groups import Group from gn3.auth.authorisation.resources import ( Resource, user_resources, create_resource, ResourceCategory, @@ -35,10 +36,14 @@ def test_create_resource(mocker, fxtr_app, fxtr_users_in_group, user, expected): """Test that resource creation works as expected.""" mocker.patch("gn3.auth.authorisation.resources.uuid4", uuid_fn) conn, _group, _users = fxtr_users_in_group - with fxtr_app.app_context() as flask_context: + with fxtr_app.app_context() as flask_context, db.cursor(conn) as cursor: flask_context.g.user = user assert create_resource(conn, "test_resource", resource_category) == expected + # Cleanup + cursor.execute( + "DELETE FROM resources WHERE resource_id=?", (str(uuid_fn()),)) + SORTKEY = lambda resource: resource.resource_id @pytest.mark.unit_test @@ -74,5 +79,5 @@ def test_user_resources(fxtr_group_user_roles, user, expected): WHEN: a particular user's resources are requested THEN: list only the resources for which the user can access """ - conn = fxtr_group_user_roles + conn, *_others = fxtr_group_user_roles assert sorted(user_resources(conn, user), key=SORTKEY) == expected |