Test more

dfinity · Jun 9, 2024 · b6d6dad · b6d6dad
1 parent 944613f
commit b6d6dad
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/src/yaml2candid/src/test.rs b/src/yaml2candid/src/test.rs
@@ -479,6 +479,20 @@ fn can_convert() {
                 val: IDLValue::None,
             }]),
         },
+        TestVec {
+            description: "Record containing None in conventional form (omission)",
+            typ: IDLType::RecordT(vec![TypeField {
+                label: Label::Named("Foo".to_string()),
+                typ: IDLType::OptT(Box::new(IDLType::PrimT(
+                    candid_parser::types::PrimType::Int8,
+                ))),
+            }]),
+            data: YamlValue::Mapping([].into_iter().collect()),
+            expected_result: IDLValue::Record(vec![IDLField {
+                id: Label::Named("Foo".to_string()),
+                val: IDLValue::None,
+            }]),
+        },
         TestVec {
             description: "Record containing Some(5) in canonical form",
             typ: IDLType::RecordT(vec![TypeField {
@@ -500,6 +514,45 @@ fn can_convert() {
                 val: IDLValue::Opt(Box::new(IDLValue::Int8(8))),
             }]),
         },
+        TestVec {
+            description: "Record containing Some(5) in conventional form",
+            typ: IDLType::RecordT(vec![TypeField {
+                label: Label::Named("Foo".to_string()),
+                typ: IDLType::OptT(Box::new(IDLType::PrimT(
+                    candid_parser::types::PrimType::Int8,
+                ))),
+            }]),
+            data: YamlValue::Mapping(
+                [(YamlValue::from("Foo"), YamlValue::from(8))]
+                    .into_iter()
+                    .collect(),
+            ),
+            expected_result: IDLValue::Record(vec![IDLField {
+                id: Label::Named("Foo".to_string()),
+                val: IDLValue::Opt(Box::new(IDLValue::Int8(8))),
+            }]),
+        },
+        TestVec {
+            description: "Record containing Some([5]) in conventional form",
+            typ: IDLType::RecordT(vec![TypeField {
+                label: Label::Named("Foo".to_string()),
+                typ: IDLType::OptT(Box::new(IDLType::VecT(Box::new(IDLType::PrimT(
+                    candid_parser::types::PrimType::Int8,
+                ))))),
+            }]),
+            data: YamlValue::Mapping(
+                [(
+                    YamlValue::from("Foo"),
+                    YamlValue::Sequence(vec![YamlValue::from(8)]),
+                )]
+                .into_iter()
+                .collect(),
+            ),
+            expected_result: IDLValue::Record(vec![IDLField {
+                id: Label::Named("Foo".to_string()),
+                val: IDLValue::Opt(Box::new(IDLValue::Vec(vec![IDLValue::Int8(8)]))),
+            }]),
+        },
     ];
 
     for TestVec {