refactor data fixtures

atrox3d · atrox3d · commit befbf028f937 · 2024-08-29T19:49:44.000+02:00
diff --git a/tests/fixtures/fakedata/__init__.py b/tests/fixtures/fakedata/__init__.py
diff --git a/tests/fixtures/fakedata/posts.py b/tests/fixtures/fakedata/posts.py
@@ -0,0 +1,25 @@
+import pytest
+
+from schemas import post
+from schemas import user
+from schemas import vote
+from db.orm import models
+
+@pytest.fixture
+def fake_posts() -> list[post.PostCreate]:
+    return [
+        post.PostCreate(title='1st title', content='1st content'),
+        post.PostCreate(title='2st title', content='1st content'),
+        post.PostCreate(title='3st title', content='1st content'),
+    ]
+
+@pytest.fixture
+def fake_post_models_db(fake_posts, add_user_db) -> list[models.Post]:
+    return [models.Post(**_post.model_dump(), owner_id=add_user_db.id) 
+            for _post in fake_posts]
+
+@pytest.fixture
+def add_fake_posts_db(session, fake_post_models_db) -> list[models.Post]:
+    session.add_all(fake_post_models_db)
+    session.commit()
+    return session.query(models.Post).all()
diff --git a/tests/fixtures/fakedata/users.py b/tests/fixtures/fakedata/users.py
@@ -0,0 +1,44 @@
+import pytest
+
+from schemas import post
+from schemas import user
+from schemas import vote
+from db.orm import models
+
+@pytest.fixture
+def fake_emails_dict() -> list[dict[str, str]]:
+    return [ 
+                {'email': 'testuser@gmail.com',},
+                {'email': 'cat@gmail.com',},
+                {'email': 'dog@gmail.com',},
+    ]
+
+@pytest.fixture
+def fake_usernames_dict() -> list[dict[str, str]]:
+    return [ 
+                {'username': 'testuser@gmail.com',},
+                {'username': 'cat@gmail.com',},
+                {'username': 'dog@gmail.com',},
+    ]
+
+@pytest.fixture
+def fake_passwords_dict() -> list[dict[str, str]]:
+    return [ 
+                {'password': 'password',},
+                {'password': 'meow',},
+                {'password': 'wof',},
+    ]
+
+@pytest.fixture
+def fake_users_creation_dict(
+            fake_emails_dict,
+            fake_passwords_dict
+) -> list[dict[str, str]]:
+    return [email | password for email, password in zip(fake_emails_dict, fake_passwords_dict)]
+
+@pytest.fixture
+def fake_users_login_dict(
+            fake_usernames_dict,
+            fake_passwords_dict
+) -> list[dict[str, str]]:
+    return [email | password for email, password in zip(fake_usernames_dict, fake_passwords_dict)]
