package table

import (
	"bytes"
	"testing"
)

func TestCSVHeaders(t *testing.T) {
	buffer := new(bytes.Buffer)
	printer := &CSVPrinter{writer: buffer}

	err := printer.Headers("Name", "Age")
	if err != nil {
		t.Errorf("Unexpected error: %v", err)
	}

	expected := "Name,Age,\n"
	if buffer.String() != expected {
		t.Errorf("Expected '%s', got '%s'", expected, buffer.String())
	}
}

func TestCSVFields(t *testing.T) {
	buffer := new(bytes.Buffer)
	printer := &CSVPrinter{writer: buffer}

	err := printer.Fields("John Doe", 30)
	if err != nil {
		t.Errorf("Unexpected error: %v", err)
	}

	expected := "John Doe,30,\n"
	if buffer.String() != expected {
		t.Errorf("Expected '%s', got '%s'", expected, buffer.String())
	}
}

func TestCSVStringFields(t *testing.T) {
	buffer := new(bytes.Buffer)
	printer := &CSVPrinter{writer: buffer}

	err := printer.StringFields("John Doe", "30")
	if err != nil {
		t.Errorf("Unexpected error: %v", err)
	}

	expected := "John Doe,30,\n"
	if buffer.String() != expected {
		t.Errorf("Expected '%s', got '%s'", expected, buffer.String())
	}
}