package utils

import (
	"testing"

	"github.com/google/go-cmp/cmp"
)

func TestParseColumns(t *testing.T) {
	tests := []struct {
		name    string
		rows    []string
		titles  []string
		want    map[string][]string
		wantErr bool
	}{

		{
			name: "vmstat's output with spaced rows",
			rows: []string{
				"r  b   swpd   free   buff  cache   si   so    bi    bo   in   cs us sy id wa st",
				"                                                                                ",
				"5  0      0 14827096      0  25608    0    0     2     5   57    2  1  0 99  0  0",
				"                                                                                ",
				"2  0      0 14827096      0  25608    0    0     0     0 1131 1594  2  1 97  0  1",
				"                                                                                ",
				"2  0      0 14827096      0  25608    0    0     0     0 5283 8037  7  3 90  0  0",
			},
			titles: []string{"us", "sy", "st"},
			want: map[string][]string{
				"us": {"1", "2", "7"},
				"sy": {"0", "1", "3"},
				"st": {"0", "1", "0"},
			},
		},
		{
			name: "empty slice",
			rows: []string{},
			want: map[string][]string{},
		},
		{
			name: "empty titles",
			rows: []string{
				"r  b   swpd   free   buff",
				"5  0      0 14827096      0",
			},
			want: map[string][]string{
				"r":    {"5"},
				"b":    {"0"},
				"swpd": {"0"},
				"free": {"14827096"},
				"buff": {"0"},
			},
		},
		{
			name: "repeated headers",
			rows: []string{
				"Device            r/s     w/s     rkB/s     wkB/s   rrqm/s   wrqm/s  %rrqm  %wrqm r_await w_await aqu-sz rareq-sz wareq-sz  svctm  %util",
				"vdb              0.01    0.58      0.86     20.79     0.00     0.19   0.24  25.02    8.44 1552.07   0.90    96.44    35.68  95.00   5.62",
				"vda              0.00    0.00      0.04      0.00     0.00     0.00   2.73   0.00    3.08    0.00   0.00    62.55     0.00   2.20   0.00",
				"																																		 ",
				"Device            r/s     w/s     rkB/s     wkB/s   rrqm/s   wrqm/s  %rrqm  %wrqm r_await w_await aqu-sz rareq-sz wareq-sz  svctm  %util",
				"																																		 ",
				"Device            r/s     w/s     rkB/s     wkB/s   rrqm/s   wrqm/s  %rrqm  %wrqm r_await w_await aqu-sz rareq-sz wareq-sz  svctm  %util",
				"																																		 ",
				"Device            r/s     w/s     rkB/s     wkB/s   rrqm/s   wrqm/s  %rrqm  %wrqm r_await w_await aqu-sz rareq-sz wareq-sz  svctm  %util",
			},
			titles: []string{"%util"},
			want: map[string][]string{
				"%util": {"5.62", "0.00"},
			},
		},
		{
			name: "missing titles on some columns",
			rows: []string{
				"              total        used        free      shared  buff/cache   available",
				"Mem:          14520          13       14481           0          25       14506",
				"Swap:             0           0           0",
			},
			titles:  []string{"total"},
			wantErr: true,
		},
		{
			name: "unknown titles",
			rows: []string{
				"r  b   swpd    free   buff   cache",
				"5  0      0 14827096     0   25608",
			},
			titles:  []string{"r", "b", "used"},
			wantErr: true,
		},
	}

	for _, test := range tests {
		got, err := ParseColumns(test.rows, test.titles...)
		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("ParseColumns(%v, %v) err %v, wantErr: %t", test.rows, test.titles, err, test.wantErr)
		}
		if diff := cmp.Diff(test.want, got); diff != "" {
			t.Errorf("Ran ParseColumns(%v, %v), but got mismatch between got and want (+got, -want): \n diff %s", test.rows, test.titles, diff)
		}
	}
}

