Simplified the stub and expect macros.

Maybe I did too much JavaScript recently...
twobitlabs · Apr 25, 2014 · 6759e85 · 6759e85
1 parent 48224f0
commit 6759e85
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/Source/OCMock/OCMock.h b/Source/OCMock/OCMock.h
@@ -24,18 +24,18 @@
 
 #define OCMObserverMock() [OCMockObject observerMock]
 
-#define OCMStub(invocation) (^ () \
-{ \
+#define OCMStub(invocation) \
+({ \
     [OCMMacroState beginStubMacro]; \
     invocation; \
-    return [OCMMacroState endStubMacro]; \
-})()
+    [OCMMacroState endStubMacro]; \
+})
 
-#define OCMExpect(invocation) (^ () \
-{ \
+#define OCMExpect(invocation) \
+({ \
     [OCMMacroState beginExpectMacro]; \
     invocation; \
-    return [OCMMacroState endExpectMacro]; \
-})()
+    [OCMMacroState endExpectMacro]; \
+})
 
 #define OCMVerifyAll(mock) [mock verifyAtLocation:OCMMakeLocation(self, __FILE__, __LINE__)]