Testlar qo'shildi

core/apps/api/models/ad/ad.py

@@ -3,6 +3,7 @@ from django_core.models.base import AbstractBaseModel
 from django.utils.translation import gettext_lazy as _
 from django.contrib.auth import get_user_model
 from core.apps.api.choices.ad_type import AdType, AdCategoryType
+from model_bakery import baker
 
 
 class AdModel(AbstractBaseModel):
@@ -18,6 +19,10 @@ class AdModel(AbstractBaseModel):
     tags = models.ManyToManyField("api.Tags", verbose_name=_("Tags"))
     image = models.ImageField(verbose_name=_("Image"))
 
+    @classmethod
+    def _baker(cls):
+        return baker.make(cls)
+
     def __str__(self):
         return str(self.pk)
 

core/apps/api/models/ad/category.py

@@ -2,6 +2,7 @@ from django.db import models
 from django_core.models.base import AbstractBaseModel
 from django.utils.translation import gettext_lazy as _
 from core.apps.api.choices import AdCategoryType
+from model_bakery import baker
 
 
 class Category(AbstractBaseModel):
@@ -13,6 +14,10 @@ class Category(AbstractBaseModel):
     category_type = models.CharField(max_length=255, verbose_name=_('Category Type'), choices=AdCategoryType,
                                      default=AdCategoryType.PRODUCT)
 
+    @classmethod
+    def _baker(cls):
+        return baker.make(cls)
+
     def __str__(self):
         return str(self.pk)