Skip to content

Commit 124961a

Browse files
committed
Fix UT
1 parent 1cf05e0 commit 124961a

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

pkg/deployment/reconcile/plan_builder_test.go

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ import (
2828
"io/ioutil"
2929
"testing"
3030

31+
policy "k8s.io/api/policy/v1beta1"
32+
3133
"github.com/arangodb/kube-arangodb/pkg/deployment/resources/inspector"
3234

3335
backupApi "github.com/arangodb/kube-arangodb/pkg/apis/backup/v1"
@@ -533,10 +535,12 @@ func TestCreatePlan(t *testing.T) {
533535
ExpectedLog string
534536
ExpectedEvent *k8sutil.Event
535537

536-
Pods map[string]*core.Pod
537-
Secrets map[string]*core.Secret
538-
Services map[string]*core.Service
539-
PVCS map[string]*core.PersistentVolumeClaim
538+
Pods map[string]*core.Pod
539+
Secrets map[string]*core.Secret
540+
Services map[string]*core.Service
541+
PVCS map[string]*core.PersistentVolumeClaim
542+
ServiceAccounts map[string]*core.ServiceAccount
543+
PDBS map[string]*policy.PodDisruptionBudget
540544
}{
541545
{
542546
Name: "Can not create plan for single deployment",
@@ -796,7 +800,7 @@ func TestCreatePlan(t *testing.T) {
796800
if testCase.Helper != nil {
797801
testCase.Helper(testCase.context.ArangoDeployment)
798802
}
799-
err, _ := r.CreatePlan(ctx, inspector.NewInspectorFromData(testCase.Pods, testCase.Secrets, testCase.PVCS, testCase.Services))
803+
err, _ := r.CreatePlan(ctx, inspector.NewInspectorFromData(testCase.Pods, testCase.Secrets, testCase.PVCS, testCase.Services, testCase.ServiceAccounts, testCase.PDBS))
800804

801805
// Assert
802806
if testCase.ExpectedEvent != nil {

0 commit comments

Comments
 (0)