@@ -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