refactor: reformatted all files

solution/services/backend/apps/campaign/tests/test_campaign_click_model.py

@@ -1,8 +1,10 @@
-from django.test import TestCase, override_settings
 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
 from apps.client.models import Client
+from config.errors import ConflictError
 
 
 class CampaignClickModelTest(TestCase):
@@ -14,7 +16,7 @@ class CampaignClickModelTest(TestCase):
             }
         }
     )
-    def setUpTestData(cls):
+    def setUpTestData(cls) -> None:
         cache.set("current_date", 1)
 
         cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
@@ -29,25 +31,25 @@ class CampaignClickModelTest(TestCase):
             start_date=1,
             end_date=10,
         )
-        cls.client = Client.objects.create(
+        cls.client_instance = Client.objects.create(
             login="test_client", age=15, location="Moscow", gender="FEMALE"
         )
         cls.click = CampaignClick.objects.create(
             campaign=cls.campaign,
-            client=cls.client,
+            client=cls.client_instance,
             price=0.10,
             date=1,
         )
 
-    def test_campaign_click_creation(self):
+    def test_campaign_click_creation(self) -> None:
         self.assertIsInstance(self.click, CampaignClick)
         self.assertEqual(self.click.price, 0.10)
 
-    def test_unique_together_constraint(self):
-        with self.assertRaises(Exception):
+    def test_unique_together_constraint(self) -> None:
+        with self.assertRaises(ConflictError):
             CampaignClick.objects.create(
                 campaign=self.campaign,
-                client=self.client,
+                client=self.client_instance,
                 price=0.10,
                 date=1,
             )

solution/services/backend/apps/campaign/tests/test_campaign_impression_model.py

@@ -1,7 +1,9 @@
 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
+from config.errors import ConflictError
 
 
 class CampaignImpressionModelTest(TestCase):
@@ -13,7 +15,7 @@ class CampaignImpressionModelTest(TestCase):
             }
         }
     )
-    def setUpTestData(cls):
+    def setUpTestData(cls) -> None:
         cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
         cls.campaign = Campaign.objects.create(
             advertiser=cls.advertiser,
@@ -26,25 +28,25 @@ class CampaignImpressionModelTest(TestCase):
             start_date=1,
             end_date=10,
         )
-        cls.client = Client.objects.create(
+        cls.client_instance = Client.objects.create(
             login="test_client", age=15, location="Moscow", gender="FEMALE"
         )
         cls.impression = CampaignImpression.objects.create(
             campaign=cls.campaign,
-            client=cls.client,
+            client=cls.client_instance,
             price=0.05,
             date=1,
         )
 
-    def test_campaign_impression_creation(self):
+    def test_campaign_impression_creation(self) -> None:
         self.assertIsInstance(self.impression, CampaignImpression)
         self.assertEqual(self.impression.price, 0.05)
 
-    def test_unique_together_constraint(self):
-        with self.assertRaises(Exception):
+    def test_unique_together_constraint(self) -> None:
+        with self.assertRaises(ConflictError):
             CampaignImpression.objects.create(
                 campaign=self.campaign,
-                client=self.client,
+                client=self.client_instance,
                 price=0.05,
                 date=1,
             )

solution/services/backend/apps/campaign/tests/test_campaign_model.py

@@ -1,7 +1,9 @@
 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
@@ -16,7 +18,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def setUpTestData(cls):
+    def setUpTestData(cls) -> None:
         cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
         cls.campaign = Campaign.objects.create(
             advertiser=cls.advertiser,
@@ -30,20 +32,20 @@ class CampaignModelTest(TestCase):
             end_date=10,
         )
 
-    def test_campaign_creation(self):
+    def test_campaign_creation(self) -> None:
         self.assertIsInstance(self.campaign, Campaign)
         self.assertEqual(self.campaign.ad_title, "Test Campaign")
 
-    def test_campaign_str_method(self):
+    def test_campaign_str_method(self) -> None:
         self.assertEqual(str(self.campaign), "Test Campaign")
 
-    def test_campaign_id_property(self):
+    def test_campaign_id_property(self) -> None:
         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):
