Allow expected calls that don't match arguments. Closes #9.

Author: ryanplusplus

spec/mach_spec.lua

@@ -55,6 +55,10 @@ describe('The mach library', function()
     end)
   end)
 
+  it('should allow you to expect a function to be called with any arguments', function()
+    f:should_be_called_with_any_arguments():when(function() f(1, '3') end)
+  end)
+
   it('should allow you to specify the return value of a mocked function', function()
     f:should_be_called():and_will_return(4):when(function()
       assert.is.equal(f(), 4)
@@ -318,6 +322,12 @@ describe('The mach library', function()
     end)
   end)
 
+  it('should fail if should_be_called_with_any_arguments is used after a call has already been specified', function()
+    should_fail_with('call already specified', function()
+      f:should_be_called():should_be_called_with_any_arguments()
+    end)
+  end)
+
   it('should allow calls to happen out of order when and_also is used', function()
     f1:should_be_called():
       and_also(f2:should_be_called()):

src/mach/Expectation.lua

@@ -111,18 +111,28 @@ function expectation:and_also(other)
   return self
 end
 
+function expectation:should_be_called_with_any_arguments()
+  if self._call_specified then
+    error('call already specified', 2)
+  end
+
+  self._call_specified = true
+  table.insert(self._calls, ExpectedCall(self._m, { required = true, ignore_args = true }))
+  return self
+end
+
 function expectation:should_be_called_with(...)
-  if self._call_specified == true then
+  if self._call_specified then
     error('call already specified', 2)
   end
 
   self._call_specified = true
-  table.insert(self._calls, ExpectedCall(self._m, true, table.pack(...)))
+  table.insert(self._calls, ExpectedCall(self._m, { required = true, args = table.pack(...) }))
   return self
 end
 
 function expectation:should_be_called()
-  if self._call_specified == true then
+  if self._call_specified then
     error('call already specified', 2)
   end
 
@@ -135,7 +145,7 @@ function expectation:may_be_called_with(...)
   end
 
   self._call_specified = true
-  table.insert(self._calls, ExpectedCall(self._m, false, table.pack(...)))
+  table.insert(self._calls, ExpectedCall(self._m, { required = false, args = table.pack(...) }))
   return self
 end
 

src/mach/ExpectedCall.lua

@@ -1,12 +1,13 @@
 local expected_call = {}
 expected_call.__index = expected_call
 
-local function create(f, required, args)
+local function create(f, config)
   local o = {
     _f = f,
     _ordered = false,
-    _required = required,
-    _args = args,
+    _required = config.required,
+    _args = config.args,
+    _ignore_args = config.ignore_args,
     _return = {}
   }
 
@@ -20,6 +21,7 @@ function expected_call:function_matches(f)
 end
 
 function expected_call:args_match(args)
+  if self._ignore_args then return true end
   if #self._args ~= #args then return false end
 
   for k in ipairs(self._args) do