testsvm-quarry 0.2.1

Quarry protocol testing utilities for Solana SVM testing framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

//! # Merge Miner Testing Utilities
//!
//! Test helpers for Quarry merge mining functionality.
//!
//! This module provides the `TestMergeMiner` struct which manages merge mining
//! operations across multiple quarries. Merge mining allows miners to participate
//! in multiple quarries simultaneously, earning rewards from both primary and
//! replica pools.
//!
//! ## Features
//!
//! - **Primary Miner Management**: Create and manage primary miners
//! - **Replica Pool Support**: Add and manage replica mining pools
//! - **Unified Staking**: Stake tokens across multiple quarries
//! - **Reward Collection**: Claim rewards from all participating pools

use anyhow::Result;
use solana_sdk::pubkey::Pubkey;
use testsvm::prelude::*;

use crate::{quarry_merge_mine, quarry_mine};

/// Helper for managing a merge miner with type-safe account references
#[derive(Debug)]
pub struct TestMergeMiner {
    pub merge_miner: AccountRef<quarry_merge_mine::accounts::MergeMiner>,
    pub primary_tokens: AccountRef<anchor_spl::token::TokenAccount>,
    pub replica_tokens: AccountRef<anchor_spl::token::TokenAccount>,
}

impl TestMergeMiner {
    /// Create primary miner and miner vault for the merge miner
    pub fn create_primary_miner(
        &self,
        env: &mut TestSVM,
        label: &str,
        rewarder: &Pubkey,
        quarry: &Pubkey,
        primary_mint: &AccountRef<anchor_spl::token::Mint>,
    ) -> Result<(
        AccountRef<quarry_mine::accounts::Miner>,
        AccountRef<anchor_spl::token::TokenAccount>,
    )> {
        let merge_pool = self.merge_miner.load(env)?.pool;
        // Get primary miner PDA
        let primary_miner = env.get_pda::<quarry_mine::accounts::Miner>(
            &format!("{label}_primary_miner"),
            &[b"Miner", quarry.as_ref(), self.merge_miner.key.as_ref()],
            crate::quarry_mine::ID,
        )?;

        // Create primary miner vault ATA
        let (create_miner_vault_ix, primary_miner_vault) = env.create_ata_ix(
            &format!("{label}_primary_miner_vault"),
            &primary_miner.into(),
            &primary_mint.into(),
        )?;

        // Create init_miner_v2 instruction for merge mine
        let init_miner_ix = anchor_instruction(
            crate::quarry_merge_mine::ID,
            quarry_merge_mine::client::accounts::InitMinerV2 {
                pool: merge_pool,
                mm: self.merge_miner.key,
                miner: primary_miner.key,
                miner_vault: primary_miner_vault.into(),
                token_mint: primary_mint.key,
                quarry: *quarry,
                payer: env.default_fee_payer(),
                system_program: solana_sdk::system_program::ID,
                token_program: anchor_spl::token::ID,
                mine_program: crate::quarry_mine::ID,
                rewarder: *rewarder,
            },
            quarry_merge_mine::client::args::InitMinerV2 {},
        );

        // Execute instructions
        env.execute_ixs(&[create_miner_vault_ix, init_miner_ix])?;

        Ok((primary_miner, primary_miner_vault))
    }

    /// Create replica miner and miner vault for the merge miner
    pub fn create_replica_miner(
        &self,
        env: &mut TestSVM,
        label: &str,
        rewarder: &Pubkey,
        quarry: &Pubkey,
        replica_mint: &AccountRef<anchor_spl::token::Mint>,
    ) -> Result<(
        AccountRef<quarry_mine::accounts::Miner>,
        AccountRef<anchor_spl::token::TokenAccount>,
    )> {
        let merge_pool = self.merge_miner.load(env)?.pool;

        // Get replica miner PDA
        let replica_miner = env.get_pda::<quarry_mine::accounts::Miner>(
            &format!("{label}_replica_miner"),
            &[b"Miner", quarry.as_ref(), self.merge_miner.key.as_ref()],
            crate::quarry_mine::ID,
        )?;

        // Create replica miner vault ATA
        let (create_miner_vault_ix, replica_miner_vault) = env.create_ata_ix(
            &format!("{label}_replica_miner_vault"),
            &replica_miner.into(),
            &replica_mint.into(),
        )?;

        // Create init_miner_v2 instruction for merge mine
        let init_miner_ix = anchor_instruction(
            crate::quarry_merge_mine::ID,
            quarry_merge_mine::client::accounts::InitMinerV2 {
                pool: merge_pool,
                mm: self.merge_miner.key,
                miner: replica_miner.key,
                miner_vault: replica_miner_vault.into(),
                token_mint: replica_mint.key,
                quarry: *quarry,
                payer: env.default_fee_payer(),
                system_program: solana_sdk::system_program::ID,
                token_program: anchor_spl::token::ID,
                mine_program: crate::quarry_mine::ID,
                rewarder: *rewarder,
            },
            quarry_merge_mine::client::args::InitMinerV2 {},
        );

        // Execute instructions
        env.execute_ixs(&[create_miner_vault_ix, init_miner_ix])?;

        Ok((replica_miner, replica_miner_vault))
    }
}