fix tests

Author: rdallman

api/agent/lb_agent_test.go

@@ -269,15 +269,15 @@ func TestEnforceLbTimeout(t *testing.T) {
 
 // SetCallType create a models.Call setting up the provided Call Type
 func SetCallType(callType string) CallOpt {
-	return func(c *call) error {
+	return func(cfg Config, c *call) error {
 		c.Call = &models.Call{Type: callType}
 		c.req, _ = http.NewRequest("GET", "http://www.example.com", nil)
 		return nil
 	}
 }
 
 func ModifyCallRequest(callType string) CallOpt {
-	return func(c *call) error {
+	return func(cfg Config, c *call) error {
 		c.Call.Type = callType
 		return nil
 	}

api/server/runner_fninvoke_test.go

@@ -218,7 +218,7 @@ func TestFnInvokeRunnerExecution(t *testing.T) {
 		{"/invoke/http_stream_fn_id", oomer, nil, http.MethodPost, http.StatusBadGateway, nil, "error receiving function response", nil},
 		{"/invoke/http_stream_fn_id", bigbuf, nil, http.MethodPost, http.StatusRequestEntityTooLarge, nil, "", nil},
 
-		{"/invoke/dne_fn_id", ``, nil, http.MethodPost, http.StatusNotFound, nil, "pull access denied", nil},
+		{"/invoke/dne_fn_id", ``, nil, http.MethodPost, http.StatusNotFound, nil, "Failed to pull image", nil},
 		{"/invoke/dnereg_fn_id", ``, nil, http.MethodPost, http.StatusBadGateway, nil, "connection refused", nil},
 
 		// XXX(reed): nil, nil, what are these?

api/server/runner_httptrigger_test.go

@@ -301,7 +301,7 @@ func TestTriggerRunnerExecution(t *testing.T) {
 		// XXX(reed): meh we really should try to get oom out, but maybe it's better left to the logs?
 		{"/t/myapp/httpstream", nil, oomer, "POST", http.StatusBadGateway, nil, "error receiving function response", nil},
 
-		{"/t/myapp/mydne", nil, ``, "GET", http.StatusNotFound, nil, "pull access denied", nil},
+		{"/t/myapp/mydne", nil, ``, "GET", http.StatusNotFound, nil, "Failed to pull image", nil},
 		{"/t/myapp/mydneregistry", nil, ``, "GET", http.StatusBadGateway, nil, "connection refused", nil},
 
 		// XXX(reed): what are these?