fix unit test

Signed-off-by: Wenqi Qiu <[email protected]>
vmware-tanzu · Jul 8, 2024 · f827a04 · f827a04
1 parent 7f1dd09
commit f827a04
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/nsx/services/nsxserviceaccount/cluster_test.go b/pkg/nsx/services/nsxserviceaccount/cluster_test.go
@@ -310,6 +310,7 @@ func TestNSXServiceAccountService_CreateOrUpdateNSXServiceAccount(t *testing.T)
 						ResourceType: &antreaClusterResourceType,
 						Certificate:  nil,
 						VhcPath:      &vpcPath,
+						NodeType:     &nodeTypeAntrea,
 						Tags: []model.Tag{{
 							Scope: &tagScopeCluster,
 							Tag:   &s.NSXConfig.CoeConfig.Cluster,
@@ -328,7 +329,7 @@ func TestNSXServiceAccountService_CreateOrUpdateNSXServiceAccount(t *testing.T)
 				}})
 				patches.ApplyMethodSeq(s.NSXClient, "NSXCheckVersion", []gomonkey.OutputCell{{
 					Values: gomonkey.Params{true},
-					Times:  1,
+					Times:  2,
 				}})
 				return patches
 			},
@@ -446,7 +447,7 @@ func TestNSXServiceAccountService_CreateOrUpdateNSXServiceAccount(t *testing.T)
 				}})
 				patches.ApplyMethodSeq(s.NSXClient, "NSXCheckVersion", []gomonkey.OutputCell{{
 					Values: gomonkey.Params{true},
-					Times:  1,
+					Times:  2,
 				}})
 				return patches
 			},
@@ -888,6 +889,10 @@ func TestNSXServiceAccountService_RestoreRealizedNSXServiceAccount(t *testing.T)
 					}, nil},
 					Times: 1,
 				}})
+				patches.ApplyMethodSeq(s.NSXClient, "NSXCheckVersion", []gomonkey.OutputCell{{
+					Values: gomonkey.Params{true},
+					Times:  1,
+				}})
 				return patches
 			},
 			args: args{