@@ -185,25 +185,34 @@ func TestSuggestMemoryAllocation(t *testing.T) {
185
185
description string
186
186
sysLimit int
187
187
containerLimit int
188
+ nodes int
188
189
want int
189
190
}{
190
- {"128GB sys" , 128000 , 0 , 6000 },
191
- {"64GB sys" , 64000 , 0 , 6000 },
192
- {"16GB sys" , 16384 , 0 , 4000 },
193
- {"odd sys" , 14567 , 0 , 3600 },
194
- {"4GB sys" , 4096 , 0 , 2200 },
195
- {"2GB sys" , 2048 , 0 , 2048 },
196
- {"Unable to poll sys" , 0 , 0 , 2200 },
197
- {"128GB sys, 16GB container" , 128000 , 16384 , 16336 },
198
- {"64GB sys, 16GB container" , 64000 , 16384 , 16000 },
199
- {"16GB sys, 4GB container" , 16384 , 4096 , 4000 },
200
- {"4GB sys, 3.5GB container" , 16384 , 3500 , 3452 },
201
- {"2GB sys, 2GB container" , 16384 , 2048 , 2048 },
202
- {"2GB sys, unable to poll container" , 16384 , 0 , 4000 },
191
+ {"128GB sys" , 128000 , 0 , 1 , 6000 },
192
+ {"64GB sys" , 64000 , 0 , 1 , 6000 },
193
+ {"32GB sys" , 32768 , 0 , 1 , 6000 },
194
+ {"16GB sys" , 16384 , 0 , 1 , 4000 },
195
+ {"odd sys" , 14567 , 0 , 1 , 3600 },
196
+ {"4GB sys" , 4096 , 0 , 1 , 2200 },
197
+ {"2GB sys" , 2048 , 0 , 1 , 2048 },
198
+ {"Unable to poll sys" , 0 , 0 , 1 , 2200 },
199
+ {"128GB sys, 16GB container" , 128000 , 16384 , 1 , 16336 },
200
+ {"64GB sys, 16GB container" , 64000 , 16384 , 1 , 16000 },
201
+ {"16GB sys, 4GB container" , 16384 , 4096 , 1 , 4000 },
202
+ {"4GB sys, 3.5GB container" , 16384 , 3500 , 1 , 3452 },
203
+ {"16GB sys, 2GB container" , 16384 , 2048 , 1 , 2048 },
204
+ {"16GB sys, unable to poll container" , 16384 , 0 , 1 , 4000 },
205
+ {"128GB sys 2 nodes" , 128000 , 0 , 2 , 6000 },
206
+ {"8GB sys 3 nodes" , 8192 , 0 , 3 , 2200 },
207
+ {"16GB sys 2 nodes" , 16384 , 0 , 2 , 2200 },
208
+ {"32GB sys 2 nodes" , 32768 , 0 , 2 , 4050 },
209
+ {"odd sys 2 nodes" , 14567 , 0 , 2 , 2200 },
210
+ {"4GB sys 2 nodes" , 4096 , 0 , 2 , 2200 },
211
+ {"2GB sys 3 nodes" , 2048 , 0 , 3 , 2048 },
203
212
}
204
213
for _ , test := range tests {
205
214
t .Run (test .description , func (t * testing.T ) {
206
- got := suggestMemoryAllocation (test .sysLimit , test .containerLimit )
215
+ got := suggestMemoryAllocation (test .sysLimit , test .containerLimit , test . nodes )
207
216
if got != test .want {
208
217
t .Errorf ("defaultMemorySize(sys=%d, container=%d) = %d, want: %d" , test .sysLimit , test .containerLimit , got , test .want )
209
218
}
0 commit comments