From eb21d5aaba406a8307cf4089f205e02c00f61707 Mon Sep 17 00:00:00 2001
From: Wenqi Qiu <wenqiq@vmware.com>
Date: Mon, 8 Jul 2024 11:08:02 +0800
Subject: [PATCH] fix unit test

Signed-off-by: Wenqi Qiu <wenqiq@vmware.com>
---
 pkg/nsx/services/nsxserviceaccount/cluster_test.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pkg/nsx/services/nsxserviceaccount/cluster_test.go b/pkg/nsx/services/nsxserviceaccount/cluster_test.go
index 985218534..fbf7d2f7a 100644
--- 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
 			},