Add more asserts

mvines · Jun 30, 2024 · 81d4dab · 81d4dab
1 parent 567dcc7
commit 81d4dab
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/priority_fee.rs b/src/priority_fee.rs
@@ -56,6 +56,7 @@ pub struct ComputeBudget {
 
 impl ComputeBudget {
     pub fn new(compute_unit_limit: u32, priority_fee_lamports: u64) -> Self {
+        assert_ne!(compute_unit_limit, 0);
         Self {
             compute_unit_price_micro_lamports: priority_fee_lamports * (1e6 as u64)
                 / compute_unit_limit as u64,
@@ -105,6 +106,8 @@ pub fn apply_priority_fee(
     compute_unit_limit: u32,
     priority_fee: PriorityFee,
 ) -> Result<u64, Box<dyn std::error::Error>> {
+    assert_ne!(compute_unit_limit, 0);
+
     let compute_budget = match priority_fee {
         PriorityFee::Exact { lamports } => ComputeBudget::new(compute_unit_limit, lamports),
         PriorityFee::Auto {