83 lines
2.8 KiB
Go
83 lines
2.8 KiB
Go
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")
|
|
}
|
|
}
|