1 files changed, 31 insertions, 0 deletions
diff --git a/third_party/ocmock/OCMockObjectHamcrestTests.mm b/third_party/ocmock/OCMockObjectHamcrestTests.mm
new file mode 100644
index 0000000..8059deb
--- /dev/null
+++ b/third_party/ocmock/OCMockObjectHamcrestTests.mm
@@ -0,0 +1,31 @@
+//---------------------------------------------------------------------------------------
+//  $Id: OCMockObjectTests.m 21 2008-01-24 18:59:39Z erik $
+//  Copyright (c) 2004-2008 by Mulle Kybernetik. See License file for details.
+//---------------------------------------------------------------------------------------
+
+#import <OCMock/OCMock.h>
+#import "OCMockObjectHamcrestTests.h"
+
+#define HC_SHORTHAND
+#import <hamcrest/hamcrest.h>
+
+
+@implementation OCMockObjectHamcrestTests
+
+- (void)testAcceptsStubbedMethodWithHamcrestConstraint
+{
+	id mock = [OCMockObject mockForClass:[NSString class]];
+	[[mock stub] hasSuffix:(id)startsWith(@"foo")];
+	[mock hasSuffix:@"foobar"];
+}
+
+
+- (void)testRejectsUnstubbedMethodWithHamcrestConstraint
+{
+	id mock = [OCMockObject mockForClass:[NSString class]];
+	[[mock stub] hasSuffix:(id)anyOf(equalTo(@"foo"), equalTo(@"bar"), NULL)];
+	STAssertThrows([mock hasSuffix:@"foobar"], @"Should have raised an exception.");
+}
+
+
+@end