fix MockBuilder (#26)

Author: h908714124

compiler/src/main/java/io/jbock/simple/processor/writing/MockBuilder.java

@@ -131,6 +131,8 @@ private List<MethodSpec> getMethods() {
                 FieldSpec auxField = FieldSpec.builder(TypeName.BOOLEAN, namedBinding.auxName(), PRIVATE).build();
                 method.addStatement("this.$N = $L", auxField, true);
             }
+            method.addStatement("return this");
+            method.returns(getClassName());
             methods.add(method.build());
         }
         return methods;

compiler/src/test/java/io/jbock/simple/processor/BuilderTest.java

@@ -205,8 +205,9 @@ void builderParameter() {
                         "      return new Builder_Impl(this);",
                         "    }",
                         "",
-                        "    public void a(TestClass.A a) {",
+                        "    public MockBuilder a(TestClass.A a) {",
                         "      this.a = a;",
+                        "      return this;",
                         "    }",
                         "  }",
                         "}");

compiler/src/test/java/io/jbock/simple/processor/PrimitiveTest.java

@@ -108,13 +108,15 @@ void providesPrimitive() {
                         "      return new Factory_Impl(this);",
                         "    }",
                         "",
-                        "    public void b(int b) {",
+                        "    public MockBuilder b(int b) {",
                         "      this.b = b;",
                         "      this.b_isSet = true;",
+                        "      return this;",
                         "    }",
                         "",
-                        "    public void a(TestClass.A a) {",
+                        "    public MockBuilder a(TestClass.A a) {",
                         "      this.a = a;",
+                        "      return this;",
                         "    }",
                         "  }",
                         "}");

compiler/src/test/java/io/jbock/simple/processor/ProcessorComponentTest.java

@@ -81,14 +81,17 @@ void staticMethodBindings() {
                         "      TestClass.A a = this.a != null ? this.a : new TestClass.A(b);",
                         "      return new TestClass_AComponent_Impl(a);",
                         "    }",
-                        "    public void c(TestClass.C c) {",
+                        "    public MockBuilder c(TestClass.C c) {",
                         "      this.c = c;",
+                        "      return this;",
                         "    }",
-                        "    public void b(TestClass.B b) {",
+                        "    public MockBuilder b(TestClass.B b) {",
                         "      this.b = b;",
+                        "      return this;",
                         "    }",
-                        "    public void a(TestClass.A a) {",
+                        "    public MockBuilder a(TestClass.A a) {",
                         "      this.a = a;",
+                        "      return this;",
                         "    }",
                         "  }",
                         "}");
@@ -172,20 +175,24 @@ void dependencyDiamond() {
                         "      return new TestClass_AComponent_Impl(a);",
                         "    }",
                         "",
-                        "    void e(TestClass.E e) {",
+                        "    MockBuilder e(TestClass.E e) {",
                         "      this.e = e;",
+                        "      return this;",
                         "    }",
                         "",
-                        "    void c(TestClass.C c) {",
+                        "    MockBuilder c(TestClass.C c) {",
                         "      this.c = c;",
+                        "      return this;",
                         "    }",
                         "",
-                        "    void b(TestClass.B b) {",
+                        "    MockBuilder b(TestClass.B b) {",
                         "      this.b = b;",
+                        "      return this;",
                         "    }",
                         "",
-                        "    void a(TestClass.A a) {",
+                        "    MockBuilder a(TestClass.A a) {",
                         "      this.a = a;",
+                        "      return this;",
                         "    }",
                         "  }",
                         "}");