export_test.go

package wordcounter_test

import (
	"os"
	"testing"

	"github.com/jedib0t/go-pretty/table"
	wcg "github.com/100gle/wordcounter"
)

func TestExporter_ExportCSV(t *testing.T) {
	e := wcg.NewExporter()
	data := []wcg.Row{
		{"Name", "Age", "Gender"},
		{"Alice", 25, "Female"},
		{"Bob", 30, "Male"},
	}
	expected := "Name,Age,Gender\nAlice,25,Female\nBob,30,Male"

	result, err := e.ExportCSV(data)
	if err != nil {
		t.Errorf("ExportCSV failed with error: %v", err)
	}

	if result != expected {
		t.Errorf("ExportCSV failed. Expected:\n%v, got:\n%v", expected, result)
	}
}

func TestExporter_ExportCSVWithFilename(t *testing.T) {
	e := wcg.NewExporter()
	data := []wcg.Row{
		{"Name", "Age", "Gender"},
		{"Alice", 25, "Female"},
		{"Bob", 30, "Male"},
	}

	filename := "test.csv"
	expected := "Name,Age,Gender\nAlice,25,Female\nBob,30,Male"
	csvData, err := e.ExportCSV(data, filename)
	if err != nil {
		t.Errorf("ExportCSV failed with error: %v", err)
	}

	if _, err := os.Stat(filename); os.IsNotExist(err) {
		t.Errorf("ExportCSV did not create file: %v", err)
	}

	if csvData != expected {
		t.Errorf("ExportCSV failed. Expected:\n%v, got:\n%v", expected, csvData)
	}

	err = os.Remove(filename)
	if err != nil {
		t.Errorf("ExportCSV could not delete file: %v", err)
	}
}

func TestExporter_ExportExcel(t *testing.T) {
	e := wcg.NewExporter()
	data := []wcg.Row{
		{"Name", "Age", "Gender"},
		{"Alice", 25, "Female"},
		{"Bob", 30, "Male"},
	}
	filename := "test.xlsx"

	err := e.ExportExcel(data, filename)

	if err != nil {
		t.Errorf("ExportExcel failed with error: %v", err)
	}

	// Check if file exists
	if _, err := os.Stat(filename); os.IsNotExist(err) {
		t.Errorf("ExportExcel did not create file: %v", err)
	}

	// Clean up file
	err = os.Remove(filename)
	if err != nil {
		t.Errorf("ExportExcel could not delete file: %v", err)
	}
}

func TestExporter_ExportTable(t *testing.T) {
	e := wcg.NewExporter()
	data := []wcg.Row{
		{"Name", "Age", "Gender"},
		{"Alice", 25, "Female"},
		{"Bob", 30, "Male"},
	}

	expectedTbl := table.NewWriter()
	expectedTbl.AppendHeader(data[0])
	for _, row := range data[1:] {
		expectedTbl.AppendRow(row)
	}

	expected := expectedTbl.Render()
	result := e.ExportTable(data)

	if result != expected {
		t.Errorf("ExportTable failed. Expected: \n%v\nGot: \n%v", expected, result)
	}
}