36 lines
670 B
Go
36 lines
670 B
Go
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)
|
|
}
|
|
}
|