38 lines
847 B
Go
38 lines
847 B
Go
|
package allocdir
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
hclog "github.com/hashicorp/go-hclog"
|
||
|
|
testing "github.com/mitchellh/go-testing-interface"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestAllocDir returns a built alloc dir in a temporary directory and cleanup
|
||
|
|
// func.
|
||
|
|
func TestAllocDir(t testing.T, l hclog.Logger, prefix string) (*AllocDir, func()) {
|
||
|
|
dir, err := ioutil.TempDir("", prefix)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Couldn't create temp dir: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
allocDir := NewAllocDir(l, dir)
|
||
|
|
|
||
|
|
cleanup := func() {
|
||
|
|
if err := os.RemoveAll(dir); err != nil {
|
||
|
|
t.Logf("error cleaning up alloc dir %q: %v", prefix, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := allocDir.Destroy(); err != nil {
|
||
|
|
t.Logf("error cleaning up alloc dir %q: %v", prefix, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := allocDir.Build(); err != nil {
|
||
|
|
cleanup()
|
||
|
|
t.Fatalf("error building alloc dir %q: %v", prefix, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return allocDir, cleanup
|
||
|
|
}
|