src/chrome/common/mac/objc_method_swizzle_unittest.mm

   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