test: fix panic in TestNotFound

4c86378b · Steven Allen · 4a78fc6e · 4c86378b
Commit 4c86378b authored May 05, 2020 by Steven Allen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

ext_test.go ext_test.go +2 -1

No files found.
--- a/ext_test.go
+++ b/ext_test.go
@@ -247,7 +247,8 @@ func TestNotFound(t *testing.T) {

 				pmes := new(pb.Message)
 				if err := pbr.ReadMsg(pmes); err != nil {
-					panic(err)
+					// this isn't an error, it just means the stream has died.
+					return
 				}

 				switch pmes.GetType() {