package formatter

import (
	"bytes"
	"testing"
)

func TestFriendly_printStatistics(t *testing.T) {
	tests := []struct {
		name     string
		stats    map[string]int
		expected string
	}{
		{
			name:     "no stats",
			stats:    map[string]int{},
			expected: "",
		},
		{
			name:     "nil stats",
			stats:    nil,
			expected: "",
		},
		{
			name:     "single stat",
			stats:    map[string]int{"rule1": 1},
			expected: "Warnings:\n  1  rule1  \n\n",
		},
		{
			name:     "multiple stats sorted by failures desc",
			stats:    map[string]int{"rule2": 2, "rule1": 1, "rule3": 3},
			expected: "Warnings:\n  3  rule3  \n  2  rule2  \n  1  rule1  \n\n",
		},
		{
			name:     "multiple stats with different length sorted by failures desc",
			stats:    map[string]int{"rule2": 2, "rule1": 1, "rule3": 3, "rule100": 40},
			expected: "Warnings:\n  40  rule100  \n  3   rule3    \n  2   rule2    \n  1   rule1    \n\n",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			f := &Friendly{}
			var buf bytes.Buffer
			f.printStatistics(&buf, "Warnings:", tt.stats)
			if got := buf.String(); got != tt.expected {
				t.Errorf("got %q, want %q", got, tt.expected)
			}
		})
	}
}

func TestFriendly_table(t *testing.T) {
	tests := []struct {
		name     string
		input    [][]string
		expected string
	}{
		{
			name:     "empty input",
			input:    [][]string{},
			expected: "",
		},
		{
			name:     "single row",
			input:    [][]string{{"1", "2", "3"}},
			expected: "  1  2  3  \n",
		},
		{
			name:     "multiple rows",
			input:    [][]string{{"1", "2", "3"}, {"4", "5", "6"}},
			expected: "  1  2  3  \n  4  5  6  \n",
		},
		{
			name:     "multiple rows with different column lengths",
			input:    [][]string{{"1", "22", "3"}, {"4", "5", "6"}},
			expected: "  1  22  3  \n  4  5   6  \n",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := table(tt.input)
			if got != tt.expected {
				t.Errorf("got %q, want %q", got, tt.expected)
			}
		})
	}
}