package export

import (
	"github.com/stretchr/testify/assert"
	"os"
	"testing"
)

func TestExcel_Write(t *testing.T) {
	pwd, _ := os.Getwd()
	e := NewExcel(pwd+"/aa-{begin}.xlsx", 5, map[string]string{"begin": "202301"})

	e.Open()

	e.WriteTitle([]string{"姓名", "年龄"})
	data := make([]interface{}, 2)
	data[0] = "张三"
	for i := 0; i < 9; i++ {
		data[1] = 10 + i
		e.Write(data)
	}
	e.Close()

	assert.FileExists(t, pwd+"/aa-202301_0.xlsx")
	assert.FileExists(t, pwd+"/aa-202301_1.xlsx")
	assert.NoFileExists(t, pwd+"/aa-202301_2.xlsx")

	_ = os.Remove(pwd + "/aa-202301_0.xlsx")
	_ = os.Remove(pwd + "/aa-202301_1.xlsx")

}