@@ -176,10 +176,14 @@ public void testShortfallScaleup2groups() throws Exception {
176176 // create 3 VMs for group1 and 1 VM for group2
177177 int hostIdx =0 ;
178178 final List <VirtualMachineLease > leases = new ArrayList <>();
179- leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier , ports , attributes1 ));
180- leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier , ports , attributes1 ));
181- leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier , ports , attributes1 ));
182- leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus2 , cpus2 * memMultiplier , ports , attributes2 ));
179+ leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier ,
180+ 0 , 0 , ports , attributes1 , Collections .singletonMap ("GPU" , 1.0 )));
181+ leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier ,
182+ 0 , 0 , ports , attributes1 , Collections .singletonMap ("GPU" , 1.0 )));
183+ leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus1 , cpus1 * memMultiplier ,
184+ 0 , 0 , ports , attributes1 , Collections .singletonMap ("GPU" , 1.0 )));
185+ leases .add (LeaseProvider .getLeaseOffer ("host" + hostIdx ++, cpus2 , cpus2 * memMultiplier ,
186+ 0 , 0 , ports , attributes2 , Collections .singletonMap ("GPU" , 2.0 )));
183187 // create 1-cpu tasks to fill VMS equal to two times the max size of group1, and also group2
184188 for (int i =0 ; i < (cpus1 *rule1 .getMaxSize () + cpus2 * rule2 .getMaxSize ()); i ++) {
185189 queue .queueTask (QueuableTaskProvider .wrapTask (qA1 , TaskRequestProvider .getTaskRequest (1 , memMultiplier , 1 )));
0 commit comments