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