opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server: export ServerTransportStreamFromContext for unary interceptors to control headers/trailers (#2019)

Browse Source
This commit is contained in:
dfawley
2018-04-26 17:38:15 -07:00
committed by Menghan Li
parent 07709e8a3d
commit fc37cf1364
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server_test.go
View File

@ -128,7 +128,7 @@ func TestGetServiceInfo(t *testing.T) {
func TestStreamContext(t *testing.T) {
expectedStream := &transport.Stream{}
ctx := NewContextWithServerTransportStream(context.Background(), expectedStream)
s := serverTransportStreamFromContext(ctx)
s := ServerTransportStreamFromContext(ctx)
stream, ok := s.(*transport.Stream)
if !ok || expectedStream != stream {
t.Fatalf("GetStreamFromContext(%v) = %v, %t, want: %v, true", ctx, stream, ok, expectedStream)