nim-ethers/testmodule/testReturns.nim

55 lines
2.1 KiB
Nim
Raw Normal View History

import pkg/asynctest
import pkg/ethers
import ./hardhat
type
TestReturns = ref object of Contract
Static = (UInt256, UInt256)
Dynamic = (string, UInt256)
suite "Contract return values":
var contract: TestReturns
var provider: JsonRpcProvider
var snapshot: JsonNode
setup:
provider = JsonRpcProvider.new()
snapshot = await provider.send("evm_snapshot")
let deployment = readDeployment()
contract = TestReturns.new(!deployment.address(TestReturns), provider)
teardown:
discard await provider.send("evm_revert", @[snapshot])
await provider.close()
test "handles static size structs":
proc getStatic(contract: TestReturns): Static {.contract, pure.}
proc getStatics(contract: TestReturns): (Static, Static) {.contract, pure.}
check (await contract.getStatic()) == (1.u256, 2.u256)
check (await contract.getStatics()) == ((1.u256, 2.u256), (3.u256, 4.u256))
test "handles dynamic size structs":
proc getDynamic(contract: TestReturns): Dynamic {.contract, pure.}
proc getDynamics(contract: TestReturns): (Dynamic, Dynamic) {.contract, pure.}
check (await contract.getDynamic()) == ("1", 2.u256)
check (await contract.getDynamics()) == (("1", 2.u256), ("3", 4.u256))
test "handles mixed dynamic and static size structs":
proc getDynamicAndStatic(contract: TestReturns): (Dynamic, Static) {.contract, pure.}
check (await contract.getDynamicAndStatic()) == (("1", 2.u256), (3.u256, 4.u256))
test "handles return type that is a tuple with a single element":
proc getDynamic(contract: TestReturns): (Dynamic,) {.contract, pure.}
check (await contract.getDynamic()) == (("1", 2.u256),)
test "handles parentheses around return type":
proc getDynamic(contract: TestReturns): (Dynamic) {.contract, pure.}
check (await contract.getDynamic()) == ("1", 2.u256)
test "handles return type that is an explicit tuple":
proc getDynamics(contract: TestReturns): tuple[a, b: Dynamic] {.contract, pure.}
let values = await contract.getDynamics()
check values.a == ("1", 2.u256)
check values.b == ("3", 4.u256)