Fix flaky iOS unit tests

lib/ios/ReactNativeNavigationTests/RNNCommandsHandlerTest.m

@@ -290,14 +290,10 @@
 }
 
 - (void)testSetStackRoot_callRenderTreeAndWaitOnce {
-	id vc1Mock = OCMPartialMock(_vc1);
-	id vc2Mock = OCMPartialMock(_vc2);
-	OCMStub([vc1Mock renderTreeAndWait:NO perform:[OCMArg any]]);
-	OCMStub([vc2Mock renderTreeAndWait:NO perform:[OCMArg any]]);
-	
+	id vc1Mock = [OCMockObject partialMockForObject:_vc1];
+	id vc2Mock = [OCMockObject partialMockForObject:_vc2];
 	NSArray* newViewControllers = @[vc1Mock, vc2Mock];
-	id classMock = OCMClassMock([RNNLayoutManager class]);
-	OCMStub(ClassMethod([classMock findComponentForId:@"vc1"])).andReturn(_nvc);
+	
 	OCMStub([self.controllerFactory createChildrenLayout:[OCMArg any]]).andReturn(newViewControllers);
 	[self.uut setReadyToReceiveCommands:true];
 	[self.uut setStackRoot:@"vc1" commandId:@"" children:nil completion:^{
@@ -314,12 +310,11 @@
 	_vc2.options.animations.setStackRoot.waitForRender = [[Bool alloc] initWithBOOL:YES];
 	id vc1Mock = OCMPartialMock(_vc1);
 	id vc2Mock = OCMPartialMock(_vc2);
-	OCMStub([vc1Mock renderTreeAndWait:NO perform:[OCMArg any]]);
-	OCMStub([vc2Mock renderTreeAndWait:YES perform:[OCMArg any]]);
+	
 	NSArray* newViewControllers = @[vc1Mock, vc2Mock];
-	id classMock = OCMClassMock([RNNLayoutManager class]);
-	OCMStub(ClassMethod([classMock findComponentForId:@"vc1"])).andReturn(_nvc);
+	
 	OCMStub([self.controllerFactory createChildrenLayout:[OCMArg any]]).andReturn(newViewControllers);
+	
 	[self.uut setReadyToReceiveCommands:true];
 	[self.uut setStackRoot:@"vc1" commandId:@"" children:nil completion:^{