@@ -38,7 +38,8 @@ type FakeImageService struct {
38
38
39
39
pulledImages []* pulledImage
40
40
41
- FakeFilesystemUsage []* runtimeapi.FilesystemUsage
41
+ FakeFilesystemUsage []* runtimeapi.FilesystemUsage
42
+ FakeContainerFilesystemUsage []* runtimeapi.FilesystemUsage
42
43
}
43
44
44
45
// SetFakeImages sets the list of fake images for the FakeImageService.
@@ -93,6 +94,14 @@ func (r *FakeImageService) SetFakeFilesystemUsage(usage []*runtimeapi.Filesystem
93
94
r .FakeFilesystemUsage = usage
94
95
}
95
96
97
+ // SetFakeFilesystemUsage sets the FilesystemUsage for FakeImageService.
98
+ func (r * FakeImageService ) SetFakeContainerFilesystemUsage (usage []* runtimeapi.FilesystemUsage ) {
99
+ r .Lock ()
100
+ defer r .Unlock ()
101
+
102
+ r .FakeContainerFilesystemUsage = usage
103
+ }
104
+
96
105
// NewFakeImageService creates a new FakeImageService.
97
106
func NewFakeImageService () * FakeImageService {
98
107
return & FakeImageService {
@@ -218,7 +227,7 @@ func (r *FakeImageService) RemoveImage(_ context.Context, image *runtimeapi.Imag
218
227
}
219
228
220
229
// ImageFsInfo returns information of the filesystem that is used to store images.
221
- func (r * FakeImageService ) ImageFsInfo (_ context.Context ) ([] * runtimeapi.FilesystemUsage , error ) {
230
+ func (r * FakeImageService ) ImageFsInfo (_ context.Context ) (* runtimeapi.ImageFsInfoResponse , error ) {
222
231
r .Lock ()
223
232
defer r .Unlock ()
224
233
@@ -227,7 +236,10 @@ func (r *FakeImageService) ImageFsInfo(_ context.Context) ([]*runtimeapi.Filesys
227
236
return nil , err
228
237
}
229
238
230
- return r .FakeFilesystemUsage , nil
239
+ return & runtimeapi.ImageFsInfoResponse {
240
+ ImageFilesystems : r .FakeFilesystemUsage ,
241
+ ContainerFilesystems : r .FakeContainerFilesystemUsage ,
242
+ }, nil
231
243
}
232
244
233
245
// AssertImagePulledWithAuth validates whether the image was pulled with auth and asserts if it wasn't.
0 commit comments