+    def test_ad_id_property(self) -> None:
         self.assertEqual(self.campaign.ad_id, self.campaign.id)
 
     @override_settings(
@@ -53,7 +55,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def test_started_property(self):
+    def test_started_property(self) -> None:
         cache.set("current_date", 5)
         self.assertTrue(self.campaign.started)
         cache.set("current_date", 0)
@@ -66,7 +68,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def test_active_property(self):
+    def test_active_property(self) -> None:
         cache.set("current_date", 5)
         self.assertTrue(self.campaign.active)
         cache.set("current_date", 11)
@@ -80,7 +82,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def test_clean_method(self):
+    def test_clean_method(self) -> None:
         self.campaign.start_date = -1
 
         with self.assertRaises(ValidationError):
@@ -93,7 +95,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def test_view_method(self):
+    def test_view_method(self) -> None:
         client = Client.objects.create(
             login="test_client", age=15, location="Moscow", gender="FEMALE"
         )
@@ -108,7 +110,7 @@ class CampaignModelTest(TestCase):
             }
         }
     )
-    def test_click_method(self):
+    def test_click_method(self) -> None:
         client = Client.objects.create(
             login="test_client", age=15, location="Moscow", gender="FEMALE"
         )

solution/services/backend/apps/campaign/tests/test_campaign_report_model.py

@@ -1,7 +1,9 @@
 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
+from config.errors import ConflictError
 
 
 class CampaignReportModelTest(TestCase):
@@ -13,7 +15,7 @@ class CampaignReportModelTest(TestCase):
             }
         }
     )
-    def setUpTestData(cls):
+    def setUpTestData(cls) -> None:
         cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
         cls.campaign = Campaign.objects.create(
             advertiser=cls.advertiser,
@@ -33,7 +35,7 @@ class CampaignReportModelTest(TestCase):
             location="Test Location",
         )
 
-    def test_campaign_report_creation(self):
+    def test_campaign_report_creation(self) -> None:
         report = CampaignReport.objects.create(
             campaign=self.campaign,
             client=self.client_instance,
@@ -49,15 +51,14 @@ class CampaignReportModelTest(TestCase):
         self.assertEqual(report.message, "Inappropriate content")
         self.assertTrue(report.flagged_by_llm)
 
-
-    def test_campaign_report_unique_together_constraint(self):
+    def test_campaign_report_unique_together_constraint(self) -> None:
         CampaignReport.objects.create(
             campaign=self.campaign,
             client=self.client_instance,
             state=CampaignReport.CampaignReportState.SENT,
         )
 
-        with self.assertRaises(Exception):
+        with self.assertRaises(ConflictError):
             CampaignReport.objects.create(
                 campaign=self.campaign,
                 client=self.client_instance,

solution/services/backend/apps/campaign/tests/test_campaign_statistics.py

@@ -1,5 +1,6 @@
 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
@@ -14,7 +15,7 @@ class CampaignStatisticsTest(TestCase):
             }
         }
     )
-    def setUpTestData(cls):
+    def setUpTestData(cls) -> None:
         cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
         cls.campaign = Campaign.objects.create(
             advertiser=cls.advertiser,
@@ -41,11 +42,11 @@ class CampaignStatisticsTest(TestCase):
             }
         }
     )
-    def setUp(self):
+    def setUp(self) -> None:
         cache.clear()
         cache.set("current_date", 5)
 
-    def test_get_statistics_no_data(self):
+    def test_get_statistics_no_data(self) -> None:
         stats = self.campaign.get_statistics()
         expected_stats = {
             "impressions_count": 0,
@@ -58,7 +59,7 @@ class CampaignStatisticsTest(TestCase):
 
         self.assertEqual(stats, expected_stats)
 
-    def test_get_statistics_with_data(self):
+    def test_get_statistics_with_data(self) -> None:
         CampaignImpression.objects.create(
             campaign=self.campaign,
             client=self.client_instance,
@@ -90,7 +91,7 @@ class CampaignStatisticsTest(TestCase):
             }
         }
     )
-    def test_get_daily_statistics_no_data(self):
+    def test_get_daily_statistics_no_data(self) -> None:
         daily_stats = self.campaign.get_daily_statistics()
         expected_stats = [
             {
@@ -116,7 +117,7 @@ class CampaignStatisticsTest(TestCase):
             }
         }
     )
-    def test_get_daily_statistics_with_data(self):
+    def test_get_daily_statistics_with_data(self) -> None:
         CampaignImpression.objects.create(
             campaign=self.campaign,
             client=self.client_instance,