区块链国赛笔记 10｜Rust 智能合约初识：ink! 存储、消息与单元测试

#![cfg_attr(not(feature = "std"), no_std)]
// 编译属性声明
// #! 表示创建级别属性 no_std 模式
// no_std 表示不使用 标准库 合约 不使用标准库很重要


// 整体合约基于 !ink 框架
use ink_lang as ink;
// 使用 ink_lang

#[ink::contract] // 宏标记 ink 合约
mod hello_world { // 定义合约模块
    #[ink(storage)]  // 标记合约存储结构
    pub struct HelloWorld {
        value: i32, // 合约结构体
    }

    impl HelloWorld {   // 定义实现块
        #[ink(constructor)] // 标记构造函数
        pub fn new(init_value: i32) -> Self {
            Self { value: init_value } // 接收初始值 创建合约实例
        }

        #[ink(constructor)]
        pub fn default() -> Self {  // 创建默认值为0的合约实例
            Self::new(0)
        }

        #[ink(message)]
        pub fn get(&self) -> i32 { // 构造get 返回值
            self.value
        }

        #[ink(message)]
        pub fn set(&mut self, new_value: i32) { // 构造set 设置值
            self.value = new_value;
        }
    }
}

#![cfg(test)] // 测试属性声明

use ink_lang as ink;

use ink_env::{
    test::{
        default_accounts, // 获取测试账户
        set_caller,  // 模拟不同账户
        set_value_transferred, // 设置转账金额
        get_account_balance // 查询账户余额
    },
    DefaultEnvironment,
};
use ink_primitives::AccountId;

// 导入合约模块
use hello_world::HelloWorld;

#[ink::test]
fn new_works() {
    // 测试构造函数
    let contract = HelloWorld::new(42);
    assert_eq!(contract.get(), 42); // 构造12 测试能不能get 12
}

#[ink::test]
fn default_works() {
    // 测试默认构造函数
    let contract = HelloWorld::default(); // 测试无参构造函数
    assert_eq!(contract.get(), 0); // 验证0
}

#[ink::test]
fn set_and_get_work() {
    // 测试 set 和 get 方法
    let mut contract = HelloWorld::new(10);

    // 初始值检查
    assert_eq!(contract.get(), 10);

    // 设置新值
    contract.set(100);

    // 验证新值
    assert_eq!(contract.get(), 100);
}

#[ink::test]
fn multiple_set_operations() {
    // 测试多次设置操作
    let mut contract = HelloWorld::default();

    contract.set(5);
    assert_eq!(contract.get(), 5);

    contract.set(-10);
    assert_eq!(contract.get(), -10);

    contract.set(0);
    assert_eq!(contract.get(), 0);
}

#[ink::test]
fn test_with_different_accounts() {
    // 测试不同账户调用
    let accounts = default_accounts::<DefaultEnvironment>();

    // 使用 Alice 账户部署合约
    set_caller::<DefaultEnvironment>(accounts.alice);
    let mut contract = HelloWorld::new(50);
    assert_eq!(contract.get(), 50);

    // 使用 Bob 账户调用（在真实环境中会有权限检查）
    set_caller::<DefaultEnvironment>(accounts.bob);
    contract.set(75);
    assert_eq!(contract.get(), 75);
}

#[ink::test]
fn test_edge_cases() {
    // 测试边界情况
    let mut contract = HelloWorld::new(i32::MAX);
    assert_eq!(contract.get(), i32::MAX);

    contract.set(i32::MIN);
    assert_eq!(contract.get(), i32::MIN);

    contract.set(0);
    assert_eq!(contract.get(), 0);
}

#[ink::test]
fn test_constructor_variants() {
    // 测试不同的构造函数
    let contract1 = HelloWorld::new(123);
    let contract2 = HelloWorld::default();

    assert_eq!(contract1.get(), 123);
    assert_eq!(contract2.get(), 0);
}

// 集成测试示例
#[cfg(test)]
mod integration_tests {
    use super::*;

    #[ink::test]
    fn complete_workflow() {
        // 完整的工作流程测试
        let mut contract = HelloWorld::default();

        // 初始状态
        assert_eq!(contract.get(), 0);

        // 执行操作
        contract.set(42);

        // 验证结果
        assert_eq!(contract.get(), 42);

        // 再次更新
        contract.set(100);
        assert_eq!(contract.get(), 100);
    }
}