Skip to content

Instantly share code, notes, and snippets.

@rogeralsing
Created March 22, 2016 09:08
Show Gist options
  • Save rogeralsing/6ee1f7e8d05128868313 to your computer and use it in GitHub Desktop.
Save rogeralsing/6ee1f7e8d05128868313 to your computer and use it in GitHub Desktop.
func TestStreamClient(t *testing.T) {
const address = "127.0.0.1:8090"
lis, err := net.Listen("tcp", address)
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
defer lis.Close()
s := grpc.NewServer()
server := Server{}
RegisterPeopleServer(s,&server)
go s.Serve(lis)
conn, err := grpc.Dial(address, grpc.WithInsecure())
defer conn.Close()
client := NewPeopleClient(conn)
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
stream, _ := client.GetPeople(context.Background(), &EmptyMessage{})
for i := 0; i < 1000000; i++ {
stream.Recv()
}
println("Done")
}()
wg.Wait()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment