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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
//! ink! message IR.
use ra_ap_syntax::ast;
use crate::traits::IsInkCallable;
/// An ink! message.
#[ink_analyzer_macro::entity(arg_kind = Message)]
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Message {
// ASTNode type.
ast: ast::Fn,
}
impl_ast_type_trait!(Message, IsInkFn);
impl IsInkCallable for Message {}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_utils::*;
use crate::traits::{InkEntity, IsInkFn};
use test_utils::quote_as_str;
#[test]
fn cast_works() {
for (code, is_payable, has_selector, is_default, has_name) in [
(
quote_as_str! {
#[ink(message)]
pub fn my_message(&self) {}
},
false,
false,
false,
false,
),
(
quote_as_str! {
#[ink(message, payable, default, selector=1, name="myMessage")]
pub fn my_message(&self) {}
},
true,
true,
true,
true,
),
(
quote_as_str! {
#[ink(message)]
#[ink(payable, default, selector=1, name="myMessage")]
pub fn my_message(&self) {}
},
true,
true,
true,
true,
),
(
quote_as_str! {
#[ink(message)]
#[ink(payable)]
#[ink(default)]
#[ink(selector=1)]
#[ink(name="myMessage")]
pub fn my_message(&self) {}
},
true,
true,
true,
true,
),
(
quote_as_str! {
#[ink(message, selector=0xA)]
pub fn my_message(&self) {}
},
false,
true,
false,
false,
),
(
quote_as_str! {
#[ink(message, selector=_)]
pub fn my_message(&self) {}
},
false,
true,
false,
false,
),
(
quote_as_str! {
#[ink(message, selector=@)]
pub fn my_message(&self) {}
},
false,
true,
false,
false,
),
] {
let node = parse_first_syntax_node(code);
let message = Message::cast(node).unwrap();
// `payable` argument exists.
assert_eq!(message.payable_arg().is_some(), is_payable);
// `selector` argument exists.
assert_eq!(message.selector_arg().is_some(), has_selector);
// `default` argument exists.
assert_eq!(message.default_arg().is_some(), is_default);
// `name` argument exists.
assert_eq!(message.name_arg().is_some(), has_name);
// composed selector exists.
assert!(message.composed_selector().is_some());
// `fn` item exists.
assert!(message.fn_item().is_some());
}
}
}