package main

import "testing"

func TestScrubTopLevel(t *testing.T) {
	m := map[string]any{
		"build_meta": map[string]any{"tag": "v1"},
		"other":      "keep",
	}
	scrub(m, []string{"build_meta"})
	if _, ok := m["build_meta"]; ok {
		t.Error("expected build_meta to be removed")
	}
	if m["other"] != "keep" {
		t.Error("expected other to be preserved")
	}
}

func TestScrubNested(t *testing.T) {
	m := map[string]any{
		"render_time": map[string]any{
			"total":     "0.123",
			"breakdown": "fetch:0.1s",
		},
		"other": "keep",
	}
	scrub(m, []string{"render_time.total"})
	rt, ok := m["render_time"].(map[string]any)
	if !ok {
		t.Fatal("render_time should still be present")
	}
	if _, ok := rt["total"]; ok {
		t.Error("expected render_time.total to be removed")
	}
	if rt["breakdown"] != "fetch:0.1s" {
		t.Error("expected render_time.breakdown to be preserved")
	}
	if m["other"] != "keep" {
		t.Error("expected other to be preserved")
	}
}

func TestScrubMissingPath(t *testing.T) {
	m := map[string]any{"foo": "bar"}
	scrub(m, []string{"nonexistent", "render_time.total"})
	if m["foo"] != "bar" {
		t.Error("expected foo to be preserved")
	}
}

func TestScrubNestedParentNotMap(t *testing.T) {
	m := map[string]any{"render_time": "not-a-map"}
	scrub(m, []string{"render_time.total"})
	if m["render_time"] != "not-a-map" {
		t.Error("expected render_time to be unchanged when it is not a map")
	}
}