surfpool-core 1.3.1

Where you train before surfing Solana
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

#![allow(dead_code)]
use std::net::TcpListener;

use crossbeam_channel::Sender;
use solana_clock::Clock;
use solana_epoch_info::EpochInfo;
use solana_transaction::versioned::VersionedTransaction;
use surfpool_types::{CheatcodeConfig, RpcConfig, SimnetCommand};

use crate::{
    rpc::RunloopContext,
    surfnet::{PluginCommand, locker::SurfnetSvmLocker, svm::SurfnetSvm},
};

pub fn get_free_port() -> Result<u16, String> {
    let listener =
        TcpListener::bind("127.0.0.1:0").map_err(|e| format!("Failed to bind to port 0: {}", e))?;
    let port = listener
        .local_addr()
        .map_err(|e| format!("failed to parse address: {}", e))?
        .port();
    drop(listener);
    Ok(port)
}

#[derive(Clone)]
pub struct TestSetup<T>
where
    T: Clone,
{
    pub context: RunloopContext,
    pub rpc: T,
}

impl<T> TestSetup<T>
where
    T: Clone,
{
    pub fn new(rpc: T) -> Self {
        let (simnet_commands_tx, _rx) = crossbeam_channel::unbounded();

        let (mut surfnet_svm, _, _) = SurfnetSvm::default();
        let clock = Clock {
            slot: 123,
            epoch_start_timestamp: 123,
            epoch: 1,
            leader_schedule_epoch: 1,
            unix_timestamp: 123,
        };
        surfnet_svm.inner.set_sysvar::<Clock>(&clock);
        surfnet_svm.latest_epoch_info = EpochInfo {
            epoch: clock.epoch,
            slot_index: clock.slot,
            slots_in_epoch: 100,
            absolute_slot: clock.slot,
            block_height: 42,
            transaction_count: Some(2),
        };
        surfnet_svm.transactions_processed = 69;

        let (plugin_commands_tx, _plugin_commands_rx) =
            crossbeam_channel::unbounded::<PluginCommand>();
        TestSetup {
            context: RunloopContext {
                simnet_commands_tx: simnet_commands_tx.clone(),
                id: None,
                svm_locker: SurfnetSvmLocker::new(surfnet_svm),
                remote_rpc_client: None,
                rpc_config: RpcConfig::default(),
                cheatcode_config: CheatcodeConfig::new(),
                plugin_commands_tx,
            },
            rpc,
        }
    }

    pub fn new_with_epoch_info(rpc: T, epoch_info: EpochInfo) -> Self {
        let setup = TestSetup::new(rpc);
        setup
            .context
            .svm_locker
            .0
            .blocking_write()
            .latest_epoch_info = epoch_info;
        setup
    }

    pub fn new_with_mempool(rpc: T, simnet_commands_tx: Sender<SimnetCommand>) -> Self {
        let mut setup = TestSetup::new(rpc);
        setup.context.simnet_commands_tx = simnet_commands_tx;
        setup
    }

    pub async fn without_blockhash(self) -> Self {
        let mut state_writer = self.context.svm_locker.0.write().await;
        state_writer.skip_blockhash_check = true;
        let svm = state_writer.inner.clone();
        let svm = svm.with_blockhash_check(false);
        state_writer.inner = svm;
        drop(state_writer);
        self
    }

    pub async fn process_txs(&mut self, txs: Vec<VersionedTransaction>) {
        let (status_tx, _rx) = crossbeam_channel::unbounded();
        for tx in txs {
            let _ = self
                .context
                .svm_locker
                .process_transaction(&None, tx.clone(), status_tx.clone(), true, false)
                .await
                .unwrap();
        }
    }
}