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,53 @@
+from django.test import TestCase, override_settings
+from django.core.cache import cache
+from apps.advertiser.models import Advertiser
+from apps.campaign.models import Campaign, CampaignClick
+from apps.client.models import Client
+
+
+class CampaignClickModelTest(TestCase):
+    @classmethod
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def setUpTestData(cls):
+        cache.set("current_date", 1)
+
+        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,
+        )
+        cls.client = Client.objects.create(
+            login="test_client", age=15, location="Moscow", gender="FEMALE"
+        )
+        cls.click = CampaignClick.objects.create(
+            campaign=cls.campaign,
+            client=cls.client,
+            price=0.10,
+            date=1,
+        )
+
+    def test_campaign_click_creation(self):
+        self.assertIsInstance(self.click, CampaignClick)
+        self.assertEqual(self.click.price, 0.10)
+
+    def test_unique_together_constraint(self):
+        with self.assertRaises(Exception):
+            CampaignClick.objects.create(
+                campaign=self.campaign,
+                client=self.client,
+                price=0.10,
+                date=1,
+            )
@@ -0,0 +1,50 @@
+from django.test import TestCase, override_settings
+from apps.advertiser.models import Advertiser
+from apps.campaign.models import Campaign, CampaignImpression
+from apps.client.models import Client
+
+
+class CampaignImpressionModelTest(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,
+        )
+        cls.client = Client.objects.create(
+            login="test_client", age=15, location="Moscow", gender="FEMALE"
+        )
+        cls.impression = CampaignImpression.objects.create(
+            campaign=cls.campaign,
+            client=cls.client,
+            price=0.05,
+            date=1,
+        )
+
+    def test_campaign_impression_creation(self):
+        self.assertIsInstance(self.impression, CampaignImpression)
+        self.assertEqual(self.impression.price, 0.05)
+
+    def test_unique_together_constraint(self):
+        with self.assertRaises(Exception):
+            CampaignImpression.objects.create(
+                campaign=self.campaign,
+                client=self.client,
+                price=0.05,
+                date=1,
+            )
@@ -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)
@@ -0,0 +1,65 @@
+from django.test import TestCase, override_settings
+from apps.advertiser.models import Advertiser
+from apps.campaign.models import Campaign, CampaignReport
+from apps.client.models import Client
+
+
+class CampaignReportModelTest(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,
+        )
+        cls.client_instance = Client.objects.create(
+            login="test_client",
+            age=30,
+            gender="MALE",
+            location="Test Location",
+        )
+
+    def test_campaign_report_creation(self):
+        report = CampaignReport.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            state=CampaignReport.CampaignReportState.SENT,
+            message="Inappropriate content",
+            flagged_by_llm=True,
+        )
+
+        self.assertIsInstance(report, CampaignReport)
+        self.assertEqual(report.campaign, self.campaign)
+        self.assertEqual(report.client, self.client_instance)
+        self.assertEqual(report.state, CampaignReport.CampaignReportState.SENT)
+        self.assertEqual(report.message, "Inappropriate content")
+        self.assertTrue(report.flagged_by_llm)
+
+
+    def test_campaign_report_unique_together_constraint(self):
+        CampaignReport.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            state=CampaignReport.CampaignReportState.SENT,
+        )
+
+        with self.assertRaises(Exception):
+            CampaignReport.objects.create(
+                campaign=self.campaign,
+                client=self.client_instance,
+                state=CampaignReport.CampaignReportState.UNDER_REVIEW,
+            )
@@ -0,0 +1,148 @@
+from django.core.cache import cache
+from django.test import TestCase, override_settings
+from apps.advertiser.models import Advertiser
+from apps.campaign.models import Campaign, CampaignClick, CampaignImpression
+from apps.client.models import Client
+
+
+class CampaignStatisticsTest(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,
+        )
+        cls.client_instance = Client.objects.create(
+            login="test_client",
+            age=30,
+            gender="MALE",
+            location="Test Location",
+        )
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def setUp(self):
+        cache.clear()
+        cache.set("current_date", 5)
+
+    def test_get_statistics_no_data(self):
+        stats = self.campaign.get_statistics()
+        expected_stats = {
+            "impressions_count": 0,
+            "clicks_count": 0,
+            "conversion": 0,
+            "spent_impressions": 0,
+            "spent_clicks": 0,
+            "spent_total": 0,
+        }
+
+        self.assertEqual(stats, expected_stats)
+
+    def test_get_statistics_with_data(self):
+        CampaignImpression.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            price=self.campaign.cost_per_impression,
+            date=5,
+        )
+        CampaignClick.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            price=self.campaign.cost_per_click,
+            date=5,
+        )
+        stats = self.campaign.get_statistics()
+        expected_stats = {
+            "impressions_count": 1,
+            "clicks_count": 1,
+            "conversion": 100.0,
+            "spent_impressions": 0.05,
+            "spent_clicks": 0.10,
+            "spent_total": 0.15,
+        }
+
+        self.assertEqual(stats, expected_stats)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_get_daily_statistics_no_data(self):
+        daily_stats = self.campaign.get_daily_statistics()
+        expected_stats = [
+            {
+                "date": day,
+                "impressions_count": 0,
+                "clicks_count": 0,
+                "conversion": 0,
+                "spent_impressions": 0,
+                "spent_clicks": 0,
+                "spent_total": 0,
+            }
+            for day in range(
+                self.campaign.start_date, cache.get("current_date") + 1
+            )
+        ]
+
+        self.assertEqual(daily_stats, expected_stats)
+
+    @override_settings(
+        CACHES={
+            "default": {
+                "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
+            }
+        }
+    )
+    def test_get_daily_statistics_with_data(self):
+        CampaignImpression.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            price=self.campaign.cost_per_impression,
+            date=5,
+        )
+        CampaignClick.objects.create(
+            campaign=self.campaign,
+            client=self.client_instance,
+            price=self.campaign.cost_per_click,
+            date=5,
+        )
+        daily_stats = self.campaign.get_daily_statistics()
+        expected_stats = [
+            {
+                "date": day,
+                "impressions_count": 1 if day == 5 else 0,
+                "clicks_count": 1 if day == 5 else 0,
+                "conversion": 100.0 if day == 5 else 0,
+                "spent_impressions": 0.05 if day == 5 else 0,
+                "spent_clicks": 0.10 if day == 5 else 0,
+                "spent_total": 0.15 if day == 5 else 0,
+            }
+            for day in range(
+                self.campaign.start_date, cache.get("current_date") + 1
+            )
+        ]
+
+        self.assertEqual(daily_stats, expected_stats)