updated for afrouterTest to work with new protos
Change-Id: Ibf1b74e73e3726c6dc83a79d6ff93acc44943103
diff --git a/tests/afrouter/templates/client.go b/tests/afrouter/templates/client.go
index db34efd..fe559d6 100644
--- a/tests/afrouter/templates/client.go
+++ b/tests/afrouter/templates/client.go
@@ -81,17 +81,15 @@
{{range .Methods}}
case "{{.Name}}":
var hdr metadata.MD
- switch t := param.(type) {
+ switch param.(type) {
case *{{.Param}}:
{{if .Ss}}
- _=t
_=hdr
{{else if .Cs}}
- _=t
_=hdr
{{else}}
client := {{.Pkg}}.New{{.Svc}}Client({{$.Name}}Client.conn)
- res, err := client.{{.Name}}(ctx, t, grpc.Header(&hdr))
+ res, err := client.{{.Name}}(ctx, param.(*{{.Param}}), grpc.Header(&hdr))
if err != nil {
return errors.New("Error sending method {{.Name}}")
}