diff --git a/pkg/volume/testing/testing.go b/pkg/volume/testing/testing.go index c102e551957..17aa4106351 100644 --- a/pkg/volume/testing/testing.go +++ b/pkg/volume/testing/testing.go @@ -223,6 +223,9 @@ type FakeVolumePlugin struct { LastProvisionerOptions VolumeOptions NewAttacherCallCount int NewDetacherCallCount int + VolumeLimits map[string]int64 + VolumeLimitsError error + LimitKey string Mounters []*FakeVolume Unmounters []*FakeVolume @@ -238,6 +241,7 @@ var _ RecyclableVolumePlugin = &FakeVolumePlugin{} var _ DeletableVolumePlugin = &FakeVolumePlugin{} var _ ProvisionableVolumePlugin = &FakeVolumePlugin{} var _ AttachableVolumePlugin = &FakeVolumePlugin{} +var _ VolumePluginWithAttachLimits = &FakeVolumePlugin{} func (plugin *FakeVolumePlugin) getFakeVolume(list *[]*FakeVolume) *FakeVolume { volume := &FakeVolume{} @@ -448,6 +452,14 @@ func (plugin *FakeVolumePlugin) RequiresFSResize() bool { return true } +func (plugin *FakeVolumePlugin) GetVolumeLimits() (map[string]int64, error) { + return plugin.VolumeLimits, plugin.VolumeLimitsError +} + +func (plugin *FakeVolumePlugin) VolumeLimitKey(spec *Spec) string { + return plugin.LimitKey +} + type FakeFileVolumePlugin struct { }