elasticsearch: various test fixes (#4299)

* elasticsearch: tolerate context.Canceled in v8 test stream goroutine

When StopWithin cancels the stream context, Run returns
context.Canceled. Match the mssqlserver fix so the goroutine does not
fail the test on expected shutdown.

Fixes CON-438

* elasticsearch: tolerate context.Canceled in v9 test stream goroutine

When StopWithin cancels the stream context, Run returns
context.Canceled. Match the v8 fix so the goroutine does not fail the
test on expected shutdown.

Fixes CON-449

Author: mmatczuk

internal/impl/elasticsearch/v8/integration_test.go

@@ -14,7 +14,9 @@
 package elasticsearch
 
 import (
+	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"testing"
 	"time"
@@ -79,7 +81,9 @@ elasticsearch_v8:
 	require.NoError(t, err)
 
 	go func() {
-		require.NoError(t, stream.Run(ctx))
+		if err := stream.Run(ctx); err != nil && !errors.Is(err, context.Canceled) {
+			t.Error(err)
+		}
 	}()
 	defer func() {
 		err := stream.StopWithin(time.Second * 3)

internal/impl/elasticsearch/v9/integration_test.go

@@ -14,7 +14,9 @@
 package elasticsearch
 
 import (
+	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"testing"
 	"time"
@@ -78,7 +80,9 @@ elasticsearch_v9:
 	require.NoError(t, err)
 
 	go func() {
-		require.NoError(t, stream.Run(ctx))
+		if err := stream.Run(ctx); err != nil && !errors.Is(err, context.Canceled) {
+			t.Error(err)
+		}
 	}()
 	defer func() {
 		err := stream.StopWithin(time.Second * 3)