fix tests

Author: shivaji-kharse

dgraph/cmd/dgraphimport/import_test.go

@@ -14,6 +14,7 @@ import (
 	"strconv"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/hypermodeinc/dgraph/v25/dgraphapi"
 	"github.com/hypermodeinc/dgraph/v25/dgraphtest"
@@ -102,6 +103,7 @@ func TestImportApis(t *testing.T) {
 		negativeTestCase bool // True if this is an expected failure case
 		description      string
 		err              string
+		sleep            bool
 	}{
 		{
 			name:             "SingleGroupShutTwoAlphasPerGroup",
@@ -152,6 +154,7 @@ func TestImportApis(t *testing.T) {
 			negativeTestCase: false,
 			description:      "Single group with multiple alphas, shutdown 1 alpha",
 			err:              "",
+			sleep:            true,
 		},
 		{
 			name:             "TwoGroupShutOneAlphaPerGroup",
@@ -162,6 +165,7 @@ func TestImportApis(t *testing.T) {
 			negativeTestCase: false,
 			description:      "Multiple groups with multiple alphas, shutdown 2 alphas per group",
 			err:              "",
+			sleep:            true,
 		},
 		{
 			name:             "ThreeGroupShutOneAlphaPerGroup",
@@ -172,6 +176,7 @@ func TestImportApis(t *testing.T) {
 			negativeTestCase: false,
 			description:      "Three groups with 3 alphas each, shutdown 1 alpha per group",
 			err:              "",
+			sleep:            true,
 		},
 		{
 			name:             "SingleGroupAllAlphasOnline",
@@ -212,12 +217,14 @@ func TestImportApis(t *testing.T) {
 			} else {
 				t.Logf("Running test case: %s", tt.description)
 			}
-			runImportTest(t, tt.bulkAlphas, tt.targetAlphas, tt.replicasFactor, tt.downAlphas, tt.negativeTestCase, tt.err)
+			runImportTest(t, tt.bulkAlphas, tt.targetAlphas, tt.replicasFactor, tt.downAlphas, tt.negativeTestCase,
+				tt.err, tt.sleep)
 		})
 	}
 }
 
-func runImportTest(t *testing.T, bulkAlphas, targetAlphas, replicasFactor, numDownAlphas int, negative bool, errStr string) {
+func runImportTest(t *testing.T, bulkAlphas, targetAlphas, replicasFactor, numDownAlphas int, negative bool,
+	errStr string, sleep bool) {
 	bulkCluster, baseDir := setupBulkCluster(t, bulkAlphas)
 	defer func() { bulkCluster.Cleanup(t.Failed()) }()
 
@@ -296,14 +303,18 @@ func runImportTest(t *testing.T, bulkAlphas, targetAlphas, replicasFactor, numDo
 	for group, alphas := range alphaGroups {
 		for i := 0; i < numDownAlphas; i++ {
 			alphaID := alphas[i]
-			t.Logf("Shutting down alpha %v from group %v", alphaID, group)
+			t.Logf("Starting alpha %v from group %v", alphaID, group)
 			require.NoError(t, targetCluster.StartAlpha(alphaID))
 		}
 	}
 
 	require.NoError(t, targetCluster.HealthCheck(false))
 
-	fmt.Println("Import completed")
+	if sleep {
+		time.Sleep(1 * time.Minute)
+	}
+
+	t.Log("Import completed")
 
 	for i := 0; i < targetAlphas; i++ {
 		gc, cleanup, err := targetCluster.AlphaClient(i)