[github/fretlink/terraform-provider-statuscake.git] / vendor / github.com / hashicorp / terraform / config / module / testing.go

package module

import (
	"io/ioutil"
	"os"
	"testing"
)

// TestTree loads a module at the given path and returns the tree as well
// as a function that should be deferred to clean up resources.
func TestTree(t *testing.T, path string) (*Tree, func()) {
	// Create a temporary directory for module storage
	dir, err := ioutil.TempDir("", "tf")
	if err != nil {
		t.Fatalf("err: %s", err)
		return nil, nil
	}

	// Load the module
	mod, err := NewTreeModule("", path)
	if err != nil {
		t.Fatalf("err: %s", err)
		return nil, nil
	}

	// Get the child modules
	s := &Storage{StorageDir: dir, Mode: GetModeGet}
	if err := mod.Load(s); err != nil {
		t.Fatalf("err: %s", err)
		return nil, nil
	}

	return mod, func() {
		os.RemoveAll(dir)
	}
}
Commit	Line	Data
bae9f6d2 JC	1	package module
	2
	3	import (
	4	"io/ioutil"
	5	"os"
	6	"testing"
bae9f6d2 JC	7	)
	8
	9	// TestTree loads a module at the given path and returns the tree as well
	10	// as a function that should be deferred to clean up resources.
	11	func TestTree(t testing.T, path string) (Tree, func()) {
	12	// Create a temporary directory for module storage
	13	dir, err := ioutil.TempDir("", "tf")
	14	if err != nil {
	15	t.Fatalf("err: %s", err)
	16	return nil, nil
	17	}
	18
	19	// Load the module
	20	mod, err := NewTreeModule("", path)
	21	if err != nil {
	22	t.Fatalf("err: %s", err)
	23	return nil, nil
	24	}
	25
	26	// Get the child modules
15c0b25d AP	27	s := &Storage{StorageDir: dir, Mode: GetModeGet}
15c0b25d AP	28	if err := mod.Load(s); err != nil {
bae9f6d2 JC	29	t.Fatalf("err: %s", err)
	30	return nil, nil
	31	}
	32
	33	return mod, func() {
	34	os.RemoveAll(dir)
	35	}
	36	}