package main

import (
	"context"
	"strings"
	"testing"
	"time"

	fleetpb "fleet-manager/proto"
)

func TestSnapshotPoint(t *testing.T) {
	at := time.Unix(1700000000, 0).UTC()
	snap := &fleetpb.HealthSnapshot{
		LoadAvg: 0.5, CpuPct: 12, TmpfsUsedPct: 63, ChromiumCrashes: 2, KioskActive: true, NvmeRoot: "ok",
	}
	p := snapshotPoint("beef01", "nqn:host-beef01", "orion-drifter-07", snap, at)

	if p.Name() != influxMeasurement {
		t.Errorf("measurement = %q", p.Name())
	}
	tags := map[string]string{}
	for _, tg := range p.TagList() {
		tags[tg.Key] = tg.Value
	}
	if tags["serial"] != "beef01" || tags["starship"] != "orion-drifter-07" || tags["hostnqn"] != "nqn:host-beef01" {
		t.Errorf("identity tags = %+v", tags)
	}
	fields := map[string]any{}
	for _, f := range p.FieldList() {
		fields[f.Key] = f.Value
	}
	if fields["tmpfs_used_pct"] != float64(63) {
		t.Errorf("tmpfs_used_pct field = %v (%T)", fields["tmpfs_used_pct"], fields["tmpfs_used_pct"])
	}
	if fields["kiosk_active"] != int64(1) {
		t.Errorf("kiosk_active = %v", fields["kiosk_active"])
	}
	if fields["chromium_crashes"] != int64(2) {
		t.Errorf("chromium_crashes = %v", fields["chromium_crashes"])
	}
	if fields["nvme_root"] != "ok" {
		t.Errorf("nvme_root = %v", fields["nvme_root"])
	}
	if !p.Time().Equal(at) {
		t.Errorf("time = %v", p.Time())
	}
}

func TestFluxQueryContentsAndQuoting(t *testing.T) {
	q := fluxQuery("fleet_health", "beef01", "tmpfs_used_pct", time.Hour)
	for _, want := range []string{
		`from(bucket:"fleet_health")`,
		`range(start: -3600s)`,
		`r._measurement == "fleet_health"`,
		`r.serial == "beef01"`,
		`r._field == "tmpfs_used_pct"`,
	} {
		if !strings.Contains(q, want) {
			t.Errorf("flux query missing %q in:\n%s", want, q)
		}
	}
}

func TestQueryFieldRejectsUnsafeInput(t *testing.T) {
	s := &influxSink{bucket: "b"} // query is never reached for these
	if _, err := s.QueryField(context.Background(), "not-hex!", "tmpfs_used_pct", time.Hour); err == nil {
		t.Error("expected rejection of a non-hex serial")
	}
	if _, err := s.QueryField(context.Background(), "beef01", "evil) |> drop(", time.Hour); err == nil {
		t.Error("expected rejection of a non-whitelisted field")
	}
}