Allow other calls to be ignored. Closes #4.

Author: ryanplusplus

README.md

@@ -150,6 +150,19 @@ f:should_be_called_with(1):
   end)
 ```
 
+## Ignoring Other calls
+
+```javascript
+local mach = require 'mach'
+
+local f = mach.mock_function('f')
+
+f:should_be_called().and_other_calls_should_be_ignored():when(function()
+  f()
+  f(1)
+end)
+```
+
 ## Flexible Syntax
 
 ```lua

spec/mach_spec.lua

@@ -439,4 +439,11 @@ describe('The mach library', function()
       mach.mock_method()(1, 2, 3)
     end)
   end)
+
+  it('should allow additional mocked calls to be ignored', function()
+    f1:should_be_called():and_other_calls_should_be_ignored():when(function()
+      f1()
+      f2()
+    end)
+  end)
 end)

src/mach/Expectation.lua

@@ -43,8 +43,6 @@ function expectation:when(thunk)
   end
 
   local function called(m, name, args)
-    assert(#self._calls > 0, 'unexpected call')
-
     local valid_function_found = false
 
     for i, call in ipairs(self._calls) do
@@ -69,10 +67,12 @@ function expectation:when(thunk)
       if call:has_fixed_order() then break end
     end
 
-    if not valid_function_found then
-      unexpected_call_error(name, args, 2)
-    else
-      unexpected_args_error(name, args, 2)
+    if not self._ignore_other_calls then
+      if not valid_function_found then
+        unexpected_call_error(name, args, 2)
+      else
+        unexpected_args_error(name, args, 2)
+      end
     end
   end
 
@@ -155,4 +155,9 @@ function expectation:multiple_times(times)
   return self
 end
 
+function expectation:and_other_calls_should_be_ignored()
+  self._ignore_other_calls = true
+  return self
+end
+
 return create