server_test: wait for OnClose in TestClientEOF.

klihub · klihub · commit e0cd80111675 · 2023-07-27T17:59:04.000+03:00
In the test for client Call failing with ErrClosed on a closed
server, wait for the client's OnClose handler to get triggered
to make sure closing the socket had properly been administered
on the client's side. Otherwise trying a new Call() might fail
with some other error than ErrClosed, for instance ENOTCONN.

Signed-off-by: Krisztian Litkey &lt;krisztian.litkey@intel.com&gt;
diff --git a/server_test.go b/server_test.go
@@ -363,6 +363,8 @@ func TestClientEOF(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	client.UserOnCloseWait(ctx)
+
 	// server shutdown, but we still make a call.
 	if err := client.Call(ctx, serviceName, "Test", tp, tp); err == nil {
 		t.Fatalf("expected error when calling against shutdown server")

Original file line number	Diff line number	Diff line change
`@@ -363,6 +363,8 @@ func TestClientEOF(t *testing.T) {`
`363`	`363`	`t.Fatal(err)`
`364`	`364`	`}`
`365`	`365`
	`366`	`+ client.UserOnCloseWait(ctx)`
	`367`	`+`
`366`	`368`	`// server shutdown, but we still make a call.`
`367`	`369`	`if err := client.Call(ctx, serviceName, "Test", tp, tp); err == nil {`
`368`	`370`	`t.Fatalf("expected error when calling against shutdown server")`