Skip to content

Commit 849f7ee

Browse files
committed
controllers: replace logr.NewContext shims
Signed-off-by: Hidde Beydals <[email protected]>
1 parent ad0993e commit 849f7ee

File tree

3 files changed

+19
-35
lines changed

3 files changed

+19
-35
lines changed

controllers/bucket_controller_test.go

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@ import (
3232
"time"
3333

3434
"github.com/darkowlzz/controller-check/status"
35-
"github.com/go-logr/logr"
35+
"github.com/fluxcd/pkg/apis/meta"
36+
"github.com/fluxcd/pkg/runtime/conditions"
3637
. "github.com/onsi/gomega"
3738
raw "google.golang.org/api/storage/v1"
3839
corev1 "k8s.io/api/core/v1"
@@ -41,10 +42,6 @@ import (
4142
"k8s.io/client-go/tools/record"
4243
"sigs.k8s.io/controller-runtime/pkg/client"
4344
fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake"
44-
"sigs.k8s.io/controller-runtime/pkg/log"
45-
46-
"github.com/fluxcd/pkg/apis/meta"
47-
"github.com/fluxcd/pkg/runtime/conditions"
4845

4946
sourcev1 "github.com/fluxcd/source-controller/api/v1beta2"
5047
sreconcile "github.com/fluxcd/source-controller/internal/reconcile"
@@ -959,9 +956,7 @@ func TestBucketReconciler_reconcileArtifact(t *testing.T) {
959956
tt.beforeFunc(g, obj, artifact, tmpDir)
960957
}
961958

962-
dlog := log.NewDelegatingLogSink(log.NullLogSink{})
963-
nullLogger := logr.New(dlog)
964-
got, err := r.reconcileArtifact(logr.NewContext(ctx, nullLogger), obj, &artifact, tmpDir)
959+
got, err := r.reconcileArtifact(context.TODO(), obj, &artifact, tmpDir)
965960
g.Expect(err != nil).To(Equal(tt.wantErr))
966961
g.Expect(got).To(Equal(tt.want))
967962

controllers/gitrepository_controller_test.go

Lines changed: 10 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ limitations under the License.
1717
package controllers
1818

1919
import (
20+
"context"
2021
"fmt"
2122
"net/url"
2223
"os"
@@ -26,13 +27,17 @@ import (
2627
"time"
2728

2829
"github.com/darkowlzz/controller-check/status"
30+
"github.com/fluxcd/pkg/apis/meta"
31+
"github.com/fluxcd/pkg/gittestserver"
32+
"github.com/fluxcd/pkg/runtime/conditions"
33+
"github.com/fluxcd/pkg/ssh"
34+
"github.com/fluxcd/pkg/testserver"
2935
"github.com/go-git/go-billy/v5/memfs"
3036
gogit "github.com/go-git/go-git/v5"
3137
"github.com/go-git/go-git/v5/config"
3238
"github.com/go-git/go-git/v5/plumbing"
3339
"github.com/go-git/go-git/v5/plumbing/object"
3440
"github.com/go-git/go-git/v5/storage/memory"
35-
"github.com/go-logr/logr"
3641
. "github.com/onsi/gomega"
3742
sshtestdata "golang.org/x/crypto/ssh/testdata"
3843
corev1 "k8s.io/api/core/v1"
@@ -45,13 +50,6 @@ import (
4550
"sigs.k8s.io/controller-runtime/pkg/client"
4651
fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake"
4752
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
48-
"sigs.k8s.io/controller-runtime/pkg/log"
49-
50-
"github.com/fluxcd/pkg/apis/meta"
51-
"github.com/fluxcd/pkg/gittestserver"
52-
"github.com/fluxcd/pkg/runtime/conditions"
53-
"github.com/fluxcd/pkg/ssh"
54-
"github.com/fluxcd/pkg/testserver"
5553

5654
sourcev1 "github.com/fluxcd/source-controller/api/v1beta2"
5755
sreconcile "github.com/fluxcd/source-controller/internal/reconcile"
@@ -499,9 +497,8 @@ func TestGitRepositoryReconciler_reconcileSource_authStrategy(t *testing.T) {
499497

500498
var artifact sourcev1.Artifact
501499
var includes artifactSet
502-
dlog := log.NewDelegatingLogSink(log.NullLogSink{})
503-
nullLogger := logr.New(dlog)
504-
got, err := r.reconcileSource(logr.NewContext(ctx, nullLogger), obj, &artifact, &includes, tmpDir)
500+
501+
got, err := r.reconcileSource(context.TODO(), obj, &artifact, &includes, tmpDir)
505502
g.Expect(obj.Status.Conditions).To(conditions.MatchConditions(tt.assertConditions))
506503
g.Expect(err != nil).To(Equal(tt.wantErr))
507504
g.Expect(got).To(Equal(tt.want))
@@ -1209,9 +1206,7 @@ func TestGitRepositoryReconciler_verifyCommitSignature(t *testing.T) {
12091206
tt.beforeFunc(obj)
12101207
}
12111208

1212-
dlog := log.NewDelegatingLogSink(log.NullLogSink{})
1213-
nullLogger := logr.New(dlog)
1214-
got, err := r.verifyCommitSignature(logr.NewContext(ctx, nullLogger), obj, tt.commit)
1209+
got, err := r.verifyCommitSignature(context.TODO(), obj, tt.commit)
12151210
g.Expect(obj.Status.Conditions).To(conditions.MatchConditions(tt.assertConditions))
12161211
g.Expect(err != nil).To(Equal(tt.wantErr))
12171212
g.Expect(got).To(Equal(tt.want))
@@ -1335,9 +1330,7 @@ func TestGitRepositoryReconciler_ConditionsUpdate(t *testing.T) {
13351330
}
13361331

13371332
key := client.ObjectKeyFromObject(obj)
1338-
dlog := log.NewDelegatingLogSink(log.NullLogSink{})
1339-
nullLogger := logr.New(dlog)
1340-
res, err := r.Reconcile(logr.NewContext(ctx, nullLogger), ctrl.Request{NamespacedName: key})
1333+
res, err := r.Reconcile(context.TODO(), ctrl.Request{NamespacedName: key})
13411334
g.Expect(err != nil).To(Equal(tt.wantErr))
13421335
g.Expect(res).To(Equal(tt.want))
13431336

controllers/helmrepository_controller_test.go

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -27,20 +27,17 @@ import (
2727
"time"
2828

2929
"github.com/darkowlzz/controller-check/status"
30-
"github.com/go-logr/logr"
30+
"github.com/fluxcd/pkg/apis/meta"
31+
"github.com/fluxcd/pkg/helmtestserver"
32+
"github.com/fluxcd/pkg/runtime/conditions"
33+
"github.com/fluxcd/pkg/runtime/patch"
3134
. "github.com/onsi/gomega"
3235
corev1 "k8s.io/api/core/v1"
3336
apierrors "k8s.io/apimachinery/pkg/api/errors"
3437
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3538
"k8s.io/client-go/tools/record"
3639
"sigs.k8s.io/controller-runtime/pkg/client"
3740
fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake"
38-
"sigs.k8s.io/controller-runtime/pkg/log"
39-
40-
"github.com/fluxcd/pkg/apis/meta"
41-
"github.com/fluxcd/pkg/helmtestserver"
42-
"github.com/fluxcd/pkg/runtime/conditions"
43-
"github.com/fluxcd/pkg/runtime/patch"
4441

4542
sourcev1 "github.com/fluxcd/source-controller/api/v1beta2"
4643
serror "github.com/fluxcd/source-controller/internal/error"
@@ -596,9 +593,8 @@ func TestHelmRepositoryReconciler_reconcileArtifact(t *testing.T) {
596593
if tt.beforeFunc != nil {
597594
tt.beforeFunc(g, obj, artifact, chartRepo)
598595
}
599-
dlog := log.NewDelegatingLogSink(log.NullLogSink{})
600-
nullLogger := logr.New(dlog)
601-
got, err := r.reconcileArtifact(logr.NewContext(ctx, nullLogger), obj, &artifact, chartRepo)
596+
597+
got, err := r.reconcileArtifact(context.TODO(), obj, &artifact, chartRepo)
602598
g.Expect(err != nil).To(Equal(tt.wantErr))
603599
g.Expect(got).To(Equal(tt.want))
604600

0 commit comments

Comments
 (0)