codama_nodes/pda_seed_nodes/
variable_pda_seed_node.rs

1use crate::{CamelCaseString, Docs, TypeNode};
2use codama_nodes_derive::node;
3
4#[node]
5pub struct VariablePdaSeedNode {
6    // Data.
7    pub name: CamelCaseString,
8    #[serde(default)]
9    #[serde(skip_serializing_if = "Docs::is_empty")]
10    pub docs: Docs,
11
12    // Children.
13    pub r#type: TypeNode,
14}
15
16impl From<VariablePdaSeedNode> for crate::Node {
17    fn from(val: VariablePdaSeedNode) -> Self {
18        crate::Node::PdaSeed(val.into())
19    }
20}
21
22impl VariablePdaSeedNode {
23    pub fn new<T, U>(name: T, r#type: U) -> Self
24    where
25        T: Into<CamelCaseString>,
26        U: Into<TypeNode>,
27    {
28        Self {
29            name: name.into(),
30            docs: Docs::default(),
31            r#type: r#type.into(),
32        }
33    }
34}
35
36#[cfg(test)]
37mod tests {
38    use super::*;
39    use crate::{NumberTypeNode, U32};
40
41    #[test]
42    fn new() {
43        let node = VariablePdaSeedNode::new("my_seed", NumberTypeNode::le(U32));
44        assert_eq!(node.name, CamelCaseString::new("mySeed"));
45        assert_eq!(node.r#type, TypeNode::Number(NumberTypeNode::le(U32)));
46    }
47
48    #[test]
49    fn direct_instantiation() {
50        let node = VariablePdaSeedNode {
51            name: "mySeed".into(),
52            docs: vec!["Hello".to_string()].into(),
53            r#type: NumberTypeNode::le(U32).into(),
54        };
55        assert_eq!(node.name, CamelCaseString::new("mySeed"));
56        assert_eq!(*node.docs, vec!["Hello".to_string()]);
57        assert_eq!(node.r#type, TypeNode::Number(NumberTypeNode::le(U32)));
58    }
59
60    #[test]
61    fn to_json() {
62        let node = VariablePdaSeedNode::new("mySeed", NumberTypeNode::le(U32));
63        let json = serde_json::to_string(&node).unwrap();
64        assert_eq!(
65            json,
66            r#"{"kind":"variablePdaSeedNode","name":"mySeed","type":{"kind":"numberTypeNode","format":"u32","endian":"le"}}"#
67        );
68    }
69
70    #[test]
71    fn from_json() {
72        let json = r#"{"kind":"variablePdaSeedNode","name":"mySeed","type":{"kind":"numberTypeNode","format":"u32","endian":"le"}}"#;
73        let node: VariablePdaSeedNode = serde_json::from_str(json).unwrap();
74        assert_eq!(
75            node,
76            VariablePdaSeedNode::new("mySeed", NumberTypeNode::le(U32))
77        );
78    }
79}