Temp compute budget fix

Author: mvines

src/helius_rpc.rs

@@ -53,7 +53,7 @@ pub fn get_priority_fee_estimate_for_transaction(
     priority_level: HeliusPriorityLevel,
     transaction: &Transaction,
 ) -> Result<u64, String> {
-    println!("Invoking Helius RPC method: getPriorityFeeEstimate");
+    //println!("Invoking Helius RPC method: getPriorityFeeEstimate");
 
     let request = serde_json::json!([HeliusGetPriorityFeeEstimateRequest {
         options: Some(HeliusGetPriorityFeeEstimateOptions {
@@ -84,7 +84,7 @@ pub fn get_priority_fee_estimate_for_instructions(
     priority_level: HeliusPriorityLevel,
     instructions: &[Instruction],
 ) -> Result<u64, String> {
-    println!("Invoking Helius RPC method: getPriorityFeeEstimate");
+    //println!("Invoking Helius RPC method: getPriorityFeeEstimate");
 
     let mut account_keys: Vec<_> = instructions
         .iter()
@@ -113,7 +113,7 @@ pub fn get_priority_fee_estimate_for_instructions(
             solana_client::rpc_request::RpcRequest::Custom {
                 method: "getPriorityFeeEstimate",
             },
-            dbg!(request),
+            request,
         )
         .map(|response| {
             response

src/main.rs

@@ -3180,10 +3180,14 @@ async fn process_account_sweep<T: Signers>(
         None => {
             let mut message = Message::new(&instructions, Some(&from_authority_address));
             message.recent_blockhash = recent_blockhash;
-            assert_eq!(
-                rpc_client.get_fee_for_message(&message)?,
-                num_transaction_signatures * fee_calculator.lamports_per_signature
-            );
+            if compute_unit_price_micro_lamports.is_none() {
+                assert_eq!(
+                    rpc_client.get_fee_for_message(&message)?,
+                    num_transaction_signatures * fee_calculator.lamports_per_signature
+                );
+            } else {
+                println!("TODO: account for priority fee...");
+            }
 
             let mut transaction = Transaction::new_unsigned(message);
             let simulation_result = rpc_client.simulate_transaction(&transaction)?.value;