31 lines
1.8 KiB
Python
31 lines
1.8 KiB
Python
|
from datetime import timedelta
|
|||
|
|
from django.utils import timezone
|
|||
|
|
from tracker.models import Site, Visitor, VisitSession, Event
|
|||
|
|
from tracker.services.stats import refresh_daily
|
|||
|
|
|
|||
|
|
# 用上海本地时刻做基准：refresh_daily 的日界是上海自然日（make_aware），
|
|||
|
|
# 若直接用 UTC 的 timezone.now()，在本地 00:00-08:00 之间跑测试时 NOW.date() 会错一天
|
|||
|
|
NOW = timezone.localtime(timezone.now())
|
|||
|
|
YESTERDAY = NOW - timedelta(days=1)
|
|||
|
|
|
|||
|
|
def _mk(site, vid, sid, day, n_events, stay=0):
|
|||
|
|
v, _ = Visitor.objects.get_or_create(site=site, visitor_id=vid)
|
|||
|
|
sess = VisitSession.objects.create(site=site, visitor=v, session_id=sid,
|
|||
|
|
started_at=day, ended_at=day + timedelta(minutes=5),
|
|||
|
|
stay_ms=stay, event_count=n_events)
|
|||
|
|
for i in range(n_events):
|
|||
|
|
Event.objects.create(site=site, visitor=v, session=sess, event_id=f"{vid}-{sid}-{day:%Y%m%d}-{i}",
|
|||
|
|
event_name="page_view", occurred_at=day + timedelta(seconds=i), path="/")
|
|||
|
|
|
|||
|
|
def test_refresh_daily_two_days(db):
|
|||
|
|
site = Site.objects.create(name="s", domains=["a.com"])
|
|||
|
|
_mk(site, "v1", "s1", YESTERDAY, 3) # v1 昨天 3 事件
|
|||
|
|
_mk(site, "v1", "s2", NOW, 2, stay=8000) # v1 今天 2 事件
|
|||
|
|
_mk(site, "v2", "s3", NOW, 4) # v2 今天 4 事件
|
|||
|
|
y = refresh_daily(site, YESTERDAY.date())
|
|||
|
|
t = refresh_daily(site, NOW.date())
|
|||
|
|
# new_visitors 口径 = first_seen_at 落入当日。测试里 v1 虽有昨日事件，但访客记录是测试运行时（今天）建档的
|
|||
|
|
# （auto_now_add），故昨天 new=0、今天 new=2（v1+v2）——按 first_seen_at 真实语义断言
|
|||
|
|
assert (y.pv, y.uv, y.sessions, y.new_visitors) == (3, 1, 1, 0)
|
|||
|
|
assert (t.pv, t.uv, t.sessions, t.new_visitors) == (6, 2, 2, 2) # 今天：pv=6 uv=2(v1+v2) sessions=2 new=2(v1+v2)
|