package profiler

import (
	"testing"

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

func TestCollectUtilization(t *testing.T) {
	tests := []struct {
		name    string
		outputs map[string]utils.ParsedOutput
		want    float64
		wantErr bool
	}{
		{
			name: "simple",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1", "2", "7"},
					"sy": {"0", "1", "3"},
					"st": {"0", "0", "0"},
				},
			},
			want: 4.67,
		},
		{
			name: "missing columns",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1", "2", "7"},
					"sy": {"0", "1", "3"},
				},
			},
			wantErr: true,
		},
		{
			name:    "empty outputs",
			outputs: map[string]utils.ParsedOutput{},
			wantErr: true,
		},
	}

	for _, test := range tests {
		fakeCPU := &CPU{name: "fakeCPU", metrics: &USEMetrics{}}
		err := fakeCPU.CollectUtilization(test.outputs)
		got := fakeCPU.metrics.Utilization

		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("CollectUtilization(%v) err %q, wantErr %v", 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
		outputs map[string]utils.ParsedOutput
		want    bool
		wantErr bool
	}{
		{
			name: "non-saturated",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"5", "2", "2"},
				},
				"lscpu": {
					"CPU(s):": {"8"},
				},
			},
		},
		{
			name: "saturated",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"10", "10", "10"},
				},
				"lscpu": {
					"CPU(s):": {"8"},
				},
			},
			want: true,
		},
		{
			name: "wrong columns",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"us": {"1", "2", "7"},
					"sy": {"0", "1", "3"},
					"st": {"0", "0", "0"},
				},
				"lscpu": {
					"CPU(s):": {"8"},
				},
			},
			wantErr: true,
		},
		{
			name: "missing lscpu",
			outputs: map[string]utils.ParsedOutput{
				"vmstat": {
					"r": {"10", "10", "10"},
				},
			},
			wantErr: true,
		},
		{
			name:    "empty outputs",
			outputs: map[string]utils.ParsedOutput{},
			wantErr: true,
		},
	}

	for _, test := range tests {
		fakeCPU := &CPU{name: "fakeCPU", metrics: &USEMetrics{}}
		err := fakeCPU.CollectSaturation(test.outputs)
		got := fakeCPU.metrics.Saturation

		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("CollectSaturation(%v) err %q, wantErr %v", test.outputs, err, test.wantErr)
		}
		if got != test.want {
			t.Errorf("CollectSaturation(%v) = %v, want: %v", test.outputs, test.want, got)
		}
	}
}