piecrust 0.30.0

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) DUSK NETWORK. All rights reserved.

use piecrust::{ContractData, Error, SessionData, VM, contract_bytecode};

const OWNER: [u8; 32] = [0u8; 32];
const LIMIT: u64 = 1_000_000;

#[test]
pub fn push_pop() -> Result<(), Error> {
    let vm = VM::ephemeral()?;

    let mut session = vm.session(SessionData::builder())?;

    let id = session.deploy(
        contract_bytecode!("stack"),
        ContractData::builder().owner(OWNER),
        LIMIT,
    )?;

    let val = 42;

    session.call::<_, ()>(id, "push", &val, LIMIT)?;

    let len: u32 = session.call(id, "len", &(), LIMIT)?.data;
    assert_eq!(len, 1);

    let popped: Option<i32> = session.call(id, "pop", &(), LIMIT)?.data;
    let len: i32 = session.call(id, "len", &(), LIMIT)?.data;

    assert_eq!(len, 0);
    assert_eq!(popped, Some(val));

    Ok(())
}

#[test]
pub fn multi_push_pop() -> Result<(), Error> {
    let vm = VM::ephemeral()?;

    let mut session = vm.session(SessionData::builder())?;

    let id = session.deploy(
        contract_bytecode!("stack"),
        ContractData::builder().owner(OWNER),
        LIMIT,
    )?;

    const N: i32 = 16;

    for i in 0..N {
        session.call::<_, ()>(id, "push", &i, LIMIT)?;
        let len: i32 = session.call(id, "len", &(), LIMIT)?.data;

        assert_eq!(len, i + 1);
    }

    for i in (0..N).rev() {
        let popped: Option<i32> = session.call(id, "pop", &(), LIMIT)?.data;
        let len: i32 = session.call(id, "len", &(), LIMIT)?.data;

        assert_eq!(len, i);
        assert_eq!(popped, Some(i));
    }

    let popped: Option<i32> = session.call(id, "pop", &(), LIMIT)?.data;
    assert_eq!(popped, None);

    Ok(())
}