mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 04:18:01 +00:00
36 lines
1.5 KiB
Go
36 lines
1.5 KiB
Go
|
package managed
|
||
|
|
|
||
|
|
import (
|
||
|
|
kubernetesmock "github.com/caos/orbos/pkg/kubernetes/mock"
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
core "k8s.io/api/core/v1"
|
||
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func SetClean(
|
||
|
|
k8sClient *kubernetesmock.MockClientInt,
|
||
|
|
namespace string,
|
||
|
|
replicas int,
|
||
|
|
) {
|
||
|
|
k8sClient.EXPECT().ScaleStatefulset(namespace, gomock.Any(), 0).Return(nil)
|
||
|
|
k8sClient.EXPECT().ListPersistentVolumeClaims(namespace).Return(&core.PersistentVolumeClaimList{
|
||
|
|
Items: []core.PersistentVolumeClaim{
|
||
|
|
{ObjectMeta: metav1.ObjectMeta{
|
||
|
|
Name: "datadir-cockroachdb-0",
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
}, nil)
|
||
|
|
k8sClient.EXPECT().ScaleStatefulset(namespace, gomock.Any(), 1).Return(nil)
|
||
|
|
k8sClient.EXPECT().DeletePersistentVolumeClaim(namespace, gomock.Any(), gomock.Any()).Times(replicas).Return(nil)
|
||
|
|
k8sClient.EXPECT().WaitUntilStatefulsetIsReady(namespace, gomock.Any(), true, false, gomock.Any())
|
||
|
|
k8sClient.EXPECT().WaitUntilStatefulsetIsReady(namespace, gomock.Any(), true, true, time.Second*1)
|
||
|
|
k8sClient.EXPECT().WaitUntilStatefulsetIsReady(namespace, gomock.Any(), true, true, gomock.Any())
|
||
|
|
|
||
|
|
/*
|
||
|
|
k8sClient.EXPECT().ApplyJob(gomock.Any()).Times(1).Return(nil)
|
||
|
|
k8sClient.EXPECT().GetJob(namespace, clean.GetJobName(backupName)).Times(1).Return(nil, macherrs.NewNotFound(schema.GroupResource{"batch", "jobs"}, clean.GetJobName(backupName)))
|
||
|
|
k8sClient.EXPECT().WaitUntilJobCompleted(namespace, clean.GetJobName(backupName), gomock.Any()).Times(1).Return(nil)
|
||
|
|
k8sClient.EXPECT().DeleteJob(namespace, clean.GetJobName(backupName)).Times(1).Return(nil)*/
|
||
|
|
}
|