1 files changed, 12 insertions, 0 deletions

diff --git a/vendor/github.com/hashicorp/terraform/dag/marshal.go b/vendor/github.com/hashicorp/terraform/dag/marshal.go
index 16d5dd6..c567d27 100644
--- a/vendor/github.com/hashicorp/terraform/dag/marshal.go
+++ b/vendor/github.com/hashicorp/terraform/dag/marshal.go
@@ -273,6 +273,9 @@ func (e *encoder) Encode(i interface{}) {
 }
 
 func (e *encoder) Add(v Vertex) {
+        if e == nil {
+                return
+        }
         e.Encode(marshalTransform{
                 Type:      typeTransform,
                 AddVertex: newMarshalVertex(v),
@@ -281,6 +284,9 @@ func (e *encoder) Add(v Vertex) {
 
 // Remove records the removal of Vertex v.
 func (e *encoder) Remove(v Vertex) {
+        if e == nil {
+                return
+        }
         e.Encode(marshalTransform{
                 Type:         typeTransform,
                 RemoveVertex: newMarshalVertex(v),
@@ -288,6 +294,9 @@ func (e *encoder) Remove(v Vertex) {
 }
 
 func (e *encoder) Connect(edge Edge) {
+        if e == nil {
+                return
+        }
         e.Encode(marshalTransform{
                 Type:    typeTransform,
                 AddEdge: newMarshalEdge(edge),
@@ -295,6 +304,9 @@ func (e *encoder) Connect(edge Edge) {
 }
 
 func (e *encoder) RemoveEdge(edge Edge) {
+        if e == nil {
+                return
+        }
         e.Encode(marshalTransform{
                 Type:       typeTransform,
                 RemoveEdge: newMarshalEdge(edge),