Added a factory file for the Category Model

project/accounts/factory.py

@@ -0,0 +1,49 @@
+import factory
+import factory.fuzzy
+from django.contrib.auth.models import User
+
+from .modes import Profile
+
+
+class UserFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = User
+
+    username = factory.Faker("sentence", nb_words=20)
+
+
+class ProfileFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Profile
+
+    user = factory.SubFactory("app.factories.UserFactory")
+    first_name = factory.Faker("lorem")
+    last_name = factory.Faker("lorem")
+    about_me = factory.Faker("lorem")
+    is_verified = factory.fuzzy.FuzzyChoice(choices=[False, False, False, False])
+    profile_image = factory.Faker("image_url")
+    profile_image_thumb = factory.Faker("image_url")
+
+    @factory.post_generation
+    def categories(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for category in extracted:
+                self.categories.add(category)
+
+    @factory.post_generation
+    def tags(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for tag in extracted:
+                self.tags.add(tag)
+
+    @factory.post_generation
+    def following(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for follower in extracted:
+                self.following.add(follower)

project/categories/factory.py

@@ -0,0 +1,10 @@
+import factory
+
+from .models import Category
+
+
+class CategoryFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Category
+
+    name = factory.Faker("lorem")

project/notification/factory.py

@@ -0,0 +1,17 @@
+import factory
+import factory.fuzzy
+from account.factory import ProfileFactory
+from threads.factory import CiviFactory, ThreadFactory
+
+from .models import Notification
+
+
+class NotificationFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Notification
+
+    account = factory.SubFactory(ProfileFactory)
+    thread = factory.SubFactory(ThreadFactory)
+    civi = factory.SubFactory(CiviFactory)
+    activity_type = factory.Faker("lorem")
+    read = factory.fuzzy.FuzzyChoice(choices=[True, True, True, False])

project/threads/factory.py

@@ -0,0 +1,92 @@
+import factory
+import factory.fuzzy
+from categories.factory import CategoryFactory
+from django.contrib.auth.models import User
+
+from .models import Civi, Fact, Thread
+
+
+class FactFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Fact
+
+    body = factory.Sequence(lambda n: "Fact #%s" % n)
+
+
+class ThreadFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Thread
+
+    author = User.objects.get_or_create(username="admin")
+    category = factory.SubFactory(CategoryFactory)
+    title = factory.Faker("lorem")
+    summary = factory.Faker("lorem")
+    image = factory.Faker("image_url")
+    is_draft = factory.fuzzy.FuzzyChoice(choices=[True, True, True, True])
+    num_views = 0
+    num_civis = 0
+    num_solutions = 0
+
+    @factory.post_generation
+    def facts(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for fact in extracted:
+                self.facts.add(fact)
+
+    @factory.post_generation
+    def tags(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for tag in extracted:
+                self.tags.add(tag)
+
+    @factory.post_generation
+    def objects(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for obje in extracted:
+                self.objects.add(obje)
+
+
+class CiviFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Civi
+
+    author = User.objects.get_or_create(username="admin")
+    thread = factory.SubFactory(ThreadFactory)
+    title = factory.Faker("lorem")
+    body = factory.Faker("lorem")
+    c_type = factory.Faker("lorem")
+    votes_vneg = 0
+    votes_neg = 0
+    votes_neutral = 0
+    votes_pos = 0
+    votes_vpos = 0
+
+    @factory.post_generation
+    def linked_civis(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for linked_civi in extracted:
+                self.linked_civis.add(linked_civi)
+
+    @factory.post_generation
+    def tags(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for tag in extracted:
+                self.tags.add(tag)
+
+    @factory.post_generation
+    def objects(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for obje in extracted:
+                self.objects.add(obje)

pyproject.toml

@@ -30,6 +30,7 @@ ipython = "^8.5.0"
 django-debug-toolbar = "^3.7.0"
 django-linear-migrations = "^2.5.1"
 pre-commit = "^2.20.0"
+factory-boy = "^3.2.1"
 
 [tool.isort]
 profile = "black"