Merge pull request #132 from ldmonster/master

[template] add when/then for functions without results

Author: zcolleen

internal/templates/body.tmpl

@@ -185,7 +185,7 @@ func New{{$mock}}{{(params)}}(t minimock.Tester) *{{$mock}}{{(paramsRef)}} {
         return {{$m}}.mock
     }
 
-    {{if (and $method.HasParams $method.HasResults)}}
+    {{if $method.HasParams }}
         // When sets expectation for the {{$.Interface.Name}}.{{$method.Name}} which will trigger the result defined by the following
         // Then helper
         func ({{$m}} *m{{$mock}}{{$method.Name}}{{(paramsRef)}}) When({{$method.Params}}) *{{$mock}}{{$method.Name}}Expectation{{(paramsRef)}} {
@@ -202,11 +202,17 @@ func New{{$mock}}{{(params)}}(t minimock.Tester) *{{$mock}}{{(paramsRef)}} {
             return expectation
         }
 
-        // Then sets up {{$.Interface.Name}}.{{$method.Name}} return parameters for the expectation previously defined by the When method
+        // Then sets up {{$.Interface.Name}}.{{$method.Name}} return parameters for the expectation previously defined by the When method{{if $method.HasResults }}
         func (e *{{$mock}}{{$method.Name}}Expectation{{(paramsRef)}}) Then({{$method.Results}}) *{{$mock}}{{(paramsRef)}} {
             e.results = &{{$mock}}{{$method.Name}}Results{{(paramsRef)}}{ {{ $method.ResultsNames }} }
             return e.mock
         }
+        {{end}}
+        {{if not $method.HasResults }}
+        func (e *{{$mock}}{{$method.Name}}Expectation{{(paramsRef)}}) Then() *{{$mock}}{{(paramsRef)}} {
+            return e.mock
+        }
+        {{end}}
     {{end}}
 
     // Times sets number of times {{$.Interface.Name}}.{{$method.Name}} should be invoked

tests/context_accepter_mock.go

@@ -48,6 +48,16 @@ func TestContextAccepterMock_WhenThenMatchAnycontext(t *testing.T) {
 	assert.Equal(t, 42, result)
 }
 
+func TestContextAccepterMock_WhenThenMatchAnycontextWithoutArgs(t *testing.T) {
+	tester := NewTesterMock(t)
+	tester.CleanupMock.Return().HelperMock.Return()
+
+	mock := NewContextAccepterMock(tester).
+		AcceptContextMock.When(minimock.AnyContext).Then()
+
+	mock.AcceptContext(context.TODO())
+}
+
 func TestContextAccepterMock_DiffWithoutAnyContext(t *testing.T) {
 	tester := NewTesterMock(t)
 	tester.CleanupMock.Return().HelperMock.Return()