update RequiredNodesAreReady to get nodesNumber as argument

Author: Eslam-Nawara

integrationtests/k8s_test.go

@@ -12,7 +12,7 @@ import (
 )
 
 // RequireNodesAreReady runs `kubectl get node` on the master node and requires that all nodes are ready
-func RequireNodesAreReady(t *testing.T, terraformOptions *terraform.Options, privateKey string) {
+func RequireNodesAreReady(t *testing.T, terraformOptions *terraform.Options, privateKey string, nodesNumber int) {
 	t.Helper()
 
 	masterYggIP := terraform.Output(t, terraformOptions, "mr_ygg_ip")
@@ -24,7 +24,6 @@ func RequireNodesAreReady(t *testing.T, terraformOptions *terraform.Options, pri
 	output = strings.TrimSpace(output)
 	require.NoError(t, err)
 
-	nodesNumber := 2
 	numberOfReadyNodes := strings.Count(output, "Ready")
 	fmt.Println(output)
 	require.True(t, numberOfReadyNodes == nodesNumber, "number of ready nodes is not equal to number of nodes only %d nodes are ready", numberOfReadyNodes)
@@ -84,7 +83,7 @@ func TestK8s(t *testing.T) {
 		require.True(t, ok)
 
 		// ssh to master node
-		RequireNodesAreReady(t, terraformOptions, privateKey)
+		RequireNodesAreReady(t, terraformOptions, privateKey, 2)
 	})
 
 	t.Run("k8s_invalid_names", func(t *testing.T) {

integrationtests/k8s_using_module_test.go

@@ -101,7 +101,7 @@ func TestModuleK8s(t *testing.T) {
 	require.NoError(t, err)
 	defer terraform.Destroy(t, terraformOptions)
 
-	RequireNodesAreReady(t, terraformOptions, privateKey)
+	RequireNodesAreReady(t, terraformOptions, privateKey, 2)
 
 	terraformOptions.Vars["workers"] = []map[string]interface{}{
 		{
@@ -149,5 +149,5 @@ func TestModuleK8s(t *testing.T) {
 	_, err = terraform.ApplyE(t, terraformOptions)
 	require.NoError(t, err)
 
-	RequireNodesAreReady(t, terraformOptions, privateKey)
+	RequireNodesAreReady(t, terraformOptions, privateKey, 3)
 }