Add Option<T> deserializer support

Signed-off-by: Gris Ge <[email protected]>
nispor · Jan 3, 2025 · 0163763 · 0163763
1 parent fafd671
commit 0163763
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/deserializer.rs b/src/deserializer.rs
@@ -170,11 +170,14 @@ impl<'de> Deserializer<'de> for &mut RmsdDeserializer {
         todo!()
     }
 
-    fn deserialize_option<V>(self, _visitor: V) -> Result<V::Value, Self::Error>
+    fn deserialize_option<V>(self, visitor: V) -> Result<V::Value, Self::Error>
     where
         V: Visitor<'de>,
     {
-        todo!()
+        match self.parsed.data {
+            YamlValueData::Null => visitor.visit_none(),
+            _ => visitor.visit_some(self),
+        }
     }
 
     fn deserialize_unit<V>(self, _visitor: V) -> Result<V::Value, Self::Error>

diff --git a/tests/from_str.rs b/tests/from_str.rs
@@ -290,7 +290,7 @@ fn test_de_array_of_enum_of_struct() -> Result<(), RmsdError> {
 fn test_de_struct_with_enum_member() -> Result<(), RmsdError> {
     #[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
     struct FooTest {
-        uint_a: EnumTest,
+        uint_a: Option<EnumTest>,
     }
 
     #[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
@@ -305,7 +305,7 @@ fn test_de_struct_with_enum_member() -> Result<(), RmsdError> {
 
     assert_eq!(
         FooTest {
-            uint_a: EnumTest::Bar(BarTest { uint_b: 32 })
+            uint_a: Some(EnumTest::Bar(BarTest { uint_b: 32 }))
         },
         rmsd_yaml::from_str::<FooTest>(
             r#"