forked from argotorg/fe
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathenum.fe
More file actions
77 lines (66 loc) · 1.72 KB
/
Copy pathenum.fe
File metadata and controls
77 lines (66 loc) · 1.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
enum MyOption {
None,
Some(u256),
}
msg EnumMsg {
#[selector = 0x6c56cb0c]
MakeSome { value: u256 } -> u256,
#[selector = 0x455dd373]
MakeNone -> u256,
#[selector = 0xd4eee422]
IsSomeCheck { value: u256 } -> u256,
}
pub contract EnumContract {
recv EnumMsg {
MakeSome { value } -> u256 {
let opt = MyOption::Some(value)
return match opt {
MyOption::Some(val) => val
MyOption::None => 0
}
}
MakeNone -> u256 {
let opt = MyOption::None
return match opt {
MyOption::Some(val) => val
MyOption::None => 0
}
}
IsSomeCheck { value } -> u256 {
let opt = MyOption::Some(value)
return match opt {
MyOption::Some(_) => 1
MyOption::None => 0
}
}
}
}
#[test]
fn test_enum() uses (evm: mut Evm) {
let contract_addr = evm.create2<EnumContract>(value: 0, args: (), salt: 0)
assert!(contract_addr.inner != 0)
// Test make_some(42) - should return 42
let res: u256 = evm.call(
addr: contract_addr,
gas: 100000,
value: 0,
message: EnumMsg::MakeSome { value: 42 }
)
assert!(res == 42)
// Test is_some_check(99) - should return 1 (true)
let res: u256 = evm.call(
addr: contract_addr,
gas: 100000,
value: 0,
message: EnumMsg::IsSomeCheck { value: 99 }
)
assert!(res == 1)
// Test make_none() - should return 0
let res: u256 = evm.call(
addr: contract_addr,
gas: 100000,
value: 0,
message: EnumMsg::MakeNone {}
)
assert!(res == 0)
}