120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
//
|
|
// Copyright (c) 2019-2021 Red Hat, Inc.
|
|
// This program and the accompanying materials are made
|
|
// available under the terms of the Eclipse Public License 2.0
|
|
// which is available at https://www.eclipse.org/legal/epl-2.0/
|
|
//
|
|
// SPDX-License-Identifier: EPL-2.0
|
|
//
|
|
// Contributors:
|
|
// Red Hat, Inc. - initial API and implementation
|
|
//
|
|
package deploy
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/eclipse-che/che-operator/pkg/common/chetypes"
|
|
"github.com/eclipse-che/che-operator/pkg/common/test"
|
|
"github.com/stretchr/testify/assert"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/reconcile"
|
|
)
|
|
|
|
type TestReconcilable struct {
|
|
shouldFailReconcileOnce bool
|
|
shouldFailFinalizerOnce bool
|
|
alreadyFailedReconcile bool
|
|
alreadyFailedFinalizer bool
|
|
}
|
|
|
|
func NewTestReconcilable(shouldFailReconcileOnce bool, shouldFailFinalizerOnce bool) *TestReconcilable {
|
|
return &TestReconcilable{
|
|
shouldFailReconcileOnce,
|
|
shouldFailFinalizerOnce,
|
|
false,
|
|
false}
|
|
}
|
|
|
|
func (tr *TestReconcilable) Reconcile(ctx *chetypes.DeployContext) (reconcile.Result, bool, error) {
|
|
// Fails on first invocation passes on others
|
|
if !tr.alreadyFailedReconcile && tr.shouldFailReconcileOnce {
|
|
tr.alreadyFailedReconcile = true
|
|
return reconcile.Result{}, false, fmt.Errorf("reconcile error")
|
|
} else {
|
|
return reconcile.Result{}, true, nil
|
|
}
|
|
}
|
|
|
|
func (tr *TestReconcilable) Finalize(ctx *chetypes.DeployContext) bool {
|
|
// Fails on first invocation passes on others
|
|
if !tr.alreadyFailedFinalizer && tr.shouldFailFinalizerOnce {
|
|
tr.alreadyFailedFinalizer = true
|
|
return false
|
|
} else {
|
|
return true
|
|
}
|
|
}
|
|
|
|
func TestShouldUpdateAndCleanStatus(t *testing.T) {
|
|
deployContext := test.GetDeployContext(nil, []runtime.Object{})
|
|
|
|
tr := NewTestReconcilable(true, false)
|
|
|
|
rm := NewReconcileManager()
|
|
rm.RegisterReconciler(tr)
|
|
|
|
_, done, err := rm.ReconcileAll(deployContext)
|
|
|
|
assert.False(t, done)
|
|
assert.NotNil(t, err)
|
|
assert.NotEmpty(t, deployContext.CheCluster.Status.Reason)
|
|
assert.Equal(t, "Reconciler failed deploy.TestReconcilable, cause: reconcile error", deployContext.CheCluster.Status.Message)
|
|
assert.Equal(t, tr, rm.failedReconciler)
|
|
|
|
_, done, err = rm.ReconcileAll(deployContext)
|
|
|
|
assert.True(t, done)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, deployContext.CheCluster.Status.Reason)
|
|
assert.Empty(t, deployContext.CheCluster.Status.Message)
|
|
assert.Nil(t, rm.failedReconciler)
|
|
}
|
|
|
|
func TestShouldCleanUpAllFinalizers(t *testing.T) {
|
|
ctx := test.GetDeployContext(nil, []runtime.Object{})
|
|
|
|
rm := NewReconcileManager()
|
|
rm.RegisterReconciler(NewTestReconcilable(false, false))
|
|
|
|
_, done, err := rm.ReconcileAll(ctx)
|
|
assert.True(t, done)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 1, len(ctx.CheCluster.Finalizers))
|
|
|
|
done = rm.FinalizeAll(ctx)
|
|
assert.True(t, done)
|
|
assert.Empty(t, ctx.CheCluster.Finalizers)
|
|
}
|
|
|
|
func TestShouldNotCleanUpAllFinalizersIfFailure(t *testing.T) {
|
|
ctx := test.GetDeployContext(nil, []runtime.Object{})
|
|
|
|
rm := NewReconcileManager()
|
|
rm.RegisterReconciler(NewTestReconcilable(false, true))
|
|
|
|
_, done, err := rm.ReconcileAll(ctx)
|
|
assert.True(t, done)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 1, len(ctx.CheCluster.Finalizers))
|
|
|
|
done = rm.FinalizeAll(ctx)
|
|
assert.False(t, done)
|
|
assert.Equal(t, 1, len(ctx.CheCluster.Finalizers))
|
|
|
|
done = rm.FinalizeAll(ctx)
|
|
assert.True(t, done)
|
|
assert.Empty(t, ctx.CheCluster.Finalizers)
|
|
}
|