mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-16 12:58:00 +00:00
41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
|
package backup
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/caos/orbos/mntr"
|
||
|
|
kubernetesmock "github.com/caos/orbos/pkg/kubernetes/mock"
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBackup_Cleanup1(t *testing.T) {
|
||
|
|
client := kubernetesmock.NewMockClientInt(gomock.NewController(t))
|
||
|
|
monitor := mntr.Monitor{}
|
||
|
|
name := "test"
|
||
|
|
namespace := "testNs"
|
||
|
|
|
||
|
|
cleanupFunc := getCleanupFunc(monitor, namespace, name)
|
||
|
|
client.EXPECT().WaitUntilJobCompleted(namespace, name, timeout).Times(1).Return(nil)
|
||
|
|
client.EXPECT().DeleteJob(namespace, name).Times(1)
|
||
|
|
assert.NoError(t, cleanupFunc(client))
|
||
|
|
|
||
|
|
client.EXPECT().WaitUntilJobCompleted(namespace, name, timeout).Times(1).Return(errors.New("fail"))
|
||
|
|
assert.Error(t, cleanupFunc(client))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackup_Cleanup2(t *testing.T) {
|
||
|
|
client := kubernetesmock.NewMockClientInt(gomock.NewController(t))
|
||
|
|
monitor := mntr.Monitor{}
|
||
|
|
name := "test2"
|
||
|
|
namespace := "testNs2"
|
||
|
|
|
||
|
|
cleanupFunc := getCleanupFunc(monitor, namespace, name)
|
||
|
|
client.EXPECT().WaitUntilJobCompleted(namespace, name, timeout).Times(1).Return(nil)
|
||
|
|
client.EXPECT().DeleteJob(namespace, name).Times(1)
|
||
|
|
assert.NoError(t, cleanupFunc(client))
|
||
|
|
|
||
|
|
client.EXPECT().WaitUntilJobCompleted(namespace, name, timeout).Times(1).Return(errors.New("fail"))
|
||
|
|
assert.Error(t, cleanupFunc(client))
|
||
|
|
}
|