32 lines
874 B
Nim
32 lines
874 B
Nim
import pkg/asynctest
|
|
import pkg/ethers
|
|
import ./hardhat
|
|
|
|
type
|
|
TestEnums = ref object of Contract
|
|
SomeEnum = enum
|
|
One
|
|
Two
|
|
|
|
suite "Contract enum parameters and return values":
|
|
|
|
var contract: TestEnums
|
|
var provider: JsonRpcProvider
|
|
var snapshot: JsonNode
|
|
|
|
setup:
|
|
provider = JsonRpcProvider.new()
|
|
snapshot = await provider.send("evm_snapshot")
|
|
let deployment = readDeployment()
|
|
contract = TestEnums.new(!deployment.address(TestEnums), provider)
|
|
|
|
teardown:
|
|
discard await provider.send("evm_revert", @[snapshot])
|
|
await provider.close()
|
|
|
|
test "handles enum parameter and return value":
|
|
proc returnValue(contract: TestEnums,
|
|
value: SomeEnum): SomeEnum {.contract, pure.}
|
|
check (await contract.returnValue(SomeEnum.One)) == SomeEnum.One
|
|
check (await contract.returnValue(SomeEnum.Two)) == SomeEnum.Two
|