52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
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())
|
|
}
|
|
}
|