@@ -66,22 +66,27 @@ func (m *mockPodKiller) killPodNow(pod *v1.Pod, evict bool, gracePeriodOverride
66
66
67
67
// mockDiskInfoProvider is used to simulate testing.
68
68
type mockDiskInfoProvider struct {
69
- dedicatedImageFs * bool
69
+ dedicatedImageFs * bool
70
+ dedicatedContainerFs * bool
70
71
}
71
72
72
73
// HasDedicatedImageFs returns the mocked value
73
74
func (m * mockDiskInfoProvider ) HasDedicatedImageFs (_ context.Context ) (bool , error ) {
74
75
return ptr .Deref (m .dedicatedImageFs , false ), nil
75
76
}
76
77
78
+ // HasDedicatedContainerFs returns the mocked value
79
+ func (m * mockDiskInfoProvider ) HasDedicatedContainerFs (_ context.Context ) (bool , error ) {
80
+ return ptr .Deref (m .dedicatedContainerFs , false ), nil
81
+ }
82
+
77
83
// mockDiskGC is used to simulate invoking image and container garbage collection.
78
84
type mockDiskGC struct {
79
- err error
80
- imageGCInvoked bool
81
- containerGCInvoked bool
82
- readAndWriteSeparate bool
83
- fakeSummaryProvider * fakeSummaryProvider
84
- summaryAfterGC * statsapi.Summary
85
+ err error
86
+ imageGCInvoked bool
87
+ containerGCInvoked bool
88
+ fakeSummaryProvider * fakeSummaryProvider
89
+ summaryAfterGC * statsapi.Summary
85
90
}
86
91
87
92
// DeleteUnusedImages returns the mocked values.
@@ -102,10 +107,6 @@ func (m *mockDiskGC) DeleteAllUnusedContainers(_ context.Context) error {
102
107
return m .err
103
108
}
104
109
105
- func (m * mockDiskGC ) IsContainerFsSeparateFromImageFs (_ context.Context ) bool {
106
- return m .readAndWriteSeparate
107
- }
108
-
109
110
func makePodWithMemoryStats (name string , priority int32 , requests v1.ResourceList , limits v1.ResourceList , memoryWorkingSet string ) (* v1.Pod , statsapi.PodStats ) {
110
111
pod := newPod (name , priority , []v1.Container {
111
112
newContainer (name , requests , limits ),
@@ -384,7 +385,7 @@ func TestPIDPressure_VerifyPodStatus(t *testing.T) {
384
385
385
386
fakeClock := testingclock .NewFakeClock (time .Now ())
386
387
podKiller := & mockPodKiller {}
387
- diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : ptr .To (false )}
388
+ diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : ptr .To (false ), dedicatedContainerFs : ptr . To ( false ) }
388
389
diskGC := & mockDiskGC {err : nil }
389
390
nodeRef := & v1.ObjectReference {Kind : "Node" , Name : "test" , UID : types .UID ("test" ), Namespace : "" }
390
391
@@ -562,8 +563,8 @@ func TestDiskPressureNodeFs_VerifyPodStatus(t *testing.T) {
562
563
563
564
fakeClock := testingclock .NewFakeClock (time .Now ())
564
565
podKiller := & mockPodKiller {}
565
- diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs }
566
- diskGC := & mockDiskGC {err : nil , readAndWriteSeparate : tc . writeableSeparateFromReadOnly }
566
+ diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs , dedicatedContainerFs : & tc . writeableSeparateFromReadOnly }
567
+ diskGC := & mockDiskGC {err : nil }
567
568
nodeRef := & v1.ObjectReference {Kind : "Node" , Name : "test" , UID : types .UID ("test" ), Namespace : "" }
568
569
569
570
config := Config {
@@ -1307,8 +1308,8 @@ func TestDiskPressureNodeFs(t *testing.T) {
1307
1308
1308
1309
fakeClock := testingclock .NewFakeClock (time .Now ())
1309
1310
podKiller := & mockPodKiller {}
1310
- diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs }
1311
- diskGC := & mockDiskGC {err : nil , readAndWriteSeparate : tc . writeableSeparateFromReadOnly }
1311
+ diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs , dedicatedContainerFs : & tc . writeableSeparateFromReadOnly }
1312
+ diskGC := & mockDiskGC {err : nil }
1312
1313
nodeRef := & v1.ObjectReference {Kind : "Node" , Name : "test" , UID : types .UID ("test" ), Namespace : "" }
1313
1314
1314
1315
config := Config {
@@ -1829,7 +1830,7 @@ func TestNodeReclaimFuncs(t *testing.T) {
1829
1830
1830
1831
fakeClock := testingclock .NewFakeClock (time .Now ())
1831
1832
podKiller := & mockPodKiller {}
1832
- diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs }
1833
+ diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs , dedicatedContainerFs : & tc . writeableSeparateFromReadOnly }
1833
1834
nodeRef := & v1.ObjectReference {Kind : "Node" , Name : "test" , UID : types .UID ("test" ), Namespace : "" }
1834
1835
1835
1836
config := Config {
@@ -1846,7 +1847,7 @@ func TestNodeReclaimFuncs(t *testing.T) {
1846
1847
// This is a constant that we use to test that disk pressure is over. Don't change!
1847
1848
diskStatConst := diskStatStart
1848
1849
summaryProvider := & fakeSummaryProvider {result : summaryStatsMaker (diskStatStart )}
1849
- diskGC := & mockDiskGC {fakeSummaryProvider : summaryProvider , err : nil , readAndWriteSeparate : tc . writeableSeparateFromReadOnly }
1850
+ diskGC := & mockDiskGC {fakeSummaryProvider : summaryProvider , err : nil }
1850
1851
manager := & managerImpl {
1851
1852
clock : fakeClock ,
1852
1853
killPodFunc : podKiller .killPodNow ,
@@ -2292,8 +2293,8 @@ func TestInodePressureFsInodes(t *testing.T) {
2292
2293
2293
2294
fakeClock := testingclock .NewFakeClock (time .Now ())
2294
2295
podKiller := & mockPodKiller {}
2295
- diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs }
2296
- diskGC := & mockDiskGC {err : nil , readAndWriteSeparate : tc . writeableSeparateFromReadOnly }
2296
+ diskInfoProvider := & mockDiskInfoProvider {dedicatedImageFs : tc .dedicatedImageFs , dedicatedContainerFs : & tc . writeableSeparateFromReadOnly }
2297
+ diskGC := & mockDiskGC {err : nil }
2297
2298
nodeRef := & v1.ObjectReference {Kind : "Node" , Name : "test" , UID : types .UID ("test" ), Namespace : "" }
2298
2299
2299
2300
config := Config {
0 commit comments