/* Copyright 2017 The Kubernetes Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ package volumebinding import ( "context" v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/util/sets" "k8s.io/klog/v2" ) // FakeVolumeBinderConfig holds configurations for fake volume binder. type FakeVolumeBinderConfig struct { AllBound bool FindReasons ConflictReasons FindErr error AssumeErr error BindErr error } // NewFakeVolumeBinder sets up all the caches needed for the scheduler to make // topology-aware volume binding decisions. func NewFakeVolumeBinder(config *FakeVolumeBinderConfig) *FakeVolumeBinder { return &FakeVolumeBinder{ config: config, } } // FakeVolumeBinder represents a fake volume binder for testing. type FakeVolumeBinder struct { config *FakeVolumeBinderConfig AssumeCalled bool BindCalled bool } var _ SchedulerVolumeBinder = &FakeVolumeBinder{} // GetPodVolumeClaims implements SchedulerVolumeBinder.GetPodVolumes. func (b *FakeVolumeBinder) GetPodVolumeClaims(_ klog.Logger, pod *v1.Pod) (podVolumeClaims *PodVolumeClaims, err error) { return &PodVolumeClaims{}, nil } // GetEligibleNodes implements SchedulerVolumeBinder.GetEligibleNodes. func (b *FakeVolumeBinder) GetEligibleNodes(_ klog.Logger, boundClaims []*v1.PersistentVolumeClaim) (eligibleNodes sets.Set[string]) { return nil } // FindPodVolumes implements SchedulerVolumeBinder.FindPodVolumes. func (b *FakeVolumeBinder) FindPodVolumes(_ klog.Logger, pod *v1.Pod, _ *PodVolumeClaims, node *v1.Node) (podVolumes *PodVolumes, reasons ConflictReasons, err error) { return nil, b.config.FindReasons, b.config.FindErr } // AssumePodVolumes implements SchedulerVolumeBinder.AssumePodVolumes. func (b *FakeVolumeBinder) AssumePodVolumes(_ klog.Logger, assumedPod *v1.Pod, nodeName string, podVolumes *PodVolumes) (bool, error) { b.AssumeCalled = true return b.config.AllBound, b.config.AssumeErr } // RevertAssumedPodVolumes implements SchedulerVolumeBinder.RevertAssumedPodVolumes func (b *FakeVolumeBinder) RevertAssumedPodVolumes(_ *PodVolumes) {} // BindPodVolumes implements SchedulerVolumeBinder.BindPodVolumes. func (b *FakeVolumeBinder) BindPodVolumes(ctx context.Context, assumedPod *v1.Pod, podVolumes *PodVolumes) error { b.BindCalled = true return b.config.BindErr }