gtest 2.0.0-pre.1

Testing utils for developing Gear programs.
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

// Copyright (C) Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

//! Bridge builtin storage manager.

use std::thread::LocalKey;

use crate::state::WithOverlay;
use gprimitives::U256;

thread_local! {
    pub(super) static BRIDGE_MESSAGE_NONCE: WithOverlay<U256> = Default::default();
}

fn storage() -> &'static LocalKey<WithOverlay<U256>> {
    &BRIDGE_MESSAGE_NONCE
}

pub(crate) struct BridgeBuiltinStorage;

impl BridgeBuiltinStorage {
    /// Get the current message nonce.
    pub(crate) fn fetch_nonce() -> U256 {
        storage().with(|nonce| {
            let mut data = nonce.data_mut();
            let ret = *data;
            *data = data.saturating_add(U256::one());

            ret
        })
    }

    pub(crate) fn clear() {
        storage().with(|nonce| {
            *nonce.data_mut() = U256::zero();
        });
    }
}