32 lines
637 B
Go
32 lines
637 B
Go
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")
|
|
|
|
}
|