blockifier_test_utils 0.18.0-rc.0

Test utilities for the blockifier.
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

use starknet_api::transaction::TransactionVersion;
use strum::EnumIter;

#[derive(Clone, Default, Hash, PartialEq, EnumIter, Eq, Copy, Debug)]
pub enum RunnableCairo1 {
    #[default]
    Casm,
    #[cfg(feature = "cairo_native")]
    Native,
}

impl RunnableCairo1 {
    pub fn is_cairo_native(&self) -> bool {
        match self {
            Self::Casm => false,

            #[cfg(feature = "cairo_native")]
            Self::Native => true,
        }
    }
}

// TODO(Aviv, 14/7/2024): Move from test utils module, and use it in ContractClassVersionMismatch
// error.
#[derive(Clone, Default, Hash, PartialEq, Eq, Copy, Debug)]
pub enum CairoVersion {
    #[default]
    Cairo0,
    Cairo1(RunnableCairo1),
}

impl CairoVersion {
    // A declare transaction of the given version, can be used to declare contracts of the returned
    // cairo version.
    // TODO(Dori): Make TransactionVersion an enum and use match here.
    pub fn from_declare_tx_version(tx_version: TransactionVersion) -> Self {
        if tx_version == TransactionVersion::ZERO || tx_version == TransactionVersion::ONE {
            CairoVersion::Cairo0
        } else if tx_version == TransactionVersion::TWO || tx_version == TransactionVersion::THREE {
            CairoVersion::Cairo1(RunnableCairo1::Casm)
        } else {
            panic!("Transaction version {tx_version:?} is not supported.")
        }
    }

    pub fn is_cairo0(&self) -> bool {
        match self {
            Self::Cairo0 => true,
            Self::Cairo1(_) => false,
        }
    }

    pub fn is_cairo_native(&self) -> bool {
        match self {
            Self::Cairo0 => false,

            Self::Cairo1(runnable_cairo1) => runnable_cairo1.is_cairo_native(),
        }
    }
}