Serialize Patio function declarations to Protobuf

1 files changed, 18 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 25054ae2..35ec4221 100644
--- a/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/protobuf/PatioToProtobufTests.java
@@ -1,6 +1,7 @@
 package org.shedlang.compiler.patio.protobuf;
 
 import org.junit.jupiter.api.Test;
+import org.shedlang.compiler.patio.PatioFunctionDeclaration;
 import org.shedlang.compiler.patio.PatioVisibility;
 import org.shedlang.compiler.patio.instructions.Patio;
 import org.shedlang.compiler.patio.types.PatioDataLayouts;
@@ -78,6 +79,23 @@ public class PatioToProtobufTests {
         assertThat(deserializedElement, equalTo(element));
     }
 
+    @Test
+    public void canSerializeFunctionDeclaration() {
+        var element = PatioFunctionDeclaration.builder()
+            .withName("f")
+            .addParam(Patio.param("param1", PatioTypes.INT))
+            .addParam(Patio.param("param2", PatioTypes.DATA_PTR))
+            .withReturnType(PatioTypes.VOID)
+            .build();
+
+        var deserializedElement = PatioToProtobuf.deserializeCompilationUnitElement(
+            PatioToProtobuf.serializeCompilationUnitElement(element, Sources.NULL),
+            Sources.NULL
+        );
+
+        assertThat(deserializedElement, equalTo(element));
+    }
+
     // == Data ptr values ==
 
     @Test