Support sums in rust-transient-0

2 files changed, 46 insertions, 1 deletions

diff --git a/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs b/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
index fbfc82d..0ea3f6f 100644
--- a/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
+++ b/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
@@ -170,6 +170,36 @@ pub fn decode_enum_with_variants(reader: &mut impl std::io::Read) -> ::std::io::
     })
 }
 
+pub fn encode_sum_with_variants(value: &crate::data::SumWithVariants, writer: &mut impl std::io::Write) -> ::std::io::Result::<()> {
+    match value {
+        crate::data::SumWithVariants::VariantOne(value) => {
+            {
+                crate::transient_0::encode_int_32(&0, writer)?;
+            };
+            {
+                crate::data::transient_0::encode_variant_one(value, writer)?;
+            };
+        },
+        crate::data::SumWithVariants::VariantTwo(value) => {
+            {
+                crate::transient_0::encode_int_32(&1, writer)?;
+            };
+            {
+                crate::data::transient_0::encode_variant_two(value, writer)?;
+            };
+        },
+    };
+    ::std::io::Result::Ok(())
+}
+
+pub fn decode_sum_with_variants(reader: &mut impl std::io::Read) -> ::std::io::Result::<crate::data::SumWithVariants> {
+    std::io::Result::Ok(match &crate::transient_0::decode_int_32(reader)? {
+        0 => crate::data::SumWithVariants::VariantOne(crate::data::transient_0::decode_variant_one(reader)?),
+        1 => crate::data::SumWithVariants::VariantTwo(crate::data::transient_0::decode_variant_two(reader)?),
+        _ => todo!(),
+    })
+}
+
 pub fn encode_variant_one(value: &crate::data::VariantOne, writer: &mut impl std::io::Write) -> ::std::io::Result::<()> {
     crate::transient_0::encode_int_32(&value.a, writer)?;
     ::std::io::Result::Ok(())
diff --git a/examples/10-transient-0/output/rust/src/lib.rs b/examples/10-transient-0/output/rust/src/lib.rs
index c3da14c..3aa9751 100644
--- a/examples/10-transient-0/output/rust/src/lib.rs
+++ b/examples/10-transient-0/output/rust/src/lib.rs
@@ -4,7 +4,7 @@ pub mod transient_0;
 #[cfg(test)]
 mod test {
     use std::io::Cursor;
-    use super::data::{EnumWithVariants, InnerStruct, OuterStruct, StructWithBool, StructWithInt32, StructWithInt64, StructWithList, StructWithOption, StructWithString};
+    use super::data::{EnumWithVariants, InnerStruct, OuterStruct, StructWithBool, StructWithInt32, StructWithInt64, StructWithList, StructWithOption, StructWithString, SumWithVariants, VariantOne, VariantTwo };
 
     #[test]
     fn struct_with_bool() {
@@ -128,6 +128,21 @@ mod test {
         );
     }
 
+    #[test]
+    fn sum_with_variants() {
+        assert_round_trip_encoding(
+            SumWithVariants::VariantOne(VariantOne { a: 10 }),
+            super::data::transient_0::encode_sum_with_variants,
+            super::data::transient_0::decode_sum_with_variants,
+        );
+
+        assert_round_trip_encoding(
+            SumWithVariants::VariantTwo(VariantTwo { a: 25 }),
+            super::data::transient_0::encode_sum_with_variants,
+            super::data::transient_0::decode_sum_with_variants,
+        );
+    }
+
     fn assert_round_trip_encoding<T: std::cmp::PartialEq + std::fmt::Debug>(
         value: T,
         encode: impl Fn(&T, &mut Cursor<Vec<u8>>) -> std::io::Result<()>,