40 lines
820 B
Go
40 lines
820 B
Go
package export
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestFile_FileName(t *testing.T) {
|
|
f := NewFile("/usr/file-{begin}-{end}.xlsx", 10, map[string]string{
|
|
"begin": "20230404",
|
|
"end": "20230404",
|
|
})
|
|
assert.Equal(t, "/usr/file-20230404-20230404_0.xlsx", f.FileName())
|
|
|
|
f.NextFile()
|
|
assert.Equal(t, "/usr/file-20230404-20230404_1.xlsx", f.FileName())
|
|
}
|
|
|
|
func TestFile_IsFileExist(t *testing.T) {
|
|
gwd, _ := os.Getwd()
|
|
|
|
f := NewFile(gwd+"/file-{begin}-{end}.xlsx", 10, map[string]string{
|
|
"begin": "20230404",
|
|
"end": "20230404",
|
|
})
|
|
|
|
assert.False(t, f.IsFileExist())
|
|
|
|
path := gwd + "/file-20230404-20230404_0.xlsx"
|
|
_, err := os.Create(path)
|
|
assert.Nil(t, err)
|
|
|
|
assert.FileExists(t, path)
|
|
assert.True(t, f.IsFileExist())
|
|
|
|
err = os.Remove(path)
|
|
assert.Nil(t, err)
|
|
}
|