Skip to main content

codama_nodes/display_nodes/
struct_field_display_node.rs

1use crate::{DisplaySkip, StructFieldDisplayNode};
2
3impl StructFieldDisplayNode {
4    pub fn new<T>(label: T) -> Self
5    where
6        T: Into<String>,
7    {
8        Self {
9            label: Some(label.into()),
10            ..Default::default()
11        }
12    }
13
14    pub fn flattened() -> Self {
15        Self {
16            flatten: Some(true),
17            ..Default::default()
18        }
19    }
20
21    pub fn skipped(skip: DisplaySkip) -> Self {
22        Self {
23            skip: Some(skip),
24            ..Default::default()
25        }
26    }
27}
28
29#[cfg(test)]
30mod tests {
31    use super::*;
32
33    #[test]
34    fn new() {
35        let node = StructFieldDisplayNode::new("Amount");
36        assert_eq!(node.label.as_deref(), Some("Amount"));
37        assert_eq!(node.skip, None);
38        assert_eq!(node.flatten, None);
39        assert_eq!(node.flatten_prefix, None);
40    }
41
42    #[test]
43    fn flattened() {
44        let node = StructFieldDisplayNode::flattened();
45        assert_eq!(node.flatten, Some(true));
46    }
47
48    #[test]
49    fn skipped() {
50        let node = StructFieldDisplayNode::skipped(DisplaySkip::Always);
51        assert_eq!(node.skip, Some(DisplaySkip::Always));
52    }
53
54    #[test]
55    fn to_json() {
56        let node = StructFieldDisplayNode {
57            label: Some("Amount".to_string()),
58            skip: Some(DisplaySkip::Never),
59            flatten: Some(true),
60            flatten_prefix: Some("inner".to_string()),
61        };
62        let json = serde_json::to_string(&node).unwrap();
63        assert_eq!(
64            json,
65            r#"{"kind":"structFieldDisplayNode","label":"Amount","skip":"never","flatten":true,"flattenPrefix":"inner"}"#
66        );
67    }
68
69    #[test]
70    fn from_json() {
71        let json = r#"{"kind":"structFieldDisplayNode","label":"Amount"}"#;
72        let node: StructFieldDisplayNode = serde_json::from_str(json).unwrap();
73        assert_eq!(node, StructFieldDisplayNode::new("Amount"));
74    }
75}