Fix unit tests

kubernetes-sigs · Mar 4, 2024 · 15025ef · 15025ef
1 parent c10e38c
commit 15025ef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/azure/services/agentpools/agentpools_test.go b/azure/services/agentpools/agentpools_test.go
@@ -46,6 +46,7 @@ func TestPostCreateOrUpdateResourceHook(t *testing.T) {
 		scope := mock_agentpools.NewMockAgentPoolScope(mockCtrl)
 
 		scope.EXPECT().RemoveCAPIMachinePoolAnnotation(clusterv1.ReplicasManagedByAnnotation)
+		scope.EXPECT().IsPreviewEnabled().Return(false)
 
 		managedCluster := &asocontainerservicev1.ManagedClustersAgentPool{
 			Status: asocontainerservicev1.ManagedClusters_AgentPool_STATUS{
@@ -64,6 +65,7 @@ func TestPostCreateOrUpdateResourceHook(t *testing.T) {
 
 		scope.EXPECT().SetCAPIMachinePoolAnnotation(clusterv1.ReplicasManagedByAnnotation, "true")
 		scope.EXPECT().SetCAPIMachinePoolReplicas(ptr.To(1234))
+		scope.EXPECT().IsPreviewEnabled().Return(false)
 
 		managedCluster := &asocontainerservicev1.ManagedClustersAgentPool{
 			Status: asocontainerservicev1.ManagedClusters_AgentPool_STATUS{

diff --git a/controllers/azuremanagedmachinepool_controller_test.go b/controllers/azuremanagedmachinepool_controller_test.go
@@ -72,6 +72,7 @@ func TestAzureManagedMachinePoolReconcile(t *testing.T) {
 				agentpools.SetAgentPoolProviderIDList(providerIDs)
 				agentpools.SetAgentPoolReplicas(int32(len(providerIDs))).Return()
 				agentpools.SetAgentPoolReady(true).Return()
+				agentpools.IsPreviewEnabled().Return(false)
 
 				nodelister.List(gomock2.AContext(), "fake-rg").Return(fakeVirtualMachineScaleSet, nil)
 				nodelister.ListInstances(gomock2.AContext(), "fake-rg", "vmssName").Return(fakeVirtualMachineScaleSetVM, nil)