linera_execution/test_utils/

mod.rs

// Copyright (c) Zefchain Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

#![allow(clippy::cast_possible_truncation)]

mod mock_application;
#[cfg(with_revm)]
pub mod solidity;
mod system_execution_state;

use std::{collections::BTreeMap, sync::Arc, thread, vec};

use linera_base::{
    crypto::{AccountPublicKey, ValidatorPublicKey},
    data_types::{
        Amount, Blob, BlockHeight, ChainDescription, ChainOrigin, CompressedBytecode, Epoch,
        InitialChainConfig, OracleResponse, Timestamp,
    },
    identifiers::{AccountOwner, ApplicationId, BlobId, ChainId, ModuleId},
    ownership::ChainOwnership,
    vm::VmRuntime,
};
use linera_views::{context::Context, views::View};
use proptest::{prelude::any, strategy::Strategy};

pub use self::{
    mock_application::{ExpectedCall, MockApplication, MockApplicationInstance},
    system_execution_state::SystemExecutionState,
};
use crate::{
    committee::Committee, ApplicationDescription, ExecutionRuntimeContext, ExecutionStateView,
    MessageContext, OperationContext, QueryContext, ServiceRuntimeEndpoint, ServiceSyncRuntime,
    SystemExecutionStateView,
};

/// Creates a dummy committee with a single test validator.
pub fn dummy_committee() -> Committee {
    Committee::make_simple(vec![(
        ValidatorPublicKey::test_key(0),
        AccountPublicKey::test_key(0),
    )])
}

/// Creates a dummy committee map containing a single committee at epoch zero.
pub fn dummy_committees() -> BTreeMap<Epoch, Committee> {
    let committee = dummy_committee();
    BTreeMap::from([(Epoch::ZERO, committee)])
}

/// Creates a dummy root chain description with the given index, ownership, and balance.
pub fn dummy_chain_description_with_ownership_and_balance(
    index: u32,
    ownership: ChainOwnership,
    balance: Amount,
) -> ChainDescription {
    let origin = ChainOrigin::Root(index);
    let config = InitialChainConfig {
        application_permissions: Default::default(),
        account: AccountOwner::CHAIN,
        balance,
        epoch: Epoch::ZERO,
        ownership,
    };
    ChainDescription::new(origin, config, Timestamp::default())
}

/// Creates a dummy chain description owned by the given owner, with the maximum balance.
pub fn dummy_chain_description_with_owner(index: u32, owner: AccountOwner) -> ChainDescription {
    dummy_chain_description_with_ownership_and_balance(
        index,
        ChainOwnership::single(owner),
        Amount::MAX,
    )
}

/// Creates a dummy chain description with a deterministic single owner and the maximum balance.
pub fn dummy_chain_description(index: u32) -> ChainDescription {
    let chain_key = AccountPublicKey::test_key(2 * (index % 128) as u8 + 1);
    let ownership = ChainOwnership::single(chain_key.into());
    dummy_chain_description_with_ownership_and_balance(index, ownership, Amount::MAX)
}

/// Creates a dummy [`ApplicationDescription`] for use in tests.
pub fn create_dummy_user_application_description(
    index: u32,
) -> (ApplicationDescription, Blob, Blob) {
    let chain_id = dummy_chain_description(1).id();
    let mut contract_bytes = b"contract".to_vec();
    let mut service_bytes = b"service".to_vec();
    contract_bytes.push(index as u8);
    service_bytes.push(index as u8);
    let contract_blob = Blob::new_contract_bytecode(CompressedBytecode {
        compressed_bytes: Arc::new(contract_bytes.into_boxed_slice()),
    });
    let service_blob = Blob::new_service_bytecode(CompressedBytecode {
        compressed_bytes: Arc::new(service_bytes.into_boxed_slice()),
    });

    let vm_runtime = VmRuntime::Wasm;
    (
        ApplicationDescription {
            module_id: ModuleId::new(contract_blob.id().hash, service_blob.id().hash, vm_runtime),
            creator_chain_id: chain_id,
            block_height: 0.into(),
            application_index: index,
            required_application_ids: vec![],
            parameters: vec![],
        },
        contract_blob,
        service_blob,
    )
}

/// Creates a dummy [`OperationContext`] to use in tests.
pub fn create_dummy_operation_context(chain_id: ChainId) -> OperationContext {
    OperationContext {
        chain_id,
        height: BlockHeight(0),
        round: Some(0),
        authenticated_owner: None,
        timestamp: Default::default(),
    }
}

/// Creates a dummy [`MessageContext`] to use in tests.
pub fn create_dummy_message_context(
    chain_id: ChainId,
    authenticated_owner: Option<AccountOwner>,
) -> MessageContext {
    MessageContext {
        chain_id,
        origin: chain_id,
        origin_timestamp: Default::default(),
        is_bouncing: false,
        authenticated_owner,
        refund_grant_to: None,
        height: BlockHeight(0),
        round: Some(0),
        timestamp: Default::default(),
    }
}

/// Creates a dummy [`QueryContext`] to use in tests.
pub fn create_dummy_query_context() -> QueryContext {
    QueryContext {
        chain_id: dummy_chain_description(0).id(),
        next_block_height: BlockHeight(0),
        local_time: Timestamp::from(0),
    }
}

