refactor: improve speed of mock

ab180 · Dec 1, 2023 · 25edc55 · 25edc55
1 parent b57d4b4
commit 25edc55
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Source/TestMock/MockFunction.swift b/Source/TestMock/MockFunction.swift
@@ -117,11 +117,10 @@ extension MockFunction {
 
 extension MockFunction {
     public func callAsFunction(_ argument: Argument) -> Result {
-        defer { lock.unlock() }
-        lock.lock()
-
         let result = implementation(argument)
+        lock.lock()
         calls.append(MockFunctionCall(argument: argument, result: result))
+        lock.unlock()
         self.callback(argument, result)
 
         return result