session_test.go

// TestRead_Interrupted asserts that blocked reads without a deadline are
// interrupted when a session is closed.
func TestRead_Interrupted(t *testing.T) {
	client, server := testClientServer()
	defer client.Close()
	defer server.Close()

	stream, err := client.Open()
	if err != nil {
		t.Fatalf("unexpected error opening a new stream: %v", err)
	}
	defer stream.Close()

	stream2, err := server.Accept()
	if err != nil {
		t.Fatalf("unexpected error accepting a new connection: %v", err)
	}
	defer stream2.Close()

	errCh := make(chan error, 1)
	go func() {
		buf := make([]byte, 1)
		_, err := stream.Read(buf)
		errCh <- err
	}()

	// Let Read block
	time.Sleep(100 * time.Millisecond)

	// Close the server to interrupt the waiting Read
	if err := server.Close(); err != nil {
		t.Fatalf("unexpected error closing server: %v", err)
	}

	select {
	case err := <-errCh:
		if err == nil {
			t.Fatalf("no error from waiting Read; expected EOF")
		}
		if !errors.Is(err, io.EOF) {
			t.Fatalf("unexpected error from waiting Read; expected EOF but got: %v", err)
		}
	case <-time.After(10 * time.Second):
		t.Fatalf("timed out waiting for blocked Read to unblock")
	}
}