test: update middleware test

ogen-go · Nov 14, 2022 · 790ee0e · 790ee0e
1 parent 956a120
commit 790ee0e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/integration/middleware_test.go b/internal/integration/middleware_test.go
@@ -55,12 +55,16 @@ func ModifyRequest(logger *zap.Logger) middleware.Middleware {
 	) (middleware.Response, error) {
 		switch body := req.Body.(type) {
 		case api.PetUploadAvatarByIDReq:
-			if v, ok := req.Params["petID"].(int64); ok {
+			key := middleware.ParameterKey{
+				Name: "petID",
+				In:   "query",
+			}
+			if v, ok := req.Params[key].(int64); ok {
 				logger.Info("Modifying request", zap.Int64("petID", v))
 				req.Body = api.PetUploadAvatarByIDReq{
 					Data: io.MultiReader(strings.NewReader("prefix"), body.Data),
 				}
-				req.Params["petID"] = v + 1
+				req.Params[key] = v + 1
 			}
 		default:
 			logger.Info("Skipping request modification")