package main

import (
	"fmt"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestGame_scoreWordsByCommonLetterLocations(t *testing.T) {
	type args struct {
		words []string
	}
	tests := []struct {
		name string
		args args
		want map[string]int
	}{
		{
			name: "no letters in common",
			args: args{words: []string{"aaa", "bbb"}},
			want: map[string]int{"aaa": 3, "bbb": 3},
		},
		{
			name: "one letter in common",
			args: args{words: []string{"aaa", "abb"}},
			want: map[string]int{"aaa": 4, "abb": 4},
		},
		{
			name: "three Words with some letters in common",
			args: args{words: []string{"aaa", "abb", "aab"}},
			want: map[string]int{"aaa": 6, "abb": 6, "aab": 7},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			game := NewGame()
			for _, word := range tt.args.words {
				game.Words[word] = &Word{Word: word}
			}

			err := game.scoreWordsByCommonLetterLocations()
			if err != nil {
				t.Errorf("Error scoring Words: %s\n", err)
				t.FailNow()
			}

			for word, score := range tt.want {
				if game.Words[word].Score != score {
					t.Errorf("Word %s has score %d, expected %d", word, game.Words[word].Score, score)
				}
			}
		})
	}
}

func TestGame_FilterWords(t *testing.T) {
	type fields struct {
		Words  map[string]*Word
		Length int
	}
	type args struct {
		guess string
		score int
	}
	tests := []struct {
		name   string
		fields fields
		args   args
		want   map[string]*Word
	}{
		{
			name: "no words filtered with score = 0",
			fields: fields{
				Words: map[string]*Word{
					"aaa": {Word: "aaa"},
					"bbb": {Word: "bbb"},
				},
			},
			args: args{guess: "ccc", score: 0},
			want: map[string]*Word{
				"aaa": {Word: "aaa"},
				"bbb": {Word: "bbb"},
			},
		},
		{
			name: "no words filtered with score = 1",
			fields: fields{
				Words: map[string]*Word{
					"aac": {Word: "aac"},
					"bbc": {Word: "bbc"},
				},
			},
			args: args{guess: "ccc", score: 1},
			want: map[string]*Word{
				"aac": {Word: "aac"},
				"bbc": {Word: "bbc"},
			},
		},
		{
			name: "some words filtered",
			fields: fields{
				Words: map[string]*Word{
					"aac": {Word: "aac"},
					"bbc": {Word: "bbc"},
					"ccc": {Word: "ccc"},
				},
			},
			args: args{guess: "ccc", score: 1},
			want: map[string]*Word{
				"aac": {Word: "aac"},
				"bbc": {Word: "bbc"},
			},
		},
		{
			name: "real world case",
			fields: fields{
				Words: map[string]*Word{
					"pleads": {Word: "pleads", Score: 13},
					"crimes": {Word: "crimes", Score: 16},
					"fierce": {Word: "fierce", Score: 12},
					"wagons": {Word: "wagons", Score: 14},
					"shiner": {Word: "shiner", Score: 14},
					"wastes": {Word: "wastes", Score: 20},
					"tables": {Word: "tables", Score: 17},
					"ripped": {Word: "ripped", Score: 14},
					"silver": {Word: "silver", Score: 16},
					"insane": {Word: "insane", Score: 13},
					"visage": {Word: "visage", Score: 15},
				},
			},
			args: args{guess: "wastes", score: 1},
			want: map[string]*Word{
				"pleads": {Word: "pleads", Score: 13},
				"insane": {Word: "insane", Score: 13},
				"shiner": {Word: "shiner", Score: 14},
				"visage": {Word: "visage", Score: 15},
				"silver": {Word: "silver", Score: 16},
				"ripped": {Word: "ripped", Score: 14},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			g := Game{
				Words:  tt.fields.Words,
				Length: tt.fields.Length,
			}
			g.FilterWords(tt.args.guess, tt.args.score)

			fmt.Printf("Want: %v\n", tt.want)
			fmt.Printf("Got: %v\n", g.Words)

			for word := range tt.want {
				if _, ok := g.Words[word]; !ok {
					t.Errorf("Word %s was removed, expected it to remain", word)
				}
			}

			assert.Equal(t, len(tt.want), len(g.Words), "Wrong number of matches")

		})
	}
}

func TestGame_getBestGuess(t *testing.T) {
	type fields struct {
		Words  map[string]*Word
		Length int
	}
	tests := []struct {
		name    string
		fields  fields
		want    string
		wantErr bool
	}{
		{
			name: "simple case",
			fields: fields{
				Words: map[string]*Word{
					"aaa": {Word: "aaa", Score: 3},
					"aab": {Word: "aab", Score: 2},
					"caa": {Word: "caa", Score: 2},
				},
			},
			want: "aaa",
		},
		{
			name: "real-world case",
			fields: fields{
				Words: map[string]*Word{
					"pleads": {Word: "pleads", Score: 13},
					"crimes": {Word: "crimes", Score: 16},
					"fierce": {Word: "fierce", Score: 12},
					"wagons": {Word: "wagons", Score: 14},
					"shiner": {Word: "shiner", Score: 14},
					"wastes": {Word: "wastes", Score: 20},
					"tables": {Word: "tables", Score: 17},
					"ripped": {Word: "ripped", Score: 14},
					"silver": {Word: "silver", Score: 16},
					"visage": {Word: "visage", Score: 15},
					"insane": {Word: "insane", Score: 13},
				},
			},
			want: "wastes",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			g := Game{
				Words:  tt.fields.Words,
				Length: tt.fields.Length,
			}
			got, err := g.getBestGuess()
			if (err != nil) != tt.wantErr {
				t.Errorf("getBestGuess() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != tt.want {
				t.Errorf("getBestGuess() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestGame_getSortedScores(t *testing.T) {
	type fields struct {
		Words map[string]*Word
	}
	tests := []struct {
		name   string
		fields fields
		want   []string
	}{
		{
			name: "sort words by scores",
			fields: fields{
				Words: map[string]*Word{
					"pleads": {Word: "pleads", Score: 13},
					"crimes": {Word: "crimes", Score: 16},
					"fierce": {Word: "fierce", Score: 12},
					"shiner": {Word: "shiner", Score: 14},
					"wastes": {Word: "wastes", Score: 20},
					"tables": {Word: "tables", Score: 17},
					"visage": {Word: "visage", Score: 15},
				},
			},
			want: []string{"wastes", "tables", "crimes", "visage", "shiner", "pleads", "fierce"},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			g := Game{
				Words: tt.fields.Words,
			}
			assert.Equalf(t, tt.want, g.getSortedScores(), "getSortedScores()")
		})
	}
}