From 72f75b2f2b01d0f5a8c184b326896643c59aa05f Mon Sep 17 00:00:00 2001
From: moogacs <contact@m-mooga.com>
Date: Wed, 17 Apr 2024 13:36:35 +0200
Subject: [PATCH] fix: read repair acceptance test

---
 test/acceptance/replication/read_repair_test.go | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/test/acceptance/replication/read_repair_test.go b/test/acceptance/replication/read_repair_test.go
index a57c963f93..635d5ba70d 100644
--- a/test/acceptance/replication/read_repair_test.go
+++ b/test/acceptance/replication/read_repair_test.go
@@ -31,7 +31,7 @@ func readRepair(t *testing.T) {
 	defer cancel()
 
 	compose, err := docker.New().
-		With2NodeCluster().
+		With3NodeCluster().
 		WithText2VecContextionary().
 		Start(ctx)
 	require.Nil(t, err)
@@ -47,12 +47,12 @@ func readRepair(t *testing.T) {
 
 	t.Run("create schema", func(t *testing.T) {
 		paragraphClass.ReplicationConfig = &models.ReplicationConfig{
-			Factor: 2,
+			Factor: 3,
 		}
 		paragraphClass.Vectorizer = "text2vec-contextionary"
 		helper.CreateClass(t, paragraphClass)
 		articleClass.ReplicationConfig = &models.ReplicationConfig{
-			Factor: 2,
+			Factor: 3,
 		}
 		helper.CreateClass(t, articleClass)
 	})
@@ -106,6 +106,7 @@ func readRepair(t *testing.T) {
 	})
 
 	t.Run("stop node 1", func(t *testing.T) {
+		time.Sleep(3 * time.Second) // wait before killing a leader so that data replicated to followers
 		stopNodeAt(ctx, t, compose, 1)
 		time.Sleep(10 * time.Second)
 	})
@@ -140,15 +141,17 @@ func readRepair(t *testing.T) {
 		require.True(t, exists)
 	})
 
-	t.Run("assert updated object read repair was made", func(t *testing.T) {
+	t.Run("stop node 2", func(t *testing.T) {
 		stopNodeAt(ctx, t, compose, 2)
+	})
 
-		exists, err := objectExistsCL(t, compose.GetWeaviate().URI(),
+	t.Run("assert updated object read repair was made", func(t *testing.T) {
+		exists, err := objectExistsCL(t, compose.ContainerURI(1),
 			replaceObj.Class, replaceObj.ID, replica.One)
 		require.Nil(t, err)
 		require.True(t, exists)
 
-		resp, err := getObjectCL(t, compose.GetWeaviate().URI(),
+		resp, err := getObjectCL(t, compose.ContainerURI(1),
 			repairObj.Class, repairObj.ID, replica.One)
 		require.Nil(t, err)
 		assert.Equal(t, replaceObj.ID, resp.ID)