arbiter-core 0.11.0

use std::str::FromStr;

use arbiter_bindings::bindings::{self, weth::weth};
use arbiter_core::database::fork::Fork;
use ethers::{
    prelude::Middleware,
    types::{Address, U256 as eU256, U64},
};
include!("common.rs");

#[tokio::test]
async fn receipt_data() {
    let (_environment, client) = startup();
    let arbiter_token = deploy_arbx(client.clone()).await;
    let receipt: ethers::types::TransactionReceipt = arbiter_token
        .mint(client.default_sender().unwrap(), 1000u64.into())
        .send()
        .await
        .unwrap()
        .await
        .unwrap()
        .unwrap();

    assert!(receipt.block_number.is_some());
    assert_eq!(receipt.status, Some(1.into()));

    assert!(receipt.contract_address.is_none());
    assert_eq!(receipt.to, Some(arbiter_token.address()));

    assert!(receipt.gas_used.is_some());
    assert_eq!(receipt.logs.len(), 1);
    assert_eq!(receipt.logs[0].topics.len(), 3);
    assert_eq!(receipt.transaction_index, 1.into());
    assert_eq!(receipt.from, client.default_sender().unwrap());

    let mut cumulative_gas = eU256::from(0);
    assert!(receipt.cumulative_gas_used >= cumulative_gas);
    cumulative_gas += receipt.cumulative_gas_used;

    let receipt_1 = arbiter_token
        .mint(client.default_sender().unwrap(), 1000u64.into())
        .send()
        .await
        .unwrap()
        .await
        .unwrap()
        .unwrap();

    // ensure gas in increasing
    assert!(cumulative_gas <= receipt_1.cumulative_gas_used);
}

#[tokio::test]
async fn user_update_block() {
    let (_environment, client) = startup();
    let block_number = client.get_block_number().await.unwrap();
    assert_eq!(block_number, U64::from(0));

    let block_timestamp = client.get_block_timestamp().await.unwrap();
    assert_eq!(block_timestamp, eU256::from(1));

    let new_block_number = 69;
    let new_block_timestamp = 420;

    assert!(client
        .update_block(new_block_number, new_block_timestamp,)
        .is_ok());

    let block_number = client.get_block_number().await.unwrap();
    assert_eq!(block_number, new_block_number.into());

    let block_timestamp = client.get_block_timestamp().await.unwrap();
    assert_eq!(block_timestamp, new_block_timestamp.into());
}

#[should_panic]
#[tokio::test]
async fn stop_environment() {
    let (environment, client) = startup();
    environment.stop().unwrap();
    deploy_arbx(client).await;
}

#[tokio::test]
async fn fork_into_arbiter() {
    let fork = Fork::from_disk("tests/fork.json").unwrap();

    // Get the environment going
    let environment = Environment::builder().with_state(fork.db).build();

    // Create a client
    let client = ArbiterMiddleware::new(&environment, Some("name")).unwrap();

    // Deal with the weth contract
    let weth_meta = fork.contracts_meta.get("weth").unwrap();
    let weth = weth::WETH::new(weth_meta.address, client.clone());

    let address_to_check_balance =
        Address::from_str(&weth_meta.mappings.get("balanceOf").unwrap()[0]).unwrap();

    println!("checking address: {}", address_to_check_balance);
    let balance = weth
        .balance_of(address_to_check_balance)
        .call()
        .await
        .unwrap();
    assert_eq!(balance, eU256::from(34890707020710109111_u128));

    // eoa check
    let eoa = fork.eoa.get("vitalik").unwrap();
    let eth_balance = client.get_balance(*eoa, None).await.unwrap();
    // Check the balance of the eoa with the load cheatcode
    assert_eq!(eth_balance, eU256::from(934034962177715175765_u128));
}

#[tokio::test]
async fn middleware_from_forked_eo() {
    let fork = Fork::from_disk("tests/fork.json").unwrap();

    // Get the environment going
    let environment = Environment::builder().with_state(fork.db).build();

    let vitalik_address = fork.eoa.get("vitalik").unwrap();
    let vitalik_as_a_client =
        ArbiterMiddleware::new_from_forked_eoa(&environment, *vitalik_address);
    assert!(vitalik_as_a_client.is_ok());
    let vitalik_as_a_client = vitalik_as_a_client.unwrap();

    // test a state mutating call from the forked eoa
    let weth = bindings::weth::WETH::deploy(vitalik_as_a_client.clone(), ())
        .unwrap()
        .send()
        .await;
    assert!(weth.is_ok()); // vitalik deployed the weth contract

    // test a non mutating call from the forked eoa
    let eth_balance = vitalik_as_a_client
        .get_balance(*vitalik_address, None)
        .await
        .unwrap();
    assert_eq!(eth_balance, eU256::from(934034962177715175765_u128));
}

#[tokio::test]
async fn env_returns_db() {
    let (environment, client) = startup();
    deploy_arbx(client).await;
    let db = environment.stop().unwrap();
    assert!(!db.state.read().unwrap().accounts.is_empty())
}

#[tokio::test]
async fn block_logs() {
    let (environment, client) = startup();

    let arbiter_token = deploy_arbx(client.clone()).await;
    arbiter_token
        .mint(Address::zero(), eU256::from(1000))
        .send()
        .await
        .unwrap()
        .await
        .unwrap();

    let new_block_number = 69;
    let new_block_timestamp = 420;

    client
        .update_block(new_block_number, new_block_timestamp)
        .unwrap();

    arbiter_token
        .approve(Address::zero(), eU256::from(1000))
        .send()
        .await
        .unwrap()
        .await
        .unwrap();
    client.update_block(6969, 420420).unwrap();

    let db = environment.stop().unwrap();
    let logs = db.logs.read().unwrap();
    println!("DB Logs: {:?}\n", logs);
    assert_eq!(logs.get(&revm::primitives::U256::from(0)).unwrap().len(), 1);
    assert_eq!(
        logs.get(&revm::primitives::U256::from(69)).unwrap().len(),
        1
    );
}