Serialize Patio compilation unit to Protobuf

1 files changed, 17 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java b/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java
index 4a0fb7af..9eaf4ef3 100644
--- a/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java
@@ -17,6 +17,23 @@ import static org.zwobble.precisely.AssertThat.assertThat;
 import static org.zwobble.precisely.Matchers.equalTo;
 
 public class PatioToProtobufTests {
+    // == Compilation unit ==
+
+    @Test
+    public void canSerializeCompilationUnit() {
+        var element = PatioFunctionDefinition.builder()
+            .withName("f")
+            .withReturnType(PatioTypes.VOID)
+            .build();
+        var compilationUnit = Patio.compilationUnit(List.of(element));
+
+        var deserializedCompilationUnit = PatioToProtobuf.deserializeCompilationUnit(
+            PatioToProtobuf.serializeCompilationUnit(compilationUnit)
+        );
+
+        assertThat(deserializedCompilationUnit, equalTo(compilationUnit));
+    }
+
     // == Compilation unit elements ==
 
     @Test