【Truffle】二、自定义合约测试

一、准备测试

上期我们自己安装部署了truffle，并且体验了测试用例的整个测试流程，实际开发中，我们可以对自己的合约进行测试。

我们首先先明白自定义合约测试需要几个文件

合约文件：既然要测试合约，肯定要有合约的源码文件，以.sol后缀结尾
部署文件：测试合约要写一份部署文件，用于在测试中部署合约，以.js后缀结尾
测试文件：测试合约的流程文件，用于对合约的方法流程进行测试，以.js后缀结尾

合约文件的话，我们写一个最简单的合约用来测试,合约中定义了一个名为"number" 的无符号整数变量,通过调用 store() 函数，可以将一个无符号整数值存储到 "number" 变量中。然后，可以通过调用 retrieve() 函数来获取存储在 "number" 变量中的值。

// SPDX-License-Identifier: GPL-3.0pragma solidity >=0.8.2 <0.9.0;/*** @title Storage* @dev Store & retrieve value in a variable* @custom:dev-run-script ./scripts/deploy_with_ethers.ts*/
contract Storage {uint256 number;/*** @dev Store value in variable* @param num value to store*/function store(uint256 num) public {number = num;}/*** @dev Return value * @return value of 'number'*/function retrieve() public view returns (uint256){return number;}
}

我们把它存放到工作目录中的contracts目录下，命名为storage.sol

部署文件，固定的格式，可以参考之前下载的测试用例修改一下即可

const Storage = artifacts.require("Storage");module.exports = function (deployer) {deployer.deploy(Storage);
};

我们把它存放在migrations目录下，命名为1_deploy_contracts.js

最后是测试文件，因为这个合约只有两个方法，所以我们两个方法都要进行测试，测试用例命名为Storage_test.js

const Storage = artifacts.require("Storage");contract("Storage", (accounts) => {let storageInstance;before(async () => {storageInstance = await Storage.deployed();});it("should store a value", async () => {const newValue = 42;await storageInstance.store(newValue);const storedValue = await storageInstance.retrieve();assert.equal(storedValue, newValue, "Value was not stored correctly");});it("should retrieve the stored value", async () => { const storedValue = await storageInstance.retrieve();assert.equal(storedValue, 42, "Stored value is incorrect");});
});

最后的目录结构是这样的: