From 6bfec056bae0e15d9442edcd55990efda8635baf Mon Sep 17 00:00:00 2001
From: Josh Humphries <jh@fullstory.com>
Date: Wed, 27 Feb 2019 18:09:27 -0500
Subject: [PATCH] fix flaky test where code can be 'cancelled' unexpectedly,
 instead of some error code provided by the server

---
 invoke.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/invoke.go b/invoke.go
index 75acf6b..d2f16cb 100644
--- a/invoke.go
+++ b/invoke.go
@@ -311,6 +311,7 @@ func invokeBidi(ctx context.Context, stub grpcdynamic.Stub, md *desc.MethodDescr
 				}
 				if err != nil {
 					err = fmt.Errorf("error getting request data: %v", err)
+					cancel()
 					break
 				}
 
@@ -321,7 +322,6 @@ func invokeBidi(ctx context.Context, stub grpcdynamic.Stub, md *desc.MethodDescr
 
 			if err != nil {
 				sendErr.Store(err)
-				cancel()
 			}
 		}()
 	}