108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
package io
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var (
|
|
testWriterPath = "testPath"
|
|
testWriterPathError = "test Path, @:ººººº\\/.Ç*⁺´+"
|
|
testFileContent = "test data"
|
|
testFileName = "test-file.txt"
|
|
workingDir = ""
|
|
)
|
|
|
|
func deleteAllDirs() error {
|
|
return os.RemoveAll(testWriterPath)
|
|
}
|
|
|
|
func init() {
|
|
err := deleteAllDirs()
|
|
if err != nil {
|
|
log.Fatalln("deleteAllDirs: ", err)
|
|
}
|
|
workingDir, err = os.Getwd()
|
|
if err != nil {
|
|
log.Fatalln("os.Getwd: ", err)
|
|
}
|
|
}
|
|
|
|
func Test_newWriter(t *testing.T) {
|
|
w := newWriter(testWriterPath)
|
|
require.NotNil(t, w)
|
|
}
|
|
|
|
func TestNewWriter(t *testing.T) {
|
|
w := NewWriter(testWriterPath)
|
|
require.NotNil(t, w)
|
|
}
|
|
|
|
func Test_writer_WriteFile(t *testing.T) {
|
|
err := deleteAllDirs()
|
|
require.NoError(t, err)
|
|
|
|
defer func(t *testing.T) {
|
|
err := deleteAllDirs()
|
|
require.NoError(t, err)
|
|
}(t)
|
|
|
|
w := newWriter(testWriterPath)
|
|
wgot, err := w.WriteFile(testFileName, []byte(testFileContent))
|
|
require.NoError(t, err)
|
|
|
|
btsReaded, err := os.ReadFile(wgot)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, btsReaded)
|
|
require.Equal(t, string(btsReaded), testFileContent)
|
|
}
|
|
|
|
func Test_writer_WriteFileError(t *testing.T) {
|
|
err := deleteAllDirs()
|
|
require.NoError(t, err)
|
|
|
|
defer func(t *testing.T) {
|
|
err := deleteAllDirs()
|
|
require.NoError(t, err)
|
|
}(t)
|
|
|
|
w := newWriter("")
|
|
_, err = w.WriteFile(testFileName, []byte(testFileContent))
|
|
assert.Error(t, err)
|
|
|
|
w = newWriter(testWriterPath)
|
|
_, err = w.WriteFile("", []byte(testFileContent))
|
|
require.Error(t, err)
|
|
|
|
go func(t *testing.T) {
|
|
err := deleteAllDirs()
|
|
require.NoError(t, err)
|
|
for {
|
|
f, err := os.Stat(testWriterPath)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "no such file or directory") {
|
|
continue
|
|
}
|
|
t.Errorf("os.Stat error: %s", err)
|
|
}
|
|
err = os.RemoveAll(testWriterPath)
|
|
require.NotNil(t, f)
|
|
log.Println("exists deleting dir")
|
|
require.NoError(t, err, "should not throw error because dir exists")
|
|
return
|
|
}
|
|
}(t)
|
|
|
|
_, err = w.WriteFile(testFileName, nil)
|
|
assert.Error(t, err)
|
|
|
|
w = newWriter(testWriterPathError)
|
|
_, err = w.WriteFile(testFileContent, []byte(testFileContent))
|
|
require.Error(t, err)
|
|
}
|