/// Registration of [`MockApplication`]s to use in tests.
#[allow(async_fn_in_trait)]
pub trait RegisterMockApplication {
    /// Returns the chain to use for the creation of the application.
    ///
    /// This is included in the mocked [`ApplicationId`].
    async fn creator_chain_id(&self) -> ChainId;

    /// Registers a new [`MockApplication`] and returns it with the [`ApplicationId`] that was
    /// used for it.
    async fn register_mock_application(
        &mut self,
        index: u32,
    ) -> anyhow::Result<(ApplicationId, MockApplication, [BlobId; 3])> {
        let (description, contract, service) = create_dummy_user_application_description(index);
        let description_blob_id = Blob::new_application_description(&description).id();
        let contract_blob_id = contract.id();
        let service_blob_id = service.id();

        let (app_id, application) = self
            .register_mock_application_with(description, contract, service)
            .await?;
        Ok((
            app_id,
            application,
            [description_blob_id, contract_blob_id, service_blob_id],
        ))
    }

    /// Registers a new [`MockApplication`] associated with a [`ApplicationDescription`] and
    /// its bytecode [`Blob`]s.
    async fn register_mock_application_with(
        &mut self,
        description: ApplicationDescription,
        contract: Blob,
        service: Blob,
    ) -> anyhow::Result<(ApplicationId, MockApplication)>;
}

impl<C> RegisterMockApplication for ExecutionStateView<C>
where
    C: Context + Clone + Send + Sync + 'static,
    C::Extra: ExecutionRuntimeContext,
{
    async fn creator_chain_id(&self) -> ChainId {
        self.system.creator_chain_id().await
    }

    async fn register_mock_application_with(
        &mut self,
        description: ApplicationDescription,
        contract: Blob,
        service: Blob,
    ) -> anyhow::Result<(ApplicationId, MockApplication)> {
        self.system
            .register_mock_application_with(description, contract, service)
            .await
    }
}

impl<C> RegisterMockApplication for SystemExecutionStateView<C>
where
    C: Context + Clone + Send + Sync + 'static,
    C::Extra: ExecutionRuntimeContext,
{
    async fn creator_chain_id(&self) -> ChainId {
        self.description.get().await.expect("failed to load description").as_ref().expect(
            "Can't register applications on a system state with no associated `ChainDescription`",
        ).into()
    }

    async fn register_mock_application_with(
        &mut self,
        description: ApplicationDescription,
        contract: Blob,
        service: Blob,
    ) -> anyhow::Result<(ApplicationId, MockApplication)> {
        let id = From::from(&description);
        let context = self.context();
        let extra = context.extra();
        let mock_application = MockApplication::default();

        extra
            .user_contracts()
            .pin()
            .insert(id, mock_application.clone().into());
        extra
            .user_services()
            .pin()
            .insert(id, mock_application.clone().into());
        extra
            .add_blobs([
                contract,
                service,
                Blob::new_application_description(&description),
            ])
            .await?;

        Ok((id, mock_application))
    }
}

/// Creates the given number of dummy user application registrations for use in tests.
pub fn create_dummy_user_application_registrations(
    count: u32,
) -> anyhow::Result<Vec<(ApplicationId, ApplicationDescription, Blob, Blob)>> {
    let mut ids = Vec::with_capacity(count as usize);

    for index in 0..count {
        let (description, contract_blob, service_blob) =
            create_dummy_user_application_description(index);
        let id = From::from(&description);

        ids.push((id, description, contract_blob, service_blob));
    }

    Ok(ids)
}

impl QueryContext {
    /// Spawns a thread running the [`ServiceSyncRuntime`] actor.
    ///
    /// Returns the endpoints to communicate with the actor.
    pub fn spawn_service_runtime_actor(self) -> ServiceRuntimeEndpoint {
        let (execution_state_sender, incoming_execution_requests) =
            futures::channel::mpsc::unbounded();
        let (runtime_request_sender, runtime_request_receiver) = std::sync::mpsc::channel();

        thread::spawn(move || {
            ServiceSyncRuntime::new(execution_state_sender, self).run(&runtime_request_receiver)
        });

        ServiceRuntimeEndpoint {
            incoming_execution_requests,
            runtime_request_sender,
        }
    }
}

/// Creates a [`Strategy`] for creating a [`BTreeMap`] of [`AccountOwner`]s with an initial
/// non-zero [`Amount`] of tokens.
pub fn test_accounts_strategy() -> impl Strategy<Value = BTreeMap<AccountOwner, Amount>> {
    proptest::collection::btree_map(
        any::<AccountOwner>(),
        (1_u128..).prop_map(Amount::from_tokens),
        0..5,
    )
}

/// Creates a vector of [`OracleResponse`]s for the supplied [`BlobId`]s.
pub fn blob_oracle_responses<'a>(blobs: impl Iterator<Item = &'a BlobId>) -> Vec<OracleResponse> {
    blobs
        .into_iter()
        .copied()
        .map(OracleResponse::Blob)
        .collect()
}