codama_nodes/discriminator_nodes/
discriminator_node.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
use crate::{ConstantDiscriminatorNode, FieldDiscriminatorNode, SizeDiscriminatorNode};
use codama_nodes_derive::node_union;

#[node_union]
pub enum DiscriminatorNode {
    Constant(ConstantDiscriminatorNode),
    Field(FieldDiscriminatorNode),
    Size(SizeDiscriminatorNode),
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::HasKind;

    #[test]
    fn kind() {
        let node: DiscriminatorNode = SizeDiscriminatorNode::new(42).into();
        assert_eq!(node.kind(), "sizeDiscriminatorNode");
    }
}