inital commit

marketing/tests.py

@@ -0,0 +1,91 @@
+from datetime import timedelta
+
+from django.contrib.auth import get_user_model
+from django.utils import timezone
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+from accounts.models import VendorProfile
+from booking.models import Booking
+from equipment.models import EquipmentCategory, EquipmentItem
+from marketing.models import ListingClick
+
+User = get_user_model()
+
+
+class MarketingTrackingTests(APITestCase):
+    def setUp(self):
+        self.vendor_user = User.objects.create_user(
+            email="vendor-mkt@example.com",
+            password="Pass123456!",
+            is_vendor=True,
+            is_customer=False,
+        )
+        self.vendor_profile = VendorProfile.objects.create(user=self.vendor_user, business_name="Mkt Vendor")
+        self.customer = User.objects.create_user(
+            email="customer-mkt@example.com",
+            password="Pass123456!",
+            is_vendor=False,
+            is_customer=True,
+        )
+        category = EquipmentCategory.objects.create(name="Kayak", slug="kayak")
+        self.item = EquipmentItem.objects.create(
+            vendor=self.vendor_profile,
+            category=category,
+            title="Kayak 1",
+            public_id="kayak-mkt-001",
+            price_per_day="50.00",
+            is_active=True,
+        )
+
+    def test_equipment_detail_logs_click_and_returns_ids(self):
+        res = self.client.get(f"/api/v1/equipment/items/{self.item.public_id}/")
+        self.assertEqual(res.status_code, status.HTTP_200_OK)
+        self.assertIn("marketing_click_id", res.data)
+        self.assertIn("click_traffic_type", res.data)
+        self.assertEqual(res.data["click_traffic_type"], ListingClick.TrafficType.ORGANIC)
+        self.assertEqual(ListingClick.objects.count(), 1)
+
+    def test_equipment_detail_utm_classified_marketing(self):
+        res = self.client.get(
+            f"/api/v1/equipment/items/{self.item.public_id}/",
+            {"utm_source": "google", "utm_medium": "cpc", "utm_campaign": "spring"},
+        )
+        self.assertEqual(res.status_code, status.HTTP_200_OK)
+        self.assertEqual(res.data["click_traffic_type"], ListingClick.TrafficType.MARKETING)
+        click = ListingClick.objects.get(pk=res.data["marketing_click_id"])
+        self.assertEqual(click.utm_campaign, "spring")
+
+    def test_track_click_endpoint(self):
+        res = self.client.post(
+            "/api/v1/marketing/track/click/",
+            {"listing_type": "equipment", "public_id": self.item.public_id, "utm_campaign": "email"},
+            format="json",
+        )
+        self.assertEqual(res.status_code, status.HTTP_201_CREATED)
+        self.assertIn("marketing_click_id", res.data)
+
+    def test_booking_with_marketing_click_id(self):
+        detail = self.client.get(f"/api/v1/equipment/items/{self.item.public_id}/")
+        click_id = detail.data["marketing_click_id"]
+        starts = timezone.now() + timedelta(days=10)
+        ends = starts + timedelta(days=1)
+        self.client.force_authenticate(self.customer)
+        res = self.client.post(
+            "/api/v1/booking/bookings/request/",
+            {
+                "equipment_item_id": self.item.id,
+                "starts_at": starts.isoformat(),
+                "ends_at": ends.isoformat(),
+                "marketing_click_id": click_id,
+            },
+            format="json",
+        )
+        self.assertEqual(res.status_code, status.HTTP_201_CREATED)
+        booking = Booking.objects.get(pk=res.data["id"])
+        self.assertEqual(booking.listing_click_id, click_id)
+
+    def test_vendor_summary_requires_range(self):
+        self.client.force_authenticate(self.vendor_user)
+        res = self.client.get("/api/v1/marketing/vendor/summary/")
+        self.assertEqual(res.status_code, status.HTTP_400_BAD_REQUEST)