Support sum types in Java transient-0

1 files changed, 17 insertions, 0 deletions

diff --git a/examples/10-transient-0/output/java-junit/src/test/java/org/zwobble/example/Transient0Tests.java b/examples/10-transient-0/output/java-junit/src/test/java/org/zwobble/example/Transient0Tests.java
index 7d6f7de..e28e926 100644
--- a/examples/10-transient-0/output/java-junit/src/test/java/org/zwobble/example/Transient0Tests.java
+++ b/examples/10-transient-0/output/java-junit/src/test/java/org/zwobble/example/Transient0Tests.java
@@ -19,6 +19,8 @@ import org.zwobble.example.types.data.StructWithInt64;
 import org.zwobble.example.types.data.StructWithList;
 import org.zwobble.example.types.data.StructWithOption;
 import org.zwobble.example.types.data.StructWithString;
+import org.zwobble.example.types.data.VariantOne;
+import org.zwobble.example.types.data.VariantTwo;
 
 public class Transient0Tests {
     @Test
@@ -143,6 +145,21 @@ public class Transient0Tests {
         );
     }
 
+    @Test
+    public void sumWithVariants() throws IOException {
+        assertRoundTripEncoding(
+            new VariantOne(10),
+            HobgoblinTransient0Data::encodeSumWithVariants,
+            HobgoblinTransient0Data::decodeSumWithVariants
+        );
+
+        assertRoundTripEncoding(
+            new VariantTwo(25),
+            HobgoblinTransient0Data::encodeSumWithVariants,
+            HobgoblinTransient0Data::decodeSumWithVariants
+        );
+    }
+
     private <T> void assertRoundTripEncoding(
         T value,
         Encoder<T> encoder,