Add a test using huge OrderBook account

Author: snawaz

magicblock-committor-service/src/tasks/args_task.rs

@@ -112,6 +112,7 @@ impl BaseTask for ArgsTask {
                     ),
                     allow_undelegation: value.allow_undelegation,
                 };
+                log::warn!("DIFF computed: {:?}", args.diff);
                 dlp::instruction_builder::commit_diff(
                     *validator,
                     value.committed_account.pubkey,
@@ -213,7 +214,7 @@ impl BaseTask for ArgsTask {
             ArgsTaskType::Commit(_) => TaskType::Commit,
             // TODO (snawaz): What should we use here? Commit (in the sense of "category of task"), or add a
             // new variant "CommitDiff" to indicate a specific instruction?
-            ArgsTaskType::CommitDiff(_) => unimplemented!("task_type"), //TaskType::Commit,
+            ArgsTaskType::CommitDiff(_) => TaskType::Commit,
             ArgsTaskType::BaseAction(_) => TaskType::Action,
             ArgsTaskType::Undelegate(_) => TaskType::Undelegate,
             ArgsTaskType::Finalize(_) => TaskType::Finalize,

test-integration/Cargo.lock

@@ -9,7 +9,8 @@ use solana_program::{
 };
 
 use crate::{
-    DelegateCpiArgs, ScheduleCommitCpiArgs, ScheduleCommitInstruction,
+    BookUpdate, DelegateCpiArgs, ScheduleCommitCpiArgs,
+    ScheduleCommitInstruction,
 };
 
 pub fn init_account_instruction(
@@ -30,6 +31,45 @@ pub fn init_account_instruction(
     )
 }
 
+pub fn init_order_book_instruction(
+    payer: Pubkey,
+    order_book: Pubkey,
+) -> Instruction {
+    let program_id = crate::id();
+    let account_metas = vec![
+        AccountMeta::new(payer, true),
+        AccountMeta::new(order_book, false),
+        //AccountMeta::new_readonly(system_program::id(), false),
+        AccountMeta::new_readonly(system_program::id(), false),
+    ];
+
+    Instruction::new_with_borsh(
+        program_id,
+        &ScheduleCommitInstruction::InitOrderBook,
+        account_metas,
+    )
+}
+
+pub fn grow_order_book_instruction(
+    payer: Pubkey,
+    order_book: Pubkey,
+    additional_space: u64,
+) -> Instruction {
+    let program_id = crate::id();
+    let account_metas = vec![
+        AccountMeta::new(payer, true),
+        AccountMeta::new(order_book, false),
+        //AccountMeta::new_readonly(system_program::id(), false),
+        AccountMeta::new_readonly(system_program::id(), false),
+    ];
+
+    Instruction::new_with_borsh(
+        program_id,
+        &ScheduleCommitInstruction::GrowOrderBook(additional_space),
+        account_metas,
+    )
+}
+
 pub fn init_payer_escrow(payer: Pubkey) -> [Instruction; 2] {
     let top_up_ix = dlp::instruction_builder::top_up_ephemeral_balance(
         payer,
@@ -53,9 +93,13 @@ pub fn init_payer_escrow(payer: Pubkey) -> [Instruction; 2] {
     [top_up_ix, delegate_ix]
 }
 
-pub fn delegate_account_cpi_instruction(player: Pubkey) -> Instruction {
+pub fn delegate_account_cpi_instruction(
+    player: Pubkey,
+    user_seed: &[u8],
+) -> Instruction {
     let program_id = crate::id();
-    let (pda, _) = pda_and_bump(&player);
+    let (pda, _) =
+        Pubkey::find_program_address(&[user_seed, player.as_ref()], &crate::ID);
 
     let args = DelegateCpiArgs {
         valid_until: i64::MAX,
@@ -78,7 +122,11 @@ pub fn delegate_account_cpi_instruction(player: Pubkey) -> Instruction {
 
     Instruction::new_with_borsh(
         program_id,
-        &ScheduleCommitInstruction::DelegateCpi(args),
+        &if user_seed == b"magic_schedule_commit" {
+            ScheduleCommitInstruction::DelegateCpi(args)
+        } else {
+            ScheduleCommitInstruction::DelegateOrderBook(args)
+        },
         account_metas,
     )
 }
@@ -114,6 +162,45 @@ pub fn schedule_commit_cpi_instruction(
     )
 }
 
+pub fn update_order_book_instruction(
+    payer: Pubkey,
+    order_book: Pubkey,
+    update: BookUpdate,
+) -> Instruction {
+    let program_id = crate::id();
+    let account_metas = vec![
+        AccountMeta::new(payer, true),
+        AccountMeta::new(order_book, false),
+    ];
+
+    Instruction::new_with_borsh(
+        program_id,
+        &ScheduleCommitInstruction::UpdateOrderBook(update),
+        account_metas,
+    )
+}
+
+pub fn schedule_commit_diff_instruction_for_order_book(
+    payer: Pubkey,
+    order_book: Pubkey,
+    magic_program_id: Pubkey,
+    magic_context_id: Pubkey,
+) -> Instruction {
+    let program_id = crate::id();
+    let account_metas = vec![
+        AccountMeta::new(payer, true),
+        AccountMeta::new(order_book, false),
+        AccountMeta::new(magic_context_id, false),
+        AccountMeta::new_readonly(magic_program_id, false),
+    ];
+
+    Instruction::new_with_borsh(
+        program_id,
+        &ScheduleCommitInstruction::ScheduleCommitForOrderBook,
+        account_metas,
+    )
+}
+
 pub fn schedule_commit_with_payer_cpi_instruction(
     payer: Pubkey,
     magic_program_id: Pubkey,