package platform

import "testing"

func TestInferLiveBootKind(t *testing.T) {
	tests := []struct {
		name       string
		fsType     string
		source     string
		deviceType string
		transport  string
		want       string
	}{
		{name: "ram tmpfs", fsType: "tmpfs", source: "/dev/shm/bee-live", want: "ram"},
		{name: "usb disk", source: "/dev/sdb1", deviceType: "disk", transport: "usb", want: "usb"},
		{name: "cdrom rom", source: "/dev/sr0", deviceType: "rom", want: "cdrom"},
		{name: "disk sata", source: "/dev/nvme0n1p1", deviceType: "disk", transport: "nvme", want: "disk"},
		{name: "unknown", source: "overlay", want: "unknown"},
	}
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			got := inferLiveBootKind(tc.fsType, tc.source, tc.deviceType, tc.transport)
			if got != tc.want {
				t.Fatalf("inferLiveBootKind(%q,%q,%q,%q)=%q want %q", tc.fsType, tc.source, tc.deviceType, tc.transport, got, tc.want)
			}
		})
	}
}