update tests

pkg/mic/mic_test.go

@@ -6,8 +6,6 @@ import (
 	"testing"
 	"time"
 
-	"fmt"
-
 	aadpodid "github.com/Azure/aad-pod-identity/pkg/apis/aadpodidentity/v1"
 	"github.com/Azure/aad-pod-identity/pkg/config"
 
@@ -163,11 +161,9 @@ func (c *TestCloudClient) CompareMSI(nodeName string, userIDs []string) bool {
 
 func (c *TestCloudClient) PrintMSI() {
 	for key, val := range c.ListMSI() {
-		fmt.Printf("\nNode name: %s\n", key)
 		glog.Infof("\nNode name: %s", key)
 		if val != nil {
 			for i, id := range *val {
-				fmt.Printf("%d) %s\n", i, id)
 				glog.Infof("%d) %s", i, id)
 			}
 		}

test/e2e/aadpodidentity_test.go

@@ -231,7 +231,10 @@ var _ = Describe("Kubernetes cluster using aad-pod-identity", func() {
 			identityValidatorName := fmt.Sprintf("identity-validator-%d", i)
 
 			setUpIdentityAndDeployment(keyvaultIdentity, fmt.Sprintf("%d", i))
-			time.Sleep(5 * time.Second)
+
+			ok, err := azureassignedidentity.WaitOnLengthMatched(i + 1)
+			Expect(ok).To(Equal(true))
+			Expect(err).NotTo(HaveOccurred())
 
 			identityClientID, err := azure.GetIdentityClientID(cfg.ResourceGroup, identityName)
 			Expect(err).NotTo(HaveOccurred())
@@ -530,6 +533,7 @@ var _ = Describe("Kubernetes cluster using aad-pod-identity", func() {
 func setupInfra() {
 	// Install CRDs and deploy MIC and NMI
 	err := infra.CreateInfra("default", cfg.Registry, cfg.NMIVersion, cfg.MICVersion, templateOutputPath)
+
 	Expect(err).NotTo(HaveOccurred())
 
 	ok, err := daemonset.WaitOnReady(nmiDaemonSet)