68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package file
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
)
|
|
|
|
func TestContent(t *testing.T) {
|
|
tmpDir, err := ioutil.TempDir("", "file.go")
|
|
if err != nil {
|
|
t.Fatalf("could not create test directory")
|
|
}
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
t1 := &F{
|
|
Content: []byte("this is a test"),
|
|
Path: path.Join(tmpDir, "test1"),
|
|
}
|
|
if err := t1.Ensure(); err != nil {
|
|
t.Fatalf("file failed with: %s", err)
|
|
}
|
|
raw, err := ioutil.ReadFile(t1.Path)
|
|
if err != nil {
|
|
t.Fatalf("could not read file: %s", err)
|
|
}
|
|
if bytes.Compare(raw, t1.Content) != 0 {
|
|
t.Fatalf("content mismatch")
|
|
}
|
|
|
|
t2 := &F{
|
|
Content: []byte("other content"),
|
|
Path: path.Join(tmpDir, "test1"),
|
|
}
|
|
if t2.Is() {
|
|
t.Fatalf("Is should have returned false")
|
|
}
|
|
if err := t2.Ensure(); err != nil {
|
|
t.Fatalf("it should enforce new content: %s", err)
|
|
}
|
|
if t1.Is() {
|
|
t.Fatalf("changes not recognized")
|
|
}
|
|
}
|
|
|
|
func TestDir(t *testing.T) {
|
|
tmpDir, err := ioutil.TempDir("", "file.go")
|
|
if err != nil {
|
|
t.Fatalf("could not create test directory")
|
|
}
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
t1 := &F{IsDir: true}
|
|
if err := t1.Ensure(); err == nil {
|
|
t.Fatalf("a file without path should not work")
|
|
}
|
|
t1.Path = path.Join(tmpDir, "t1")
|
|
if err := t1.Ensure(); err != nil {
|
|
t.Fatalf("directory should have been created, instead got: %s", err)
|
|
}
|
|
t1.Content = []byte("just a check")
|
|
if err := t1.Ensure(); err == nil {
|
|
t.Fatalf("content and IsDir should not work")
|
|
}
|
|
}
|