package biz

import (
	"excel_export/data"
	"fmt"
	"github.com/stretchr/testify/assert"
	"os"
	"testing"
	"time"
)

func orderBiz() *OrderBiz {
	db := data.Conn()
	repo := data.NewOrderRepo(db)
	historyRepo := data.NewHistoryOrderRepo(db)

	return NewOrderBiz(repo, historyRepo, nil)
}

func TestOrderBiz_Export(t *testing.T) {
	biz := orderBiz()
	begin := time.Date(2022, 8, 1, 0, 0, 0, 0, time.UTC)
	end := time.Date(2022, 8, 2, 10, 0, 0, 0, time.UTC)

	err := biz.Export(begin, end)
	assert.Nil(t, err)

	for i := 0; i <= biz.FileSize(); i++ {
		fileName := fmt.Sprintf(biz.FileName(), i)
		assert.FileExists(t, fileName)

		//清理文件
		os.Remove(fileName)
	}
}