Skip to content

Commit c085adc

Browse files
committed
test: adjusting unittest
1 parent f324fbe commit c085adc

File tree

3 files changed

+6
-9
lines changed

3 files changed

+6
-9
lines changed

controller/lifecycle/conditions_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"errors"
55
"fmt"
66
"testing"
7+
"time"
78

89
"github.com/stretchr/testify/assert"
910
"github.com/stretchr/testify/require"
@@ -177,7 +178,7 @@ func TestSubroutineCondition(t *testing.T) {
177178
subroutine := changeStatusSubroutine{}
178179

179180
// When
180-
setSubroutineCondition(&condition, subroutine, controllerruntime.Result{Requeue: true}, nil, false, log)
181+
setSubroutineCondition(&condition, subroutine, controllerruntime.Result{RequeueAfter: 1 * time.Second}, nil, false, log)
181182

182183
// Then
183184
assert.Equal(t, 1, len(condition))
@@ -219,7 +220,7 @@ func TestSubroutineCondition(t *testing.T) {
219220
subroutine := changeStatusSubroutine{}
220221

221222
// When
222-
setSubroutineCondition(&condition, subroutine, controllerruntime.Result{Requeue: true}, nil, true, log)
223+
setSubroutineCondition(&condition, subroutine, controllerruntime.Result{RequeueAfter: 1 * time.Second}, nil, true, log)
223224

224225
// Then
225226
assert.Equal(t, 1, len(condition))

controller/lifecycle/finalizerSubroutine_test.go

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ const subroutineFinalizer = "finalizer"
1616
type finalizerSubroutine struct {
1717
client client.Client
1818
err error
19-
requeue bool
2019
requeueAfter time.Duration
2120
}
2221

@@ -30,9 +29,6 @@ func (c finalizerSubroutine) Finalize(_ context.Context, _ RuntimeObject) (contr
3029
if c.err != nil {
3130
return controllerruntime.Result{}, errors.NewOperatorError(c.err, true, true)
3231
}
33-
if c.requeue {
34-
return controllerruntime.Result{Requeue: true}, nil
35-
}
3632
if c.requeueAfter > 0 {
3733
return controllerruntime.Result{RequeueAfter: c.requeueAfter}, nil
3834
}

controller/lifecycle/lifecycle_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,8 @@ func TestLifecycle(t *testing.T) {
133133

134134
mgr, _ := createLifecycleManager([]Subroutine{
135135
finalizerSubroutine{
136-
client: fakeClient,
137-
requeue: true,
136+
client: fakeClient,
137+
requeueAfter: 1 * time.Second,
138138
},
139139
}, fakeClient)
140140

@@ -143,7 +143,7 @@ func TestLifecycle(t *testing.T) {
143143

144144
assert.NoError(t, err)
145145
assert.Equal(t, 1, len(instance.Finalizers))
146-
assert.Equal(t, time.Duration(0), res.RequeueAfter)
146+
assert.Equal(t, time.Duration(1*time.Second), res.RequeueAfter)
147147
})
148148

149149
t.Run("Lifecycle with a finalizer - finalization(requeueAfter)", func(t *testing.T) {

0 commit comments

Comments
 (0)