Support for mocking callable tables in objects and tables

Author: ryanplusplus

mock.lua

@@ -159,12 +159,18 @@ function Mock:mockMethod(name)
   return m
 end
 
+function IsCallable(x)
+  local isFunction = type(x) == 'function'
+  local hasCallMetamethod = type((debug.getmetatable(x) or {}).__call) == 'function'
+  return isFunction or hasCallMetamethod
+end
+
 function Mock:mockTable(t, name)
   name = name or '<anonymous>'
   local mocked = {}
 
   for k, v in pairs(t) do
-    if type(v) == 'function' then
+    if IsCallable(v) then
       mocked[k] = self:mockFunction(name .. '.' .. tostring(k))
     end
   end
@@ -177,7 +183,7 @@ function Mock:mockObject(o, name)
   local mocked = {}
 
   for k, v in pairs(o) do
-    if type(v) == 'function' then
+    if IsCallable(v) then
       mocked[k] = self:mockMethod(name .. ':' .. tostring(k))
     end
   end

spec/mock_spec.lua

@@ -194,6 +194,32 @@ describe('The mock library', function()
     end)
   end)
 
+  it('should allow mocking of any callable in an object, not just functions', function()
+    local someTable = {
+      foo = {}
+    }
+
+    setmetatable(someTable.foo, {__call = function() end})
+
+    local mockedTable = mock:mockTable(someTable)
+
+    mock(mockedTable.foo):shouldBeCalled():
+    when(function() mockedTable.foo() end)
+  end)
+
+  it('should allow mocking of any callable in a table, not just functions', function()
+    local someObject = {
+      foo = {}
+    }
+
+    setmetatable(someObject.foo, {__call = function() end})
+
+    local mockedObject = mock:mockObject(someObject)
+
+    mock(mockedObject.foo):shouldBeCalled():
+    when(function() mockedObject:foo() end)
+  end)
+
   it('should fail when a method is incorrectly used as a function', function()
     shouldFail(function()
       local someObject = {}