package main

import (
	"compress/gzip"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"testing"
	"time"
)

func TestCityInstallRejectsBadDownload(t *testing.T) {
	dir := t.TempDir()
	t.Setenv("DASHBOARD_DB_PATH", filepath.Join(dir, "dashboard.db"))
	existing := []byte("existing database")
	if err := os.WriteFile(trafficCityPath(), existing, 0600); err != nil {
		t.Fatal(err)
	}
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		g := gzip.NewWriter(w)
		g.Write([]byte("corrupt database"))
		g.Close()
	}))
	defer server.Close()
	if err := installTrafficCityFrom(server.URL, cityDatabaseSHA1); err == nil {
		t.Fatal("bad checksum accepted")
	}
	got, err := os.ReadFile(trafficCityPath())
	if err != nil || string(got) != string(existing) {
		t.Fatal("existing database replaced on failure")
	}
	files, _ := filepath.Glob(filepath.Join(dir, ".city-*"))
	if len(files) != 0 {
		t.Fatal("partial file left behind")
	}
}
func TestCityDatabaseDownloaded(t *testing.T) {
	source := os.Getenv("DASHBOARD_CITY_TEST_GZIP")
	if source == "" {
		t.Skip("optional official database integration test")
	}
	t.Setenv("DASHBOARD_DB_PATH", filepath.Join(t.TempDir(), "dashboard.db"))
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.ServeFile(w, r, source) }))
	defer server.Close()
	trafficCityReady()
	if err := installTrafficCityFrom(server.URL, cityDatabaseSHA1); err != nil {
		t.Fatal(err)
	}
	g := lookupTrafficCityLocation("20.194.96.114")
	if !g.Located || g.Latitude < 30 || g.Latitude > 40 || g.Longitude < 120 || g.Longitude > 130 {
		t.Fatalf("invalid Seoul coordinates: %+v", g)
	}
	s, a := trafficFixture(t)
	s.SaveTrafficSettings(TrafficSettings{Enabled: true, AgentID: a.ID})
	now := time.Now().Unix()
	if err := s.SaveTraffic(a.ID, TrafficBatch{Events: []TrafficEvent{
		{ID: "1", Time: now, Kind: "request", IP: "20.194.96.114", Domain: "a.test"},
		{ID: "2", Time: now, Kind: "request", IP: "20.194.96.114", Domain: "a.test"},
		{ID: "3", Time: now, Kind: "alert", IP: "20.194.96.114", Domain: "b.test"},
		{ID: "4", Time: now, Kind: "request", IP: "192.168.0.10"},
	}}); err != nil {
		t.Fatal(err)
	}
	view, err := s.TrafficView("", "", false)
	if err != nil || len(view.Cities) != 1 {
		t.Fatalf("cities %+v err %v", view.Cities, err)
	}
	c := view.Cities[0]
	if c.Requests != 2 || c.UniqueIPs != 1 || c.Alerts != 1 || c.City != g.City {
		t.Fatalf("bad aggregation %+v", c)
	}
	filtered, err := s.TrafficView("b.test", "alert", false)
	if err != nil || len(filtered.Cities) != 1 || filtered.Cities[0].Requests != 0 || filtered.Cities[0].Alerts != 1 {
		t.Fatalf("city filter %+v %v", filtered.Cities, err)
	}
	city, region := lookupTrafficCity("20.194.96.114")
	t.Logf("city=%q region=%q", city, region)
	if city == "" {
		t.Fatal("public IP city not decoded")
	}
}