From d1ffe3fc2f4a62ab13135f1eb63d5da3a36dfdbc Mon Sep 17 00:00:00 2001
From: Julia Bardi <julia.bardi@elastic.co>
Date: Mon, 27 Nov 2023 13:49:01 +0100
Subject: [PATCH] try to fix test

---
 internal/pkg/bulk/bulk_test.go | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/internal/pkg/bulk/bulk_test.go b/internal/pkg/bulk/bulk_test.go
index 82dc6b7fc..4d7f85d6d 100644
--- a/internal/pkg/bulk/bulk_test.go
+++ b/internal/pkg/bulk/bulk_test.go
@@ -330,6 +330,15 @@ func TestCancelCtxChildBulkerReplaced(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	var waitBulker sync.WaitGroup
+	waitBulker.Add(1)
+	go func() {
+		defer waitBulker.Done()
+		if err := (childBulker.(*Bulker)).Run(ctx); !errors.Is(err, context.Canceled) {
+			t.Fatal(err)
+		}
+	}()
+
 	// output cfg changed
 	outputMap["remote"] = map[string]interface{}{
 		"type":          "remote_elasticsearch",
@@ -349,12 +358,16 @@ func TestCancelCtxChildBulkerReplaced(t *testing.T) {
 		err := childBulker.APIKeyUpdate(ctx, "", "", make([]byte, 0))
 
 		t.Log(err)
+		// TODO dial tcp: lookup remote-es: no such host
 		if !errors.Is(err, context.Canceled) {
-			t.Error("Expected context cancel err: ", err)
+			t.Fatal("Expected context cancel err: ", err)
 		}
+		ctx.Done()
 	}()
 
 	wg.Wait()
+	cancel()
+	waitBulker.Wait()
 }
 
 func benchmarkMockBulk(b *testing.B, samples [][]byte) {