fix solana test

espresso-cash · Jan 14, 2025 · 8bc9f84 · 8bc9f84
1 parent 3ce2f4f
commit 8bc9f84
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/solana/test/programs/compute_budget_program/program_test.dart b/packages/solana/test/programs/compute_budget_program/program_test.dart
@@ -41,10 +41,12 @@ void main() {
     );
 
     const unitPriceMicroLamports = 1000;
+    const computeUnitLimit = 200000;
 
     message = message.copyWith(
       instructions: [
         ...message.instructions,
+        ComputeBudgetInstruction.setComputeUnitLimit(units: computeUnitLimit),
         ComputeBudgetInstruction.setComputeUnitPrice(
           microLamports: unitPriceMicroLamports,
         ),
@@ -60,11 +62,10 @@ void main() {
       commitment: commitment,
     );
 
-    const defaultUnitLimit = 200000;
     const micro = 1000000;
     expect(
       feeWithComputeUnitPrice! - feeWithoutComputeUnitPrice!,
-      defaultUnitLimit * unitPriceMicroLamports / micro,
+      (computeUnitLimit * unitPriceMicroLamports) ~/ micro,
     );
   });
 }