package web

import (
	"encoding/json"
	"net/http/httptest"
	"testing"
)

// 地图上的城市圈气泡（/api/cities）必须「先按地点聚合、再按城市汇总」。
//
// 旧写法把聚合打在 places ⋈ visits 上，于是每个地点的 visit_count / dwell_minutes
// 会被按它的到访条数重复累加，count(*) 又被当成「地点数」——实测最大那个城市显示
// 「到访 72661 次 · 568 个地点」，真值是 568 次 / 86 个地点。这里按同一套口径手写一遍对账。
func TestCityAggCountsOncePerPlace(t *testing.T) {
	f := newMovementFixture(t)
	ctx, pg, dsID := f.ctx, f.pg, f.dsID

	rec := httptest.NewRecorder()
	f.srv.apiCityAgg(rec, httptest.NewRequest("GET", "/api/cities", nil).WithContext(ctx))
	if rec.Code != 200 {
		t.Fatalf("取城市聚合失败: code=%d body=%s", rec.Code, rec.Body.String())
	}
	var payload struct {
		Cities []struct {
			City   string `json:"city"`
			Visits int    `json:"visits"`
			Places int    `json:"places"`
			Dwell  int64  `json:"dwell"`
		} `json:"cities"`
	}
	if err := json.Unmarshal(rec.Body.Bytes(), &payload); err != nil {
		t.Fatalf("解析城市聚合失败: %v", err)
	}
	if len(payload.Cities) == 0 {
		t.Skip("夹具没有带城市的到访")
	}

	// 手写一份同一口径的对照（判据独立写，不复用 stats.SoloVisitSQL）：
	// 不含已并入 / 子段，也不含「家 / 工作」地点——与 apiCityAgg 保持一致。
	rows, err := pg.QueryContext(ctx, `SELECT p.city, count(*), count(DISTINCT p.id),
			COALESCE(sum(LEAST(COALESCE(v.duration_min,0),10080)),0)
		FROM places p JOIN visits v ON v.place_id = p.id AND v.dataset_id = p.dataset_id
		WHERE p.dataset_id = $1 AND p.city IS NOT NULL AND p.city <> '' AND p.lat IS NOT NULL
		  AND NOT EXISTS (SELECT 1 FROM entity_raw er
			WHERE er.dataset_id=v.dataset_id AND er.entity='ZVISIT' AND er.src_pk=v.src_pk
			  AND (er.raw->>'ZMERGEDTO' IS NOT NULL OR er.raw->>'ZPARENT' IS NOT NULL))
		  AND NOT EXISTS (SELECT 1 FROM visits vh WHERE vh.place_id = p.id
			AND (vh.is_home OR vh.is_work))
		GROUP BY p.city`, dsID)
	if err != nil {
		t.Fatal(err)
	}
	defer rows.Close()
	type want struct {
		visits, places int
		dwell          int64
	}
	truth := map[string]want{}
	for rows.Next() {
		var city string
		var w want
		if err := rows.Scan(&city, &w.visits, &w.places, &w.dwell); err != nil {
			t.Fatal(err)
		}
		truth[city] = w
	}
	if err := rows.Err(); err != nil {
		t.Fatal(err)
	}

	for _, c := range payload.Cities {
		w, ok := truth[c.City]
		if !ok {
			t.Errorf("城市「%s」不该出现（手写 SQL 里没有）", c.City)
			continue
		}
		if c.Visits != w.visits || c.Places != w.places || c.Dwell != w.dwell {
			t.Errorf("城市「%s」聚合对不上：到访 %d/%d、地点 %d/%d、停留 %d/%d",
				c.City, c.Visits, w.visits, c.Places, w.places, c.Dwell, w.dwell)
		}
	}
	// 保底的不变量：地点数不可能多于到访数（旧写法里这两个数就换位了）
	for _, c := range payload.Cities {
		if c.Places > c.Visits {
			t.Errorf("城市「%s」的地点数 %d 多于到访数 %d，说明又按到访条数重复累加了", c.City, c.Places, c.Visits)
		}
	}
}