package clean

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))
}