//A function which is the implementation of original method. 
//The function must take at least two arguments—self and _cmd.
void (*gOriginalViewDidLoad)(id, SEL);

- (void)tearDown {
    RRO_REVERT_SWIZZLE_ALL_METHOD([UIViewController class]);
}

- (void)testThatViewDidLoadIsCalledWhenShowWindow {
  __block BOOL isViewDidLoadCalled = NO;
  
  void (^viewDidLoadBlock)(UIViewController *, SEL) = ^void (UIViewController *nav, SEL _cmd) {
    isViewDidLoadCalled = true;
    NSLog(@"viewDidLoad called here");

    //call original @selector(viewDidLoad) if needed
    gOriginalViewDidLoad(nav, _cmd); 
  };
  
  RRO_SWIZZLE_BLOCK([UIViewController class], @selector(viewDidLoad), viewDidLoadBlock, &gOriginalViewDidLoad);
  
  UIWindow *window = [[UIWindow alloc] initWithFrame:[UIScreen mainScreen].bounds];
  window.rootViewController = [[UIViewController alloc] init];
  [window makeKeyAndVisible];
  
  XCTAssertTrue(isViewDidLoadCalled);
}