feat: added advertiser and campaign apps

2025-02-20 22:54:28 +03:00
parent 4793973233
commit 0f77f9eb73
24 changed files with 1580 additions and 0 deletions
@@ -0,0 +1,118 @@
+from uuid import uuid4
+from django.core.cache import cache
+from django.core.exceptions import ValidationError
+from django.test import TestCase, override_settings
+from apps.advertiser.models import Advertiser
+from apps.campaign.models import Campaign
+from apps.client.models import Client
+
+
+class CampaignModelTest(TestCase):
+    @classmethod
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def setUpTestData(cls):
+        cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
+        cls.campaign = Campaign.objects.create(
+            advertiser=cls.advertiser,
+            impressions_limit=1000,
+            clicks_limit=500,
+            cost_per_impression=0.05,
+            cost_per_click=0.10,
+            ad_title="Test Campaign",
+            ad_text="This is a test campaign.",
+            start_date=1,
+            end_date=10,
+        )
+
+    def test_campaign_creation(self):
+        self.assertIsInstance(self.campaign, Campaign)
+        self.assertEqual(self.campaign.ad_title, "Test Campaign")
+
+    def test_campaign_str_method(self):
+        self.assertEqual(str(self.campaign), "Test Campaign")
+
+    def test_campaign_id_property(self):
+        self.assertEqual(self.campaign.campaign_id, self.campaign.id)
+        new_id = uuid4()
+        self.campaign.campaign_id = new_id
+        self.assertEqual(self.campaign.id, new_id)
+
+    def test_ad_id_property(self):
+        self.assertEqual(self.campaign.ad_id, self.campaign.id)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_started_property(self):
+        cache.set("current_date", 5)
+        self.assertTrue(self.campaign.started)
+        cache.set("current_date", 0)
+        self.assertFalse(self.campaign.started)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_active_property(self):
+        cache.set("current_date", 5)
+        self.assertTrue(self.campaign.active)
+        cache.set("current_date", 11)
+        self.assertFalse(self.campaign.active)
+        cache.set("current_date", 5)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_clean_method(self):
+        self.campaign.start_date = -1
+
+        with self.assertRaises(ValidationError):
+            self.campaign.clean()
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_view_method(self):
+        client = Client.objects.create(
+            login="test_client", age=15, location="Moscow", gender="FEMALE"
+        )
+        self.campaign.view(client)
+
+        self.assertEqual(self.campaign.impressions.count(), 1)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_click_method(self):
+        client = Client.objects.create(
+            login="test_client", age=15, location="Moscow", gender="FEMALE"
+        )
+        self.campaign.view(client)
+        self.campaign.click(client)
+
+        self.assertEqual(self.campaign.clicks.count(), 1)