package sensors

import (
	"encoding/json"
	"reflect"
	"testing"
)

type fake []Sensor

func (f fake) Sensors() []Sensor { return f }

func TestSeg(t *testing.T) {
	for in, want := range map[string]string{"/": "root", "/mnt/tank": "mnt_tank", "Package id 0": "package_id_0", "enp3s0": "enp3s0", "  ": "unnamed", "Tctl+": "tctl", "nvme0n1": "nvme0n1"} {
		if got := Seg(in); got != want {
			t.Errorf("Seg(%q) = %q, want %q", in, got, want)
		}
	}
}

func TestRegistryDedupes(t *testing.T) {
	r := NewRegistry(fake{{ID: "temp.nvme.composite", Value: 40.0}}, fake{{ID: "temp.nvme.composite", Value: 50.0}})
	all := r.All()
	if len(all) != 2 || all[0].ID != "temp.nvme.composite" || all[1].ID != "temp.nvme.composite_2" {
		t.Fatalf("got %+v", all)
	}
}

func TestShape(t *testing.T) {
	all := []Sensor{{ID: "cpu.usage", Value: 12.5, Unit: "%"}, {ID: "fan.a.fan1", Value: 800.0}, {ID: "fan.a.fan2", Value: 900.0}, {ID: "net.eth0.up", Value: true}}
	got, err := Shape(all, json.RawMessage(`{"cpu":"cpu.usage","fans":"fan.*","box":{"link":"net.eth0.up","gone":"ups.battery"},"list":["cpu.usage"]}`), false)
	if err != nil {
		t.Fatal(err)
	}
	want := map[string]any{
		"cpu":  12.5,
		"fans": map[string]any{"fan.a.fan1": 800.0, "fan.a.fan2": 900.0},
		"box":  map[string]any{"link": true, "gone": nil},
		"list": []any{12.5},
	}
	if !reflect.DeepEqual(got, want) {
		t.Fatalf("got %#v\nwant %#v", got, want)
	}
	full, _ := Shape(all, json.RawMessage(`"cpu.usage"`), true)
	if s, ok := full.(Sensor); !ok || s.Unit != "%" {
		t.Fatalf("full: got %#v", full)
	}
	for _, bad := range []string{`{"x": 5}`, `{"x": "fan.["}`, `not json`} {
		if _, err := Shape(all, json.RawMessage(bad), false); err == nil {
			t.Errorf("Shape(%s) should fail", bad)
		}
	}
}