Added tests around route generation

This commits adds a set of tests around custom context specification as part of the generator arguments.

Author: Alex Davies-Moore

plugin/src/main/java/com/flit/protoc/gen/server/undertow/RpcGenerator.java

@@ -21,9 +21,9 @@ class RpcGenerator extends BaseGenerator {
             this.context = "/twirp";
         } else {
             context = context.trim();
-            if (context.equals("")) {
+            if (context.equals("") || context.equals("/")) {
                 // empty route - i.e. top level "/"
-                this.context = context;
+                this.context = "";
             } else if (context.startsWith("/")) {
                 this.context = context;
             } else {

plugin/src/test/java/com/flit/protoc/gen/server/undertow/ContextGeneratorTest.java

@@ -0,0 +1,67 @@
+package com.flit.protoc.gen.server.undertow;
+
+import com.flit.protoc.Plugin;
+import com.google.protobuf.compiler.PluginProtos;
+import org.junit.Test;
+
+import java.util.Map;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * Tests the generation of a service that has core definition imported from another file
+ */
+public class ContextGeneratorTest extends BaseGeneratorTest {
+
+    @Test
+    public void test_GenerateWithMissingRoot() throws Exception {
+        test_Route("context.missing.undertow.bin", "/twirp/com.example.context.NullService");
+    }
+
+    @Test
+    public void test_GenerateWithEmptyRoot() throws Exception {
+        test_Route("context.empty.undertow.bin", "/twirp/com.example.context.NullService");
+    }
+
+    @Test
+    public void test_GenerateWithSlashOnlyRoot() throws Exception {
+        test_Route("context.slash.undertow.bin", "/com.example.context.NullService");
+    }
+
+    @Test
+    public void test_GenerateWithSlashRoot() throws Exception {
+        test_Route("context.root.undertow.bin", "/root/com.example.context.NullService");
+    }
+
+    @Test
+    public void test_GenerateWithNameRoot() throws Exception {
+        test_Route("context.name.undertow.bin", "/fibble/com.example.context.NullService");
+    }
+
+    private void test_Route(String file, String route) throws Exception {
+        PluginProtos.CodeGeneratorRequest request = load(file);
+
+        Plugin plugin = new Plugin(request);
+        PluginProtos.CodeGeneratorResponse response = plugin.process();
+
+        assertNotNull(response);
+        assertEquals(2, response.getFileCount());
+
+        Map<String, PluginProtos.CodeGeneratorResponse.File> files = response.getFileList().stream().collect(Collectors.toMap(
+            PluginProtos.CodeGeneratorResponse.File::getName,
+            Function.identity()
+        ));
+
+        assertTrue(files.containsKey("com/example/context/rpc/RpcNullServiceService.java"));
+        assertTrue(files.containsKey("com/example/context/rpc/RpcNullServiceHandler.java"));
+
+        assertTrue(files.get("com/example/context/rpc/RpcNullServiceHandler.java").getContent().contains(
+            String.format("public static final String ROUTE = \"%s\";", route)
+        ));
+    }
+
+}

plugin/src/test/resources/context.empty.undertow.bin

@@ -0,0 +1,10 @@
+syntax = "proto3";
+
+package com.example.context;
+option java_package = "com.example.context.rpc";
+
+message Empty {}
+
+service NullService {
+    rpc SayNull(Empty) returns (Empty);
+}