Fix handling of empty payloads

Signed-off-by: Maksym Pavlenko <pavlenko.maksym@gmail.com>
containerd · Feb 2, 2024 · dea99e9 · dea99e9
1 parent 336fc1b
commit dea99e9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/integration/streaming_test.go b/integration/streaming_test.go
@@ -420,9 +420,6 @@ func emptyPayloadStream(ctx context.Context, client streaming.TTRPCStreamingClie
 			}
 		}
 
-		if err := stream.CloseSend(); err != nil {
-			t.Fatal(err)
-		}
 		if _, err := stream.Recv(); err != io.EOF {
 			t.Fatalf("Expected io.EOF, got %v", err)
 		}

diff --git a/services.go b/services.go
@@ -140,7 +140,7 @@ func (s *serviceSet) handle(ctx context.Context, req *Request, respond func(*sta
 			respond(st, p, stream.StreamingServer, true)
 		}()
 
-		if req.Payload != nil {
+		if req.Payload != nil || !info.StreamingClient {
 			unmarshal := func(obj interface{}) error {
 				return protoUnmarshal(req.Payload, obj)
 			}