func TestParseRows(t *testing.T) {
	tests := []struct {
		name    string
		rows    []string
		delim   string
		titles  []string
		want    map[string][]string
		wantErr bool
	}{
		{
			name: "lscpu's output with spaced rows",
			rows: []string{
				"Architecture:        x86_64",
				"							 ",
				"CPU op-mode(s):      32-bit, 64-bit",
				"Byte Order:          Little Endian",
				"								 ",
				"Address sizes:       39 bits physical, 48 bits virtual",
				"							 ",
				"CPU(s):              8",
				"On-line CPU(s) list: 0-7",
			},
			delim:  ":",
			titles: []string{"CPU(s)"},
			want: map[string][]string{
				"CPU(s)": {"8"},
			},
		},
		{
			name: "whitespace delimiter",
			rows: []string{
				"vdb               0.63         0.86        22.19     760760   19680952",
				"vda               0.00         0.04         0.00      37845          0",
			},
			delim: " ",
			want: map[string][]string{
				"vdb": {"0.63", "0.86", "22.19", "760760", "19680952"},
				"vda": {"0.00", "0.04", "0.00", "37845", "0"},
			},
		},
		{
			name: "empty slice",
			rows: []string{},
			want: map[string][]string{},
		},
		{
			name: "empty titles",
			rows: []string{
				"processor       : 0",
				"vendor_id       : GenuineIntel",
				"cpu family      : 6",
				"model           : 142",
				"model name      : 06/8e",
			},
			delim:  ":",
			titles: []string{},
			want: map[string][]string{
				"processor":  {"0"},
				"vendor_id":  {"GenuineIntel"},
				"cpu family": {"6"},
				"model":      {"142"},
				"model name": {"06/8e"},
			},
		},
		{
			name: "wrong delimiter",
			rows: []string{
				"vdb               0.63         0.86        22.30     760860   19808528",
				"vda               0.00         0.04         0.00      37845          0",
			},
			delim:   ":",
			wantErr: true,
		},
		{
			name: "unknown titles",
			rows: []string{
				"Device           tps    kB_read/s    kB_wrtn/s    kB_read    kB_wrtn",
				"vdb             0.60         1.54        21.54     836516   11703972",
				"vda             0.00         0.07         0.00      37901          0",
			},
			delim:   " ",
			titles:  []string{"sda"},
			wantErr: true,
		},
	}

	for _, test := range tests {
		got, err := ParseRows(test.rows, test.delim, test.titles...)
		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("ParseRows(%v, %v) = %v, wantErr: %t", test.rows, test.titles, err, test.wantErr)
		}
		if diff := cmp.Diff(test.want, got); diff != "" {
			t.Errorf("Ran ParseRows(%v, %v), but got mismatch between got and want (+got,-want): \n diff %s", test.rows, test.titles, diff)
		}
	}
}

func TestParseRowsAndColumns(t *testing.T) {
	tests := []struct {
		name    string
		rows    []string
		titles  []string
		want    map[string][]string
		wantErr bool
	}{
		{
			name: "free",
			rows: []string{
				"              total        used        free      shared  buff/cache   available",
				"Mem:          14520          13       14481           0          25       14506",
				"Swap:             0           0           0",
			},
			titles: []string{"Mem:used", "Mem:total", "Swap:used", "Swap:total"},
			want: map[string][]string{
				"Mem:used":   {"13"},
				"Mem:total":  {"14520"},
				"Swap:used":  {"0"},
				"Swap:total": {"0"},
			},
		},
		{
			name: "iostat with spaced rows",
			rows: []string{
				"Device             tps    kB_read/s    kB_wrtn/s    kB_read    kB_wrtn",
				"                                                                      ",
				"vdb               1.27         3.79        41.80     732408    8072028",
				"                                                                       ",
				"vda               0.00         0.20         0.00      37845          0",
			},
			titles: []string{"vdb:tps", "vda:kB_read", "vdb:kB_wrtn"},
			want: map[string][]string{
				"vdb:tps":     {"1.27"},
				"vdb:kB_wrtn": {"8072028"},
				"vda:kB_read": {"37845"},
			},
		},
		{
			name: "wrongly formatted titles",
			rows: []string{
				"Device             tps    kB_read/s    kB_wrtn/s    kB_read    kB_wrtn",
				"vdb               1.27         3.79        41.80     732408    8072028",
				"vda               0.00         0.20         0.00      37845          0",
			},
			titles:  []string{":tps"},
			wantErr: true,
		},
	}

	for _, test := range tests {
		got, err := ParseRowsAndColumns(test.rows, test.titles...)
		if gotErr := err != nil; gotErr != test.wantErr {
			t.Fatalf("ParseRowsAndColumns(%v, %v) = %v, wantErr: %t", test.rows, test.titles, err, test.wantErr)
		}
		if diff := cmp.Diff(test.want, got); diff != "" {
			t.Errorf("Ran ParseRowsAndColumns(%v, %v), but got mismatch between got and want (+got, -want): \n diff %s", test.rows, test.titles, diff)
		}
	}

}