operator: stub etcd remove calls in env tests

This commit is contained in:
Leonard Cohnen 2024-10-20 22:19:30 +02:00 committed by Markus Rudy
parent 3ddd6389e3
commit d6f9391c75
2 changed files with 11 additions and 0 deletions

View File

@ -891,3 +891,11 @@ func (*unimplementedNodeReplacer) CreateNode(_ context.Context, _ string) (nodeN
func (*unimplementedNodeReplacer) DeleteNode(_ context.Context, _ string) error { func (*unimplementedNodeReplacer) DeleteNode(_ context.Context, _ string) error {
panic("unimplemented") panic("unimplemented")
} }
type stubEtcdRemover struct {
deleteErr error
}
func (r *stubEtcdRemover) RemoveEtcdMemberFromCluster(_ context.Context, _ string) error {
return r.deleteErr
}

View File

@ -123,6 +123,7 @@ var _ = BeforeSuite(func() {
err = (&NodeVersionReconciler{ err = (&NodeVersionReconciler{
kubernetesServerVersionGetter: fakes.k8sVerGetter, kubernetesServerVersionGetter: fakes.k8sVerGetter,
nodeReplacer: fakes.nodeReplacer, nodeReplacer: fakes.nodeReplacer,
etcdRemover: fakes.etcdRemover,
Client: k8sManager.GetClient(), Client: k8sManager.GetClient(),
Scheme: k8sManager.GetScheme(), Scheme: k8sManager.GetScheme(),
}).SetupWithManager(k8sManager) }).SetupWithManager(k8sManager)
@ -147,6 +148,7 @@ type fakeCollection struct {
nodeStateGetter *stubNodeStateGetter nodeStateGetter *stubNodeStateGetter
nodeReplacer *stubNodeReplacer nodeReplacer *stubNodeReplacer
k8sVerGetter *stubKubernetesServerVersionGetter k8sVerGetter *stubKubernetesServerVersionGetter
etcdRemover *stubEtcdRemover
clock *testclock.FakeClock clock *testclock.FakeClock
} }
@ -162,6 +164,7 @@ func newFakes() fakeCollection {
nodeStateGetter: &stubNodeStateGetter{}, nodeStateGetter: &stubNodeStateGetter{},
nodeReplacer: &stubNodeReplacer{}, nodeReplacer: &stubNodeReplacer{},
k8sVerGetter: &stubKubernetesServerVersionGetter{}, k8sVerGetter: &stubKubernetesServerVersionGetter{},
etcdRemover: &stubEtcdRemover{},
clock: testclock.NewFakeClock(time.Now()), clock: testclock.NewFakeClock(time.Now()),
} }
} }