amseltools/internal/dirtoexcel/dirtoexcel_test.go

package dirtoexcel_test

import (
	"fmt"
	"os"
	"path"
	"testing"

	"github.com/matryer/is"
	"github.com/xuri/excelize/v2"
	"go.xsfx.dev/amseltools/internal/dirtoexcel"
)

func TestCreate(t *testing.T) {
	type want struct {
		Company string
		Value   string
	}

	tables := []struct {
		name   string
		files  []string
		sheets map[string][]want
	}{
		{
			"00",
			[]string{
				"amazon_foo_12.34_bar.jpg",
				"bma-zon_12.34_bar.jpg",
				"aldi_10.98.jpg",
				"aldi_10.00.jpg",
			},
			map[string][]want{
				"Food": {
					{
						"ALDI",
						"20.98",
					},
				},
				"Non-Food": {
					{
						"AMAZON",
						"12.34",
					},
				},
				"Other": {
					{
						"BMA-ZON",
						"12.34",
					},
				},
			},
		},
	}

	is := is.New(t)

	for _, tt := range tables {
		tt := tt

		t.Run(tt.name, func(t *testing.T) {
			dir := t.TempDir()
			for _, i := range tt.files {
				_, err := os.Create(path.Join(dir, i))
				is.NoErr(err)
			}

			outfile := path.Join(dir, "foo.xlsx")

			// Create the excel.
			err := dirtoexcel.Create(outfile, dir)
			is.NoErr(err)

			// NOTE: Uncomment this to view excel file.
			//
			// eFile, err := os.ReadFile(outfile)
			// is.NoErr(err)
			//
			// err = os.WriteFile(path.Join("/home/marv/wip/amseltools", "foo.xlsx"), eFile, 0o644)
			// is.NoErr(err)

			// Read the excel.
			f, err := excelize.OpenFile(outfile)
			is.NoErr(err)

			defer func() {
				err := f.Close()
				is.NoErr(err)
			}()

			for name, values := range tt.sheets {
				for i, j := range values {
					company, err := f.GetCellValue(name, fmt.Sprintf("A%d", i+1))
					is.NoErr(err)

					is.Equal(j.Company, company)

					value, err := f.GetCellValue(name, fmt.Sprintf("B%d", i+1))
					is.NoErr(err)

					is.Equal(j.Value, value)
				}
			}
		})
	}
}

func TestCategoriesInsert(t *testing.T) {
	is := is.New(t)

	m := dirtoexcel.Categories{}
	err := m.Insert("amazon", "11.11")
	is.NoErr(err)

	is.Equal(dirtoexcel.Categories{"Non-Food": []dirtoexcel.Value{{"AMAZON", 11.11}}}, m)
}