aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorFrederick Muriuki Muriithi2023-09-08 05:52:38 +0300
committerFrederick Muriuki Muriithi2023-09-26 03:44:27 +0300
commit7cfd3ba3e2bc37f2433301b455b70034690a5683 (patch)
tree796126f60c9059a36d73d03ebc4c7b5ad35737d6
parent666bff009e437a8bca6003b2e6ce4acd451e8e0e (diff)
downloadgn-auth-7cfd3ba3e2bc37f2433301b455b70034690a5683.tar.gz
migrations: Create `group_resources` table.
-rw-r--r--migrations/auth/20230907_04_3LnrG-refactor-create-group-resources-table.py58
-rw-r--r--tests/unit/auth/test_migrations_create_tables.py4
2 files changed, 61 insertions, 1 deletions
diff --git a/migrations/auth/20230907_04_3LnrG-refactor-create-group-resources-table.py b/migrations/auth/20230907_04_3LnrG-refactor-create-group-resources-table.py
new file mode 100644
index 0000000..a26834a
--- /dev/null
+++ b/migrations/auth/20230907_04_3LnrG-refactor-create-group-resources-table.py
@@ -0,0 +1,58 @@
+"""
+refactor: create 'group_resources' table.
+"""
+
+import uuid
+import random
+import string
+
+import sqlite3
+from yoyo import step
+
+__depends__ = {'20230907_03_BwAmf-refactor-drop-group-id-from-resources-table'}
+
+def randstr(length: int = 5):
+ """Generate random string."""
+ return "".join(random.choices(
+ string.ascii_letters + string.digits, k=length))
+
+def create_and_link_resources_for_existing_groups(conn):
+ conn.row_factory = sqlite3.Row
+ cursor = conn.cursor()
+ cursor.execute("SELECT group_id, group_name FROM groups")
+ resources = tuple({
+ "group_id": row["group_id"],
+ "resource_id": str(uuid.uuid4()),
+ "resource_name": f"{randstr(10)}: {row['group_name']}",
+ "resource_category_id": "1e0f70ee-add5-4358-8c6c-43de77fa4cce"
+ } for row in cursor.fetchall())
+ cursor.executemany(
+ "INSERT INTO "
+ "resources(resource_id, resource_name, resource_category_id) "
+ "VALUES (:resource_id, :resource_name, :resource_category_id)",
+ resources)
+ cursor.executemany(
+ "INSERT INTO group_resources(resource_id, group_id) "
+ "VALUES (:resource_id, :group_id)",
+ resources)
+
+steps = [
+ step(
+ """
+ CREATE TABLE IF NOT EXISTS group_resources(
+ -- Links groups to the resources of type 'group' that control access to
+ -- each group
+ resource_id TEXT NOT NULL,
+ group_id TEXT NOT NULL,
+ PRIMARY KEY(resource_id, group_id),
+ FOREIGN KEY (resource_id)
+ REFERENCES resources(resource_id)
+ ON UPDATE CASCADE ON DELETE RESTRICT,
+ FOREIGN KEY (group_id)
+ REFERENCES groups(group_id)
+ ON UPDATE CASCADE ON DELETE RESTRICT
+ ) WITHOUT ROWID
+ """,
+ "DROP TABLE IF EXISTS group_resources"),
+ step(create_and_link_resources_for_existing_groups)
+]
diff --git a/tests/unit/auth/test_migrations_create_tables.py b/tests/unit/auth/test_migrations_create_tables.py
index eb0e161..8b0f36e 100644
--- a/tests/unit/auth/test_migrations_create_tables.py
+++ b/tests/unit/auth/test_migrations_create_tables.py
@@ -41,7 +41,9 @@ migrations_and_tables = (
("20230410_01_8mwaf-create-linked-mrna-data-table.py", "linked_mrna_data"),
("20230410_02_WZqSf-create-mrna-resources-table.py", "mrna_resources"),
("20230907_01_pjnxz-refactor-add-resource-ownership-table.py",
- "resource_ownership"))
+ "resource_ownership"),
+ ("20230907_04_3LnrG-refactor-create-group-resources-table.py",
+ "group_resources"))
@pytest.mark.unit_test
@pytest.mark.parametrize("migration_file,the_table", migrations_and_tables)