fangke/server/tracker/tests/test_models.py

import pytest
from django.db import IntegrityError
from tracker.models import Site, Visitor, VisitSession, Event

pytestmark = pytest.mark.django_db

def test_create_site_generates_public_site_id():
    site = Site.objects.create(name="官网", domains=["www.example.com"])
    assert site.site_id.startswith("site_") and len(site.site_id) == 21
    assert site.is_active is True

def test_visitor_first_touch_fields():
    site = Site.objects.create(name="s", domains=["a.com"])
    v = Visitor.objects.create(site=site, visitor_id="vid1",
        first_referrer_domain="google.com", first_utm={"utm_source": "google"},
        first_landing_path="/p", country="美国", device_type="PC")
    assert v.visit_count == 0 and v.active_days == 0

def test_event_unique_site_event_id():
    site = Site.objects.create(name="s", domains=["a.com"])
    v = Visitor.objects.create(site=site, visitor_id="vid1")
    s = VisitSession.objects.create(site=site, visitor=v, session_id="sid1",
        started_at="2026-09-15T10:00:00Z", ended_at="2026-09-15T10:01:00Z")
    Event.objects.create(site=site, visitor=v, session=s, event_id="e1",
        event_name="page_view", occurred_at="2026-09-15T10:00:00Z", path="/")
    with pytest.raises(IntegrityError):
        Event.objects.create(site=site, visitor=v, session=s, event_id="e1",
            event_name="page_view", occurred_at="2026-09-15T10:01:00Z", path="/")

def test_site_cascade_deletes_children():
    site = Site.objects.create(name="s", domains=["a.com"])
    v = Visitor.objects.create(site=site, visitor_id="v")
    s = VisitSession.objects.create(site=site, visitor=v, session_id="s",
        started_at="2026-09-15T10:00:00Z", ended_at="2026-09-15T10:01:00Z")
    Event.objects.create(site=site, visitor=v, session=s, event_id="e",
        event_name="page_view", occurred_at="2026-09-15T10:00:00Z", path="/")
    site.delete()
    assert Visitor.objects.count() == 0 and VisitSession.objects.count() == 0 and Event.objects.count() == 0