numscript/internal/utils/pretty_csv_test.go

package utils_test

import (
	"testing"

	"github.com/hanzo-fi/numscript/internal/utils"
	"github.com/gkampitakis/go-snaps/snaps"
)

func TestPrettyCsv(t *testing.T) {
	out := utils.CsvPretty([]string{
		"Account", "Asset", "Balance",
	}, [][]string{
		{"alice", "EUR/2", "1"},
		{"alice", "USD/1234", "999999"},
		{"bob", "BTC", "3"},
	}, true)

	snaps.MatchSnapshot(t, out)
}

func TestPrettyCsvOmitEmptyCols(t *testing.T) {
	t.Run("drops a column whose cells are all empty", func(t *testing.T) {
		out := utils.CsvPrettyOmitEmptyCols([]string{
			"Account", "Scope", "Asset", "Color", "Balance",
		}, [][]string{
			{"alice", "", "EUR/2", "", "1"},
			{"bob", "", "BTC", "", "3"},
		}, false)

		snaps.MatchSnapshot(t, out)
	})

	t.Run("with no rows, still shows every header", func(t *testing.T) {
		out := utils.CsvPrettyOmitEmptyCols([]string{
			"Account", "Scope", "Asset", "Color", "Balance",
		}, [][]string{}, false)

		snaps.MatchSnapshot(t, out)
	})

	t.Run("keeps a column when at least one cell is non-empty", func(t *testing.T) {
		out := utils.CsvPrettyOmitEmptyCols([]string{
			"Account", "Scope", "Asset", "Color", "Balance",
		}, [][]string{
			{"alice", "eu", "EUR/2", "", "1"},
			{"bob", "", "BTC", "", "3"},
		}, false)

		snaps.MatchSnapshot(t, out)
	})
}

func TestPrettyCsvMap(t *testing.T) {
	out := utils.CsvPrettyMap("Name", "Value", map[string]string{
		"a":                       "0",
		"b":                       "12345",
		"very-very-very-long-key": "",
	})

	snaps.MatchSnapshot(t, out)
}