lib/file: add tests for WriteAtomic

This commit is contained in:
Mitchell Hashimoto 2018-05-03 21:58:07 -07:00
parent 2809203408
commit e9b8e5d265
No known key found for this signature in database
GPG Key ID: 744E147AA52F5B0A

32
lib/file/atomic_test.go Normal file
View File

@ -0,0 +1,32 @@
package file
import (
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
)
// This doesn't really test the "atomic" part of this function. It really
// tests that it just writes the file properly. I would love to test this
// better but I'm not sure how. -mitchellh
func TestWriteAtomic(t *testing.T) {
require := require.New(t)
td, err := ioutil.TempDir("", "lib-file")
require.NoError(err)
defer os.RemoveAll(td)
// Create a subdir that doesn't exist to test that it is created
path := filepath.Join(td, "subdir", "file")
// Write
expected := []byte("hello")
require.NoError(WriteAtomic(path, expected))
// Read and verify
actual, err := ioutil.ReadFile(path)
require.NoError(err)
require.Equal(expected, actual)
}