package profiler

import (
	"testing"

	"cos.googlesource.com/cos/tools.git/src/pkg/nodeprofiler/utils"
)

func TestCollectUtilization(t *testing.T) {
	tests := []struct {
		name      string
		component Component
		outputs   map[string]utils.ParsedOutput
		want      float64
		wantErr   bool
	}{
		{
			name:      "cpu",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1", "2", "7"},
					"sy": {"0", "1", "3"},
					"st": {"0", "0", "0"},
				},
			},
			want: 6.5,
		},
		{
			name:      "vmstat slices of length 1",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1"},
					"sy": {"0"},
					"st": {"0"},
				},
			},
			wantErr: true,
		},
		{
			name:      "empty vmstat slices",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {},
					"sy": {},
					"st": {},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing titles",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1", "2", "7"},
					"sy": {"0", "1", "3"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing commands output",
			component: &CPU{"fake", &USEMetrics{}},
			outputs:   map[string]utils.ParsedOutput{},
			wantErr:   true,
		},
		{
			name:      "memory capacity",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"free": {
					"Mem:used":   {"13"},
					"Swap:used":  {"0"},
					"Mem:total":  {"14520"},
					"Swap:total": {"0"},
				},
			},
			want: 0.09,
		},
		{
			name:      "missing titles",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"free": {},
			},
			wantErr: true,
		},
		{
			name:      "missing commands output",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"si": {"0", "0", "3"},
					"so": {"0", "1", "5"},
				},
			},
			wantErr: true,
		},
		{
			name:      "storage device I/O",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"iostat": {
					"%util": {"4.76", "0.09"},
				},
			},
			want: 2.425,
		},
		{
			name:      "missing titles",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"iostat": {
					"%idle": {"94.27"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing commands output",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs:   map[string]utils.ParsedOutput{},
			wantErr:   true,
		},
		{
			name:      "storage capacity",
			component: &StorageCap{"fake", &USEMetrics{}, []string{"/dev/vda"}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Filesystem": {"/dev/vdb", "/dev/vda"},
					"Used":       {"4764604", "50448"},
					"1K-blocks":  {"7864320", "50540"},
				},
			},
			want: 99.82,
		},
		{
			name:      "devices not set",
			component: &StorageCap{"fake", &USEMetrics{}, []string{}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Filesystem": {"/dev/sda", "/dev/vda"},
					"Used":       {"4764604", "50448"},
					"1K-blocks":  {"7864320", "50540"},
				},
			},
			want: 60.59,
		},
		{
			name:      "device (sda) with different partitions",
			component: &StorageCap{"fake", &USEMetrics{}, []string{}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Filesystem": {"/dev/sda1", "/dev/sda8"},
					"Used":       {"95384", "24"},
					"1K-blocks":  {"5971884", "11756"},
				},
			},
			want: 1.59,
		},
		{
			name:      "several occurrences of same device",
			component: &StorageCap{"fake", &USEMetrics{}, []string{"tmpfs"}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Filesystem": {"tmpfs", "tmpfs", "/dev/root", "tmpfs"},
					"Used":       {"0", "468", "1051636", "168"},
					"1K-blocks":  {"1884128", "1884128", "2003760", "1024"},
				},
			},
			want: 0.02,
		},
		{
			name:      "missing [default] device",
			component: &StorageCap{"fake", &USEMetrics{}, []string{}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Filesystem": {"/dev/vdb", "/dev/vda"},
					"Used":       {"4764604", "50448"},
					"1K-blocks":  {"7864320", "50540"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing titles",
			component: &StorageCap{"fake", &USEMetrics{}, []string{"/dev/vda"}},
			outputs: map[string]utils.ParsedOutput{
				"df": {
					"Use%": {"67%", "0%", "100%", "2%"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing commands output",
			component: &StorageCap{"fake", &USEMetrics{}, []string{"/dev/vda"}},
			outputs:   map[string]utils.ParsedOutput{},
			wantErr:   true,
		},
	}
	for _, test := range tests {
		err := test.component.CollectUtilization(test.outputs)
		got := test.component.USEMetrics().Utilization
		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("CollectUtilization(%v) err %v, wantErr %t", test.outputs, err, test.wantErr)
		}
		if got != test.want {
			t.Errorf("CollectUtilization(%v) = %v, want: %v", test.outputs, test.want, got)
		}
	}
}
func TestCollectSaturation(t *testing.T) {
	tests := []struct {
		name      string
		component Component
		outputs   map[string]utils.ParsedOutput
		want      bool
		wantErr   bool
	}{
		{
			name:      "CPU",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"15", "12", "12"},
				},
				"lscpu": {
					"CPU(s)": {"8"},
				},
			},
			want: true,
		},
		{
			name:      "vmstat slices of length 1",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"15"},
				},
				"lscpu": {
					"CPU(s)": {"8"},
				},
			},
			wantErr: true,
		},
		{
			name:      "empty vmstat slices",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {},
				},
				"lscpu": {
					"CPU(s)": {"8"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing titles",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {},
				"lscpu": {
					"CPU(s):": {"8"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing commands output",
			component: &CPU{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"10", "10", "10"},
				},
			},
			wantErr: true,
		},
		{
			name:      "Memory capacity",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"si": {"0", "7", "3"},
					"so": {"0", "8", "5"},
				},
				"free": {
					"Swap:total": {"701"},
				},
			},
		},
		{
			name:      "missing commands output",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"si": {"0", "0", "3"},
					"so": {"0", "1", "5"},
				},
			},
			wantErr: true,
		},
		{
			name:      "missing titles",
			component: &MemCap{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"si": {"0", "0", "3"},
					"so": {"0", "1", "5"},
				},
				"free": {},
			},
			wantErr: true,
		},
		{
			name:      "Storage Device",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"iostat": {
					"aqu-sz": {"0.04", "0.00"},
				},
			},
		},
		{
			name:      "missing commands output",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs:   map[string]utils.ParsedOutput{},
			wantErr:   true,
		},
		{
			name:      "missing titles",
			component: &StorageDevIO{"fake", &USEMetrics{}},
			outputs: map[string]utils.ParsedOutput{
				"iostat": {},
			},
			wantErr: true,
		},
	}
	for _, test := range tests {
		err := test.component.CollectSaturation(test.outputs)
		got := test.component.USEMetrics().Saturation
		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("CollectSaturation(%v) err %v, wantErr %t", test.outputs, err, test.wantErr)
		}
		if got != test.want {
			t.Errorf("CollectSaturation(%v) = %v, want: %v", test.outputs, test.want, got)
		}
	}
}