use ra_ap_syntax::ast;
#[ink_analyzer_macro::entity(macro_kind = StorageItem)]
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StorageItem {
ast: ast::Adt,
}
impl StorageItem {
impl_pub_ast_type_getter!(adt, Adt);
impl_pub_ink_arg_getter!(packed_arg, Packed, packed);
impl_pub_ink_arg_getter!(derive_arg, Derive, derive);
}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_utils::*;
use crate::traits::InkEntity;
use quote::quote;
use test_utils::quote_as_str;
#[test]
fn cast_works() {
for code in [
quote! {
struct MyStorageItem {
}
},
quote! {
enum MyStorageItem {
}
},
quote! {
union MyStorageItem {
}
},
] {
let node = parse_first_syntax_node(quote_as_str! {
#[ink::storage_item(packed, derive=false)]
#code
});
let storage_item = StorageItem::cast(node).unwrap();
assert!(storage_item.packed_arg().is_some());
assert!(storage_item.derive_arg().is_some());
assert!(storage_item.adt().is_some());
}
}
}