@@ -163,6 +163,9 @@ func Test_ApplyContainerResourceRequirements(t *testing.T) {
163
163
}
164
164
165
165
func Test_UpsertContainerResourceRequirements (t * testing.T ) {
166
+ v0 , err := resource .ParseQuantity ("512Mi" )
167
+ require .NoError (t , err )
168
+
166
169
v1 , err := resource .ParseQuantity ("1Gi" )
167
170
require .NoError (t , err )
168
171
@@ -174,6 +177,39 @@ func Test_UpsertContainerResourceRequirements(t *testing.T) {
174
177
175
178
var container core.Container
176
179
180
+ t .Run ("Ensure limits are applied optionally" , func (t * testing.T ) {
181
+ UpscaleContainerResourceRequirements (& container , core.ResourceRequirements {
182
+ Requests : core.ResourceList {
183
+ core .ResourceMemory : v1 ,
184
+ },
185
+ })
186
+
187
+ require .Len (t , container .Resources .Requests , 1 )
188
+ require .Contains (t , container .Resources .Requests , core .ResourceMemory )
189
+ require .Equal (t , v1 , container .Resources .Requests [core .ResourceMemory ])
190
+
191
+ require .Len (t , container .Resources .Limits , 0 )
192
+ })
193
+
194
+ t .Run ("Ensure limits are upscaled optionally" , func (t * testing.T ) {
195
+ UpscaleContainerResourceRequirements (& container , core.ResourceRequirements {
196
+ Limits : core.ResourceList {
197
+ core .ResourceMemory : v0 ,
198
+ },
199
+ Requests : core.ResourceList {
200
+ core .ResourceMemory : v1 ,
201
+ },
202
+ })
203
+
204
+ require .Len (t , container .Resources .Requests , 1 )
205
+ require .Contains (t , container .Resources .Requests , core .ResourceMemory )
206
+ require .Equal (t , v1 , container .Resources .Requests [core .ResourceMemory ])
207
+
208
+ require .Len (t , container .Resources .Limits , 1 )
209
+ require .Contains (t , container .Resources .Limits , core .ResourceMemory )
210
+ require .Equal (t , v1 , container .Resources .Limits [core .ResourceMemory ])
211
+ })
212
+
177
213
t .Run ("Ensure limits are copied" , func (t * testing.T ) {
178
214
UpscaleContainerResourceRequirements (& container , core.ResourceRequirements {
179
215
Limits : core.ResourceList {
0 commit comments