add compiler unit tests

Author: Tate

src/autogenerate/index.ts

@@ -90,7 +90,9 @@ function writeSerializers(types: ProtoTypes[], isTopLevel = true): string {
                 ) as MessageType;
                 res += `for (const key in msg.${field.name}) {
                   writer.writeMessage(${field.index}, {}, (_, mapWriter) => {
-                    mapWriter.${map.content.fields[0].write}(1, key as any);
+                    mapWriter.${
+                      map.content.fields[0].write
+                    }(1, key${printIfTypescript(" as any")});
                     mapWriter.${map.content.fields[1].write}(2, msg.foo[key]);
                   });
                 }`;
@@ -156,7 +158,11 @@ function writeSerializers(types: ProtoTypes[], isTopLevel = true): string {
                           }
                         }
                       }
-                      msg.${field.name}![key!] = value!;
+                      msg.${field.name}${printIfTypescript(
+                      "!"
+                    )}[key${printIfTypescript("!")}] = value${printIfTypescript(
+                      "!"
+                    )};
                     });`;
                   } else if (field.read === "readMessage") {
                     res += `

src/compile.ts

@@ -0,0 +1,50 @@
+import {
+  CodeGeneratorRequest,
+  CodeGeneratorResponse,
+} from "google-protobuf/google/protobuf/compiler/plugin_pb";
+import { generate } from "./autogenerate";
+import {
+  getProtobufTSFileName,
+  buildIdentifierTable,
+  getProtobufJSFileName,
+} from "./utils";
+import { format } from "prettier";
+
+export function compile(input: Uint8Array): CodeGeneratorResponse {
+  const request = CodeGeneratorRequest.deserializeBinary(input);
+  const isTypescript = request.getParameter()?.trim() === "typescript";
+  const response = new CodeGeneratorResponse();
+  response.setSupportedFeatures(
+    CodeGeneratorResponse.Feature.FEATURE_PROTO3_OPTIONAL
+  );
+
+  const identifierTable = buildIdentifierTable(request);
+
+  function writeFile(name: string, content: string) {
+    const file = new CodeGeneratorResponse.File();
+    file.setName(name);
+    file.setContent(
+      format(content, { parser: isTypescript ? "typescript" : "babel" })
+    );
+    response.addFile(file);
+  }
+
+  request.getProtoFileList().forEach((fileDescriptorProto) => {
+    const name = fileDescriptorProto.getName();
+    if (!name) {
+      return;
+    }
+
+    const protobufTs = generate(
+      fileDescriptorProto,
+      identifierTable,
+      isTypescript
+    );
+    writeFile(
+      isTypescript ? getProtobufTSFileName(name) : getProtobufJSFileName(name),
+      protobufTs
+    );
+  });
+
+  return response;
+}

src/compiler.ts

@@ -1,52 +1,8 @@
 #!/usr/bin/env node
 
 import { readFileSync } from "fs";
-import {
-  CodeGeneratorRequest,
-  CodeGeneratorResponse,
-} from "google-protobuf/google/protobuf/compiler/plugin_pb";
-import { generate } from "./autogenerate";
-import {
-  getProtobufTSFileName,
-  buildIdentifierTable,
-  getProtobufJSFileName,
-} from "./utils";
-import { format } from "prettier";
+import { compile } from "./compile";
 
 const input = readFileSync(process.stdin.fd);
-const request = CodeGeneratorRequest.deserializeBinary(input);
-const isTypescript = request.getParameter()?.trim() === "typescript";
-const response = new CodeGeneratorResponse();
-response.setSupportedFeatures(
-  CodeGeneratorResponse.Feature.FEATURE_PROTO3_OPTIONAL
-);
-
-const identifierTable = buildIdentifierTable(request);
-
-function writeFile(name: string, content: string) {
-  const file = new CodeGeneratorResponse.File();
-  file.setName(name);
-  file.setContent(
-    format(content, { parser: isTypescript ? "typescript" : "babel" })
-  );
-  response.addFile(file);
-}
-
-request.getProtoFileList().forEach((fileDescriptorProto) => {
-  const name = fileDescriptorProto.getName();
-  if (!name) {
-    return;
-  }
-
-  const protobufTs = generate(
-    fileDescriptorProto,
-    identifierTable,
-    isTypescript
-  );
-  writeFile(
-    isTypescript ? getProtobufTSFileName(name) : getProtobufJSFileName(name),
-    protobufTs
-  );
-});
-
+const response = compile(input);
 process.stdout.write(response.serializeBinary());