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)
}