Patch up tests

mweagle · Oct 7, 2016 · 3e543b1 · 3e543b1
1 parent d6932c4
commit 3e543b1
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/func_importvalue_test.go b/func_importvalue_test.go
@@ -14,7 +14,7 @@ func (testSuite *ImportValueFuncTest) TestRef(c *C) {
 	inputBuf := `{"Fn::ImportValue" : "sharedValueToImport"}`
 	f, err := unmarshalFunc([]byte(inputBuf))
 	c.Assert(err, IsNil)
-	c.Assert(f.(StringFunc).String(), DeepEquals, ImportValue("sharedValueToImport").String())
+	c.Assert(f.(StringFunc).String(), DeepEquals, ImportValue(String("sharedValueToImport")).String())
 
 	// tidy the JSON input
 	inputStruct := map[string]interface{}{}
@@ -27,14 +27,10 @@ func (testSuite *ImportValueFuncTest) TestRef(c *C) {
 }
 
 func (testSuite *ImportValueFuncTest) TestFailures(c *C) {
-	inputBuf := `{"Fn::ImportValue": {"Fn::ImportValue": "foo"}}`
+	inputBuf := `{"Fn::ImportValue": ["1"]}`
 	_, err := unmarshalFunc([]byte(inputBuf))
 	c.Assert(err, ErrorMatches, "cannot decode function")
 
-	inputBuf = `{"Fn::ImportValue": ["1"]}`
-	_, err = unmarshalFunc([]byte(inputBuf))
-	c.Assert(err, ErrorMatches, "cannot decode function")
-
 	inputBuf = `{"Fn::ImportValue": true}`
 	_, err = unmarshalFunc([]byte(inputBuf))
 	c.Assert(err, ErrorMatches, "cannot decode function")