test: fix panic in TestNotFound

Stebalien · Stebalien · commit 4c86378b8830 · 2020-05-05T18:41:47.000-07:00
diff --git 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() {

Original file line number	Diff line number	Diff line change
`@@ -247,7 +247,8 @@ func TestNotFound(t *testing.T) {`
`247`	`247`
`248`	`248`	`pmes := new(pb.Message)`
`249`	`249`	`if err := pbr.ReadMsg(pmes); err != nil {`
`250`		`- panic(err)`
	`250`	`+ // this isn't an error, it just means the stream has died.`
	`251`	`+ return`
`251`	`252`	`}`
`252`	`253`
`253`	`254`	`switch pmes.GetType() {`