mirror of
https://github.com/stakater/Reloader.git
synced 2026-08-20 12:36:26 +00:00
119 lines
3.6 KiB
Go
119 lines
3.6 KiB
Go
package controller_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/go-logr/logr/testr"
|
|
"github.com/stakater/Reloader/internal/pkg/alerting"
|
|
"github.com/stakater/Reloader/internal/pkg/config"
|
|
"github.com/stakater/Reloader/internal/pkg/controller"
|
|
"github.com/stakater/Reloader/internal/pkg/events"
|
|
"github.com/stakater/Reloader/internal/pkg/metrics"
|
|
"github.com/stakater/Reloader/internal/pkg/reload"
|
|
"github.com/stakater/Reloader/internal/pkg/testutil"
|
|
"github.com/stakater/Reloader/internal/pkg/webhook"
|
|
"github.com/stakater/Reloader/internal/pkg/workload"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/fake"
|
|
)
|
|
|
|
// newConfigMapReconciler creates a ConfigMapReconciler for testing.
|
|
func newConfigMapReconciler(t *testing.T, cfg *config.Config, objects ...runtime.Object) *controller.ConfigMapReconciler {
|
|
t.Helper()
|
|
fakeClient := fake.NewClientBuilder().
|
|
WithScheme(testutil.NewScheme()).
|
|
WithRuntimeObjects(objects...).
|
|
Build()
|
|
|
|
collectors := metrics.NewCollectors()
|
|
|
|
return &controller.ConfigMapReconciler{
|
|
Client: fakeClient,
|
|
Log: testr.New(t),
|
|
Config: cfg,
|
|
ReloadService: reload.NewService(cfg),
|
|
Registry: workload.NewRegistry(cfg.ArgoRolloutsEnabled),
|
|
Collectors: &collectors,
|
|
EventRecorder: events.NewRecorder(nil),
|
|
WebhookClient: webhook.NewClient("", testr.New(t)),
|
|
Alerter: &alerting.NoOpAlerter{},
|
|
}
|
|
}
|
|
|
|
// newSecretReconciler creates a SecretReconciler for testing.
|
|
func newSecretReconciler(t *testing.T, cfg *config.Config, objects ...runtime.Object) *controller.SecretReconciler {
|
|
t.Helper()
|
|
fakeClient := fake.NewClientBuilder().
|
|
WithScheme(testutil.NewScheme()).
|
|
WithRuntimeObjects(objects...).
|
|
Build()
|
|
|
|
collectors := metrics.NewCollectors()
|
|
|
|
return &controller.SecretReconciler{
|
|
Client: fakeClient,
|
|
Log: testr.New(t),
|
|
Config: cfg,
|
|
ReloadService: reload.NewService(cfg),
|
|
Registry: workload.NewRegistry(cfg.ArgoRolloutsEnabled),
|
|
Collectors: &collectors,
|
|
EventRecorder: events.NewRecorder(nil),
|
|
WebhookClient: webhook.NewClient("", testr.New(t)),
|
|
Alerter: &alerting.NoOpAlerter{},
|
|
}
|
|
}
|
|
|
|
// newNamespaceReconciler creates a NamespaceReconciler for testing.
|
|
func newNamespaceReconciler(t *testing.T, cfg *config.Config, cache *controller.NamespaceCache, objects ...runtime.Object) *controller.NamespaceReconciler {
|
|
t.Helper()
|
|
scheme := runtime.NewScheme()
|
|
_ = corev1.AddToScheme(scheme)
|
|
|
|
fakeClient := fake.NewClientBuilder().
|
|
WithScheme(scheme).
|
|
WithRuntimeObjects(objects...).
|
|
Build()
|
|
|
|
return &controller.NamespaceReconciler{
|
|
Client: fakeClient,
|
|
Log: testr.New(t),
|
|
Config: cfg,
|
|
Cache: cache,
|
|
}
|
|
}
|
|
|
|
// reconcileRequest creates a ctrl.Request for the given name and namespace.
|
|
func reconcileRequest(name, namespace string) ctrl.Request {
|
|
return ctrl.Request{
|
|
NamespacedName: types.NamespacedName{
|
|
Name: name,
|
|
Namespace: namespace,
|
|
},
|
|
}
|
|
}
|
|
|
|
// namespaceRequest creates a ctrl.Request for a namespace (no namespace field needed).
|
|
func namespaceRequest(name string) ctrl.Request {
|
|
return ctrl.Request{
|
|
NamespacedName: types.NamespacedName{Name: name},
|
|
}
|
|
}
|
|
|
|
// assertReconcileSuccess runs reconcile and asserts no error and no requeue.
|
|
func assertReconcileSuccess(t *testing.T, reconciler interface {
|
|
Reconcile(context.Context, ctrl.Request) (ctrl.Result, error)
|
|
}, req ctrl.Request) {
|
|
t.Helper()
|
|
result, err := reconciler.Reconcile(context.Background(), req)
|
|
if err != nil {
|
|
t.Fatalf("Reconcile failed: %v", err)
|
|
}
|
|
if result.RequeueAfter > 0 {
|
|
t.Error("Should not requeue")
|
|
}
|
|
}
|