From 23bb1827328e0b036ae1770a3ecb7881ab7b6409 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Mon, 9 Feb 2015 11:20:32 -0800
Subject: [PATCH] terraform: more tests

---
 terraform/graph_config_node_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/terraform/graph_config_node_test.go b/terraform/graph_config_node_test.go
index 1476c97e0..64127e5ca 100644
--- a/terraform/graph_config_node_test.go
+++ b/terraform/graph_config_node_test.go
@@ -1,6 +1,7 @@
 package terraform
 
 import (
+	"reflect"
 	"strings"
 	"testing"
 
@@ -62,6 +63,7 @@ func TestGraphNodeConfigResource_impl(t *testing.T) {
 	var _ dag.NamedVertex = new(GraphNodeConfigResource)
 	var _ graphNodeConfig = new(GraphNodeConfigResource)
 	var _ GraphNodeProviderConsumer = new(GraphNodeConfigResource)
+	var _ GraphNodeProvisionerConsumer = new(GraphNodeConfigResource)
 }
 
 func TestGraphNodeConfigResource_ProvidedBy(t *testing.T) {
@@ -74,6 +76,24 @@ func TestGraphNodeConfigResource_ProvidedBy(t *testing.T) {
 	}
 }
 
+func TestGraphNodeConfigResource_ProvisionedBy(t *testing.T) {
+	n := &GraphNodeConfigResource{
+		Resource: &config.Resource{
+			Type: "aws_instance",
+			Provisioners: []*config.Provisioner{
+				&config.Provisioner{Type: "foo"},
+				&config.Provisioner{Type: "bar"},
+			},
+		},
+	}
+
+	expected := []string{"foo", "bar"}
+	actual := n.ProvisionedBy()
+	if !reflect.DeepEqual(actual, expected) {
+		t.Fatalf("bad: %#v", actual)
+	}
+}
+
 const testGraphNodeModuleExpandStr = `
 aws_instance.bar
   aws_instance.foo