package app

import "testing"

func TestTechdumpBucketForPCIeLinkFiles(t *testing.T) {
	cases := map[string]string{
		"pcie-nvidia-link.txt":            "gpu",
		"pcie-nvidia-link-under-load.txt": "gpu",
		"kernel-aer-nvidia.txt":           "gpu",
	}
	for name, want := range cases {
		if got := techdumpBucketFor(name); got != want {
			t.Errorf("techdumpBucketFor(%q) = %q, want %q", name, got, want)
		}
	}
}