terraform/command/graph_test.go

107 lines
2.0 KiB
Go
Raw Normal View History

2014-07-12 05:38:03 +02:00
package command
import (
2014-07-12 05:41:47 +02:00
"os"
2014-07-12 05:38:03 +02:00
"strings"
"testing"
2014-07-13 04:25:50 +02:00
"github.com/hashicorp/terraform/config"
"github.com/hashicorp/terraform/terraform"
2014-07-12 05:38:03 +02:00
"github.com/mitchellh/cli"
)
func TestGraph(t *testing.T) {
ui := new(cli.MockUi)
c := &GraphCommand{
Meta: Meta{
ContextOpts: testCtxConfig(testProvider()),
Ui: ui,
},
2014-07-12 05:38:03 +02:00
}
args := []string{
testFixturePath("graph"),
}
if code := c.Run(args); code != 0 {
t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
}
output := ui.OutputWriter.String()
if !strings.Contains(output, "digraph {") {
t.Fatalf("doesn't look like digraph: %s", output)
}
}
2014-07-12 05:41:47 +02:00
func TestGraph_multipleArgs(t *testing.T) {
2014-07-12 05:38:03 +02:00
ui := new(cli.MockUi)
2014-07-13 04:47:31 +02:00
c := &GraphCommand{
Meta: Meta{
ContextOpts: testCtxConfig(testProvider()),
Ui: ui,
},
2014-07-12 05:38:03 +02:00
}
2014-07-12 05:41:47 +02:00
args := []string{
"bad",
"bad",
}
2014-07-12 05:38:03 +02:00
if code := c.Run(args); code != 1 {
t.Fatalf("bad: \n%s", ui.OutputWriter.String())
}
}
2014-07-12 05:41:47 +02:00
func TestGraph_noArgs(t *testing.T) {
cwd, err := os.Getwd()
if err != nil {
t.Fatalf("err: %s", err)
}
if err := os.Chdir(testFixturePath("graph")); err != nil {
t.Fatalf("err: %s", err)
}
defer os.Chdir(cwd)
2014-07-12 05:38:03 +02:00
ui := new(cli.MockUi)
2014-07-12 05:41:47 +02:00
c := &GraphCommand{
Meta: Meta{
ContextOpts: testCtxConfig(testProvider()),
Ui: ui,
},
2014-07-12 05:38:03 +02:00
}
2014-07-12 05:41:47 +02:00
args := []string{}
if code := c.Run(args); code != 0 {
t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
2014-07-12 05:38:03 +02:00
}
2014-07-12 05:41:47 +02:00
output := ui.OutputWriter.String()
if !strings.Contains(output, "digraph {") {
t.Fatalf("doesn't look like digraph: %s", output)
2014-07-12 05:38:03 +02:00
}
}
2014-07-13 04:25:50 +02:00
func TestGraph_plan(t *testing.T) {
planPath := testPlanFile(t, &terraform.Plan{
Config: new(config.Config),
})
ui := new(cli.MockUi)
c := &GraphCommand{
Meta: Meta{
ContextOpts: testCtxConfig(testProvider()),
Ui: ui,
},
2014-07-13 04:25:50 +02:00
}
args := []string{
planPath,
}
if code := c.Run(args); code != 0 {
t.Fatalf("bad: \n%s", ui.ErrorWriter.String())
}
output := ui.OutputWriter.String()
if !strings.Contains(output, "digraph {") {
t.Fatalf("doesn't look like digraph: %s", output)
}
}