console_test.go

package console

import (
	"bytes"
	"io"
	"os"
	"testing"
)

type TestUser struct {
	ID   int    `json:"id"`
	Name string `json:"name"`
}

func captureOutput(f func()) string {
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	f()

	w.Close()
	os.Stdout = old

	var buf bytes.Buffer
	io.Copy(&buf, r)
	return buf.String()
}

func TestLog(t *testing.T) {
	tests := []struct {
		name  string
		input interface{}
	}{
		{"string", "hello world"},
		{"integer", 42},
		{"float", 3.14},
		{"boolean true", true},
		{"boolean false", false},
		{"nil", nil},
		{"struct", TestUser{ID: 1, Name: "John"}},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			output := captureOutput(func() {
				Log(tt.input)
			})

			if output == "" {
				t.Errorf("Log() produced no output for %v", tt.input)
			}
		})
	}
}

func TestTable(t *testing.T) {
	tests := []struct {
		name  string
		input interface{}
	}{
		{
			"slice of structs",
			[]TestUser{
				{ID: 1, Name: "John"},
				{ID: 2, Name: "Jane"},
			},
		},
		{
			"slice of integers",
			[]int{1, 2, 3, 4, 5},
		},
		{
			"empty slice",
			[]TestUser{},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			output := captureOutput(func() {
				Table(tt.input)
			})

			if output == "" {
				t.Errorf("Table() produced no output for %v", tt.input)
			}
		})
	}
}

func TestTableWithNonSlice(t *testing.T) {
	output := captureOutput(func() {
		Table("not a slice")
	})

	if output == "" {
		t.Error("Table() should produce error output for non-slice input")
	}
}

func TestLogLevels(t *testing.T) {
	tests := []struct {
		name string
		fn   func(interface{})
	}{
		{"Info", Info},
		{"Warn", Warn},
		{"Error", Error},
		{"Success", Success},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			output := captureOutput(func() {
				tt.fn("test message")
			})

			if output == "" {
				t.Errorf("%s() produced no output", tt.name)
			}
		})
	}
}

func TestDebug(t *testing.T) {
	output := captureOutput(func() {
		Debug("test label", map[string]string{"key": "value"})
	})

	if output == "" {
		t.Error("Debug() produced no output")
	}
}

func BenchmarkLog(b *testing.B) {
	user := TestUser{ID: 1, Name: "John Doe"}

	for i := 0; i < b.N; i++ {
		captureOutput(func() {
			Log(user)
		})
	}
}

func BenchmarkTable(b *testing.B) {
	users := []TestUser{
		{ID: 1, Name: "John"},
		{ID: 2, Name: "Jane"},
		{ID: 3, Name: "Bob"},
	}

	for i := 0; i < b.N; i++ {
		captureOutput(func() {
			Table(users)
		})
	}
}