Support sums in rust-transient-0

1 files changed, 16 insertions, 1 deletions

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<()>,