From 36f7bc76bf5fd40b18db8168c57448033daa72ac Mon Sep 17 00:00:00 2001 From: Michael Taufen Date: Fri, 22 Jun 2018 19:30:02 -0700 Subject: [PATCH] extend FakeVolumePlugin to implement VolumePluginWithAttachLimits interface --- pkg/volume/testing/testing.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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 { }