OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #import "chrome/common/mac/objc_method_swizzle.h" | |
6 | |
7 #include "base/mac/scoped_nsobject.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 @interface ObjcMethodSwizzleTest : NSObject | |
11 - (id)self; | |
12 | |
13 - (NSInteger)one; | |
14 - (NSInteger)two; | |
15 @end | |
16 | |
17 @implementation ObjcMethodSwizzleTest : NSObject | |
18 - (id)self { | |
19 return [super self]; | |
20 } | |
21 | |
22 - (NSInteger)one { | |
23 return 1; | |
24 } | |
25 - (NSInteger)two { | |
26 return 2; | |
27 } | |
28 @end | |
29 | |
30 @interface ObjcMethodSwizzleTest (ObjcMethodSwizzleTestCategory) | |
31 - (NSUInteger)hash; | |
32 @end | |
33 | |
34 @implementation ObjcMethodSwizzleTest (ObjcMethodSwizzleTestCategory) | |
35 - (NSUInteger)hash { | |
36 return [super hash]; | |
37 } | |
38 @end | |
39 | |
40 namespace ObjcEvilDoers { | |
41 | |
42 TEST(ObjcMethodSwizzleTest, GetImplementedInstanceMethod) { | |
43 EXPECT_EQ(class_getInstanceMethod([NSObject class], @selector(dealloc)), | |
44 GetImplementedInstanceMethod([NSObject class], @selector(dealloc))); | |
45 EXPECT_EQ(class_getInstanceMethod([NSObject class], @selector(self)), | |
46 GetImplementedInstanceMethod([NSObject class], @selector(self))); | |
47 EXPECT_EQ(class_getInstanceMethod([NSObject class], @selector(hash)), | |
48 GetImplementedInstanceMethod([NSObject class], @selector(hash))); | |
49 | |
50 Class testClass = [ObjcMethodSwizzleTest class]; | |
51 EXPECT_EQ(class_getInstanceMethod(testClass, @selector(self)), | |
52 GetImplementedInstanceMethod(testClass, @selector(self))); | |
53 EXPECT_NE(class_getInstanceMethod([NSObject class], @selector(self)), | |
54 class_getInstanceMethod(testClass, @selector(self))); | |
55 | |
56 EXPECT_TRUE(class_getInstanceMethod(testClass, @selector(dealloc))); | |
57 EXPECT_FALSE(GetImplementedInstanceMethod(testClass, @selector(dealloc))); | |
58 } | |
59 | |
60 TEST(ObjcMethodSwizzleTest, SwizzleImplementedInstanceMethods) { | |
61 base::scoped_nsobject<ObjcMethodSwizzleTest> object( | |
62 [[ObjcMethodSwizzleTest alloc] init]); | |
63 EXPECT_EQ([object one], 1); | |
64 EXPECT_EQ([object two], 2); | |
65 | |
66 Class testClass = [object class]; | |
67 SwizzleImplementedInstanceMethods(testClass, @selector(one), @selector(two)); | |
68 EXPECT_EQ([object one], 2); | |
69 EXPECT_EQ([object two], 1); | |
70 | |
71 SwizzleImplementedInstanceMethods(testClass, @selector(one), @selector(two)); | |
72 EXPECT_EQ([object one], 1); | |
73 EXPECT_EQ([object two], 2); | |
74 } | |
75 | |
76 } // namespace ObjcEvilDoers | |
OLD | NEW |