1 files changed, 38 insertions, 1 deletions
diff --git a/tests/unit/auth/conftest.py b/tests/unit/auth/conftest.py
index 0c6b7c7..4f4f1f8 100644
--- a/tests/unit/auth/conftest.py
+++ b/tests/unit/auth/conftest.py
@@ -1,10 +1,14 @@
 """Fixtures for auth tests."""
+import uuid
+
 import pytest
 from yoyo.backends import DatabaseBackend
 from yoyo import get_backend, read_migrations
 from yoyo.migrations import Migration, MigrationList
 
 from gn3.auth import db
+from gn3.auth.authentication.users import User
+from gn3.auth.authorisation.groups import Group
 from gn3.migrations import apply_migrations, rollback_migrations
 
 @pytest.fixture(scope="session")
@@ -51,6 +55,17 @@ def migrations_up_to(migration, migrations_dir):
     return MigrationList(migrations[0:index])
 
 @pytest.fixture(scope="function")
+def test_group(conn_after_auth_migrations):# pylint: disable=[redefined-outer-name]
+    """Fixture: setup a test group."""
+    query = "INSERT INTO groups(group_id, group_name) VALUES (?, ?)"
+    group_id = uuid.UUID("9988c21d-f02f-4d45-8966-22c968ac2fbf")
+    group_name = "TheTestGroup"
+    with db.cursor(conn_after_auth_migrations) as cursor:
+        cursor.execute(query, (str(group_id), group_name))
+
+    yield (conn_after_auth_migrations, Group(group_id, group_name))
+
+@pytest.fixture(scope="function")
 def test_users(conn_after_auth_migrations):# pylint: disable=[redefined-outer-name]
     """Fixture: setup test users."""
     query = "INSERT INTO users(user_id, email, name) VALUES (?, ?, ?)"
@@ -71,7 +86,8 @@ def test_users(conn_after_auth_migrations):# pylint: disable=[redefined-outer-na
         cursor.executemany(query, the_users)
         cursor.executemany(query_user_roles, test_user_roles)
 
-    yield conn_after_auth_migrations
+    yield (conn_after_auth_migrations, tuple(
+        User(uuid.UUID(uid), email, name) for uid, email, name in the_users))
 
     with db.cursor(conn_after_auth_migrations) as cursor:
         cursor.executemany(
@@ -83,3 +99,24 @@ def test_users(conn_after_auth_migrations):# pylint: disable=[redefined-outer-na
              ("21351b66-8aad-475b-84ac-53ce528451e3",),
              ("ae9c6245-0966-41a5-9a5e-20885a96bea7",),
              ("9a0c7ce5-2f40-4e78-979e-bf3527a59579",)))
+
+@pytest.fixture(scope="function")
+def test_users_in_group(test_group, test_users):#pytest: disable=[redefined-outer-name]
+    """Link the users to the groups."""
+    conn = test_group[0]
+    group = test_group[1]
+    users = test_users[1]
+    query_params = (
+        (str(group.group_id), str(user.user_id)) for user in users
+        if user.email not in ("unaff@iliated.user",))
+    with db.cursor(conn) as cursor:
+        cursor.execute(
+            "INSERT INTO group_users(group_id, user_id) VALUES (?, ?)",
+            query_params)
+
+    yield (conn, group, users)
+
+    with db.cursor(conn) as cursor:
+        cursor.execute(
+            "DELETE FROM group_users WHERE group_id=? AND user_id=?",
+            query_params)