package main

import (
	"testing"

	"github.com/posener/complete"
	"os"
)

func TestPredictions(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name       string
		predictor  complete.Predictor
		last       string
		completion []string
	}{
		{
			name:       "predict tests ok",
			predictor:  predictTest,
			completion: []string{"TestPredictions", "Example"},
		},
		{
			name:      "predict tests not found",
			predictor: predictTest,
			last:      "X",
		},
		{
			name:       "predict benchmark ok",
			predictor:  predictBenchmark,
			completion: []string{"BenchmarkFake"},
		},
		{
			name:      "predict benchmarks not found",
			predictor: predictBenchmark,
			last:      "X",
		},
		{
			name:       "predict packages ok",
			predictor:  complete.PredictFunc(predictPackages),
			completion: []string{"./"},
		},
		{
			name:      "predict packages not found",
			predictor: complete.PredictFunc(predictPackages),
			last:      "X",
		},
		{
			name:       "predict runnable ok",
			predictor:  complete.PredictFunc(predictRunnableFiles),
			completion: []string{"./complete.go"},
		},
		{
			name:      "predict runnable not found",
			predictor: complete.PredictFunc(predictRunnableFiles),
			last:      "X",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			a := complete.Args{Last: tt.last}
			got := tt.predictor.Predict(a)
			if want := tt.completion; !equal(got, want) {
				t.Errorf("Failed %s: completion = %q, want %q", t.Name(), got, want)
			}
		})
	}
}

func BenchmarkFake(b *testing.B) {}

func Example() {
	os.Setenv("COMP_LINE", "go ru")
	main()
	// output: run

}

func equal(s1, s2 []string) bool {
	if len(s1) != len(s2) {
		return false
	}
	for i := range s1 {
		if s1[i] != s2[i] {
			return false
		}
	}
	return true
}