namada_tests 0.251.4

Namada tests setup, integration and E2E tests
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

pub mod eth_bridge_pool;
pub mod pos;

use std::cell::RefCell;
use std::collections::BTreeSet;

use namada_sdk::address::Address;
use namada_sdk::gas::{GasMeterKind, VpGasMeter};
use namada_sdk::state::StateRead;
use namada_sdk::state::testing::TestState;
use namada_sdk::storage;
use namada_vm::WasmCacheRwAccess;
use namada_vm::wasm::VpCache;
use namada_vm::wasm::run::VpEvalWasm;
use namada_vp::native_vp::{self, Ctx, NativeVp};

use crate::tx::TestTxEnv;

type NativeVpCtx<'a> = Ctx<
    'a,
    TestState,
    VpCache<WasmCacheRwAccess>,
    VpEvalWasm<
        <TestState as StateRead>::D,
        <TestState as StateRead>::H,
        WasmCacheRwAccess,
    >,
>;

#[derive(Debug)]
pub struct TestNativeVpEnv {
    pub tx_env: TestTxEnv,
    pub address: Address,
    pub verifiers: BTreeSet<Address>,
    pub keys_changed: BTreeSet<storage::Key>,
}

impl TestNativeVpEnv {
    pub fn from_tx_env(tx_env: TestTxEnv, address: Address) -> Self {
        // Find the tx verifiers and keys_changes the same way as protocol would
        let verifiers = tx_env.get_verifiers();

        let keys_changed = tx_env.all_touched_storage_keys();

        Self {
            address,
            tx_env,
            verifiers,
            keys_changed,
        }
    }
}

impl TestNativeVpEnv {
    /// Run some transaction code `apply_tx` and validate it with a native VP
    pub fn init_vp<'view, 'ctx: 'view, T>(
        &'ctx self,
        gas_meter: &'ctx RefCell<VpGasMeter>,
        init_native_vp: impl Fn(NativeVpCtx<'ctx>) -> T,
    ) -> T
    where
        T: NativeVp<'view>,
    {
        let ctx = NativeVpCtx::new(
            &self.address,
            &self.tx_env.state,
            &self.tx_env.batched_tx.tx,
            &self.tx_env.batched_tx.cmt,
            &self.tx_env.tx_index,
            gas_meter,
            &self.keys_changed,
            &self.verifiers,
            self.tx_env.vp_wasm_cache.clone(),
            GasMeterKind::MutGlobal,
        );
        init_native_vp(ctx)
    }

    /// Instantiate a native VP ctx to run a VP
    pub fn ctx<'ctx>(
        &'ctx self,
        gas_meter: &'ctx RefCell<VpGasMeter>,
    ) -> NativeVpCtx<'ctx> {
        NativeVpCtx::new(
            &self.address,
            &self.tx_env.state,
            &self.tx_env.batched_tx.tx,
            &self.tx_env.batched_tx.cmt,
            &self.tx_env.tx_index,
            gas_meter,
            &self.keys_changed,
            &self.verifiers,
            self.tx_env.vp_wasm_cache.clone(),
            GasMeterKind::MutGlobal,
        )
    }

    /// Run some transaction code `apply_tx` and validate it with a native VP
    pub fn validate_tx<'view, 'ctx: 'view, T>(
        &'ctx self,
        vp: &'view T,
    ) -> native_vp::Result<()>
    where
        T: 'view + NativeVp<'view>,
    {
        vp.validate_tx(
            &self.tx_env.batched_tx.to_ref(),
            &self.keys_changed,
            &self.verifiers,
        )
    }
}