@@ -163,6 +163,9 @@ func Test_ApplyContainerResourceRequirements(t *testing.T) {
163163}
164164
165165func Test_UpsertContainerResourceRequirements (t * testing.T ) {
166+ v0 , err := resource .ParseQuantity ("512Mi" )
167+ require .NoError (t , err )
168+
166169 v1 , err := resource .ParseQuantity ("1Gi" )
167170 require .NoError (t , err )
168171
@@ -174,6 +177,39 @@ func Test_UpsertContainerResourceRequirements(t *testing.T) {
174177
175178 var container core.Container
176179
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+
177213 t .Run ("Ensure limits are copied" , func (t * testing.T ) {
178214 UpscaleContainerResourceRequirements (& container , core.ResourceRequirements {
179215 Limits : core.ResourceList {
0 commit comments