Add serialization of sources to flatbuffers

1 files changed, 27 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/sources/serialization/SourceToFlatbuffersTests.java b/src/test/java/org/shedlang/compiler/sources/serialization/SourceToFlatbuffersTests.java
new file mode 100644
index 00000000..31d20149
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/sources/serialization/SourceToFlatbuffersTests.java
@@ -0,0 +1,27 @@
+package org.shedlang.compiler.sources.serialization;
+
+import com.google.flatbuffers.FlatBufferBuilder;
+import com.google.protobuf.InvalidProtocolBufferException;
+import org.shedlang.compiler.sources.SourceSerializationTests;
+import org.shedlang.compiler.sources.flatbuffers.SourceFbs;
+import org.shedlang.compiler.sources.protobuf.SourcesProtobuf;
+
+import java.nio.ByteBuffer;
+
+public class SourceToFlatbuffersTests extends SourceSerializationTests {
+    public SourceToFlatbuffersTests() {
+        super(
+            (source, parentSource) -> {
+                var builder = new FlatBufferBuilder();
+                var sourceFbs = SourceToFlatbuffers.serializeSource(builder, source, parentSource);
+                builder.finish(sourceFbs);
+                return builder.sizedByteArray();
+            },
+
+            (serializedSource, parentSource) -> {
+                var sourceFbs = SourceFbs.getRootAsSourceFbs(ByteBuffer.wrap(serializedSource));
+                return SourceToFlatbuffers.deserializeSource(sourceFbs, parentSource);
+            }
+        );
+    }
+}