Fix back button initialization on setStackRoot (#5246)

lib/ios/RNNNavigationControllerPresenter.m

 	[navigationController rnn_setNavigationBarLargeTitleFontFamily:[options.topBar.largeTitle.fontFamily getWithDefaultValue:nil] fontSize:[options.topBar.largeTitle.fontSize getWithDefaultValue:nil] color:[options.topBar.largeTitle.color getWithDefaultValue:nil]];
 	[navigationController rnn_setNavigationBarFontFamily:[options.topBar.title.fontFamily getWithDefaultValue:nil] fontSize:[options.topBar.title.fontSize getWithDefaultValue:nil] color:[options.topBar.title.color getWithDefaultValue:nil]];
 	[navigationController rnn_setBackButtonColor:[options.topBar.backButton.color getWithDefaultValue:nil]];
-	[navigationController rnn_setBackButtonIcon:[options.topBar.backButton.icon getWithDefaultValue:nil] withColor:[options.topBar.backButton.color getWithDefaultValue:nil] title:[options.topBar.backButton.showTitle getWithDefaultValue:YES] ? [options.topBar.backButton.title getWithDefaultValue:nil] : @""];
 }
 
 - (void)applyOptionsOnWillMoveToParentViewController:(RNNNavigationOptions *)options {
 	[super applyOptionsOnWillMoveToParentViewController:options];
-	
-	RNNNavigationController* navigationController = self.bindedViewController;
-	[navigationController rnn_setBackButtonIcon:[options.topBar.backButton.icon getWithDefaultValue:nil] withColor:[options.topBar.backButton.color getWithDefaultValue:nil] title:[options.topBar.backButton.showTitle getWithDefaultValue:YES] ? [options.topBar.backButton.title getWithDefaultValue:nil] : @""];
 }
 
 - (void)applyOptionsOnViewDidLayoutSubviews:(RNNNavigationOptions *)options {
 		[navigationController rnn_setNavigationBarLargeTitleVisible:newOptions.topBar.largeTitle.visible.get];
 	}
 	
-	if (newOptions.topBar.backButton.icon.hasValue || newOptions.topBar.backButton.showTitle.hasValue || newOptions.topBar.backButton.color.hasValue || newOptions.topBar.backButton.title.hasValue) {
-		[navigationController rnn_setBackButtonIcon:[newOptions.topBar.backButton.icon getWithDefaultValue:nil] withColor:[newOptions.topBar.backButton.color getWithDefaultValue:nil] title:[newOptions.topBar.backButton.showTitle getWithDefaultValue:YES] ? [newOptions.topBar.backButton.title getWithDefaultValue:nil] : @""];
-		
-	}
-	
 	if (newOptions.topBar.backButton.color.hasValue) {
 		[navigationController rnn_setBackButtonColor:newOptions.topBar.backButton.color.get];
 	}

lib/ios/RNNViewControllerPresenter.m

 	_navigationButtons = [[RNNNavigationButtons alloc] initWithViewController:self.bindedViewController componentRegistry:_componentRegistry];
 }
 
+- (void)applyOptionsOnWillMoveToParentViewController:(RNNNavigationOptions *)options {
+	[super applyOptionsOnWillMoveToParentViewController:options];
+	UIViewController* viewController = self.bindedViewController;
+	[viewController rnn_setBackButtonIcon:[options.topBar.backButton.icon getWithDefaultValue:nil] withColor:[options.topBar.backButton.color getWithDefaultValue:nil] title:[options.topBar.backButton.showTitle getWithDefaultValue:YES] ? [options.topBar.backButton.title getWithDefaultValue:nil] : @""];
+}
+
 - (void)applyOptions:(RNNNavigationOptions *)options {
 	[super applyOptions:options];
 	
 	if (newOptions.topBar.title.component.name.hasValue) {
 		[self setCustomNavigationTitleView:newOptions perform:nil];
 	}
+	
+	if (newOptions.topBar.backButton.icon.hasValue || newOptions.topBar.backButton.showTitle.hasValue || newOptions.topBar.backButton.color.hasValue || newOptions.topBar.backButton.title.hasValue) {
+		[viewController rnn_setBackButtonIcon:[newOptions.topBar.backButton.icon getWithDefaultValue:nil] withColor:[newOptions.topBar.backButton.color getWithDefaultValue:nil] title:[newOptions.topBar.backButton.showTitle getWithDefaultValue:YES] ? [newOptions.topBar.backButton.title getWithDefaultValue:nil] : @""];
+	}
 }
 
 - (void)renderComponents:(RNNNavigationOptions *)options perform:(RNNReactViewReadyCompletionBlock)readyBlock {

lib/ios/ReactNativeNavigationTests/RNNNavigationControllerPresenterTest.m

 	[_bindedViewController verify];
 }
 
-
-- (void)testApplyOptions_shouldSetBackButtonOnBindedViewController_withTitle {
-	Text* title = [[Text alloc] initWithValue:@"Title"];
-	self.options.topBar.backButton.title = title;
-	[[_bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:title.get];
-	[self.uut applyOptions:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptions_shouldSetBackButtonOnBindedViewController_withHideTitle {
-	Text* title = [[Text alloc] initWithValue:@"Title"];
-	self.options.topBar.backButton.title = title;
-	self.options.topBar.backButton.showTitle = [[Bool alloc] initWithValue:@(0)];
-	[[_bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:@""];
-	[self.uut applyOptions:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptions_shouldSetBackButtonOnBindedViewController_withIcon {
-	Image* image = [[Image alloc] initWithValue:[UIImage new]];
-	self.options.topBar.backButton.icon = image;
-	[[_bindedViewController expect] rnn_setBackButtonIcon:image.get withColor:nil title:nil];
-	[self.uut applyOptions:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withTitle {
-	Text* title = [[Text alloc] initWithValue:@"Title"];
-	self.options.topBar.backButton.title = title;
-	[[_bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:title.get];
-	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withHideTitle {
-	Text* title = [[Text alloc] initWithValue:@"Title"];
-	self.options.topBar.backButton.title = title;
-	self.options.topBar.backButton.showTitle = [[Bool alloc] initWithValue:@(0)];
-	[[_bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:@""];
-	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withIcon {
-	Image* image = [[Image alloc] initWithValue:[UIImage new]];
-	self.options.topBar.backButton.icon = image;
-	[[_bindedViewController expect] rnn_setBackButtonIcon:image.get withColor:nil title:nil];
-	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
-	[_bindedViewController verify];
-}
-
-- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withDefaultValues {
-	[[_bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:nil];
-	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
-	[_bindedViewController verify];
-}
-
 - (void)testApplyOptionsBeforePoppingShouldSetTopBarBackgroundForPoppingViewController {
 	_options.topBar.background.color = [[Color alloc] initWithValue:[UIColor redColor]];
 	

lib/ios/ReactNativeNavigationTests/RNNViewControllerPresenterTest.m

 	XCTAssertEqual(self.uut.bindedComponentId, @"componentId");
 }
 
+- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withTitle {
+	Text* title = [[Text alloc] initWithValue:@"Title"];
+	self.options.topBar.backButton.title = title;
+	[[(id)self.bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:title.get];
+	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
+	[(id)self.bindedViewController verify];
+}
+
+- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withHideTitle {
+	Text* title = [[Text alloc] initWithValue:@"Title"];
+	self.options.topBar.backButton.title = title;
+	self.options.topBar.backButton.showTitle = [[Bool alloc] initWithValue:@(0)];
+	[[(id)self.bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:@""];
+	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
+	[(id)self.bindedViewController verify];
+}
+
+- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withIcon {
+	Image* image = [[Image alloc] initWithValue:[UIImage new]];
+	self.options.topBar.backButton.icon = image;
+	[[(id)self.bindedViewController expect] rnn_setBackButtonIcon:image.get withColor:nil title:nil];
+	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
+	[(id)self.bindedViewController verify];
+}
+
+- (void)testApplyOptionsOnWillMoveToParent_shouldSetBackButtonOnBindedViewController_withDefaultValues {
+	[[(id)self.bindedViewController expect] rnn_setBackButtonIcon:nil withColor:nil title:nil];
+	[self.uut applyOptionsOnWillMoveToParentViewController:self.options];
+	[(id)self.bindedViewController verify];
+}
 
 
 - (RNNLayoutInfo *)createLayoutInfoWithComponentId:(NSString *)componentId {

lib/ios/ReactNativeNavigationTests/UINavigationController+RNNOptionsTest.m

     [super setUp];
 }
 
-- (void)testSetBackButtonIcon_withColor_shouldSetColor {
-	UIViewController* viewController = [UIViewController new];
-	UINavigationController* nav = [[UINavigationController alloc] initWithRootViewController:viewController];
-	UIColor* color = [UIColor blackColor];
-	
-	[nav rnn_setBackButtonIcon:nil withColor:color title:nil];
-	XCTAssertEqual(color, viewController.navigationItem.backBarButtonItem.tintColor);
-}
-
-- (void)testSetBackButtonIcon_withColor_shouldSetTitle {
-	UIViewController* viewController = [UIViewController new];
-	UINavigationController* nav = [[UINavigationController alloc] initWithRootViewController:viewController];
-	NSString* title = @"Title";
-	
-	[nav rnn_setBackButtonIcon:nil withColor:nil title:title];
-	XCTAssertEqual(title, viewController.navigationItem.backBarButtonItem.title);
-}
-
-- (void)testSetBackButtonIcon_withColor_shouldSetIcon {
-	UIViewController* viewController = [UIViewController new];
-	UINavigationController* nav = [[UINavigationController alloc] initWithRootViewController:viewController];
-	UIImage* icon = [UIImage new];
-	
-	[nav rnn_setBackButtonIcon:icon withColor:nil title:nil];
-	XCTAssertEqual(icon, viewController.navigationItem.backBarButtonItem.image);
-}
-
-- (void)testSetBackButtonIcon_shouldSetTitleOnPreviousViewControllerIfExists {
-	UIViewController* viewController = [UIViewController new];
-	UIViewController* viewController2 = [UIViewController new];
-	UINavigationController* nav = [[UINavigationController alloc] init];
-	[nav setViewControllers:@[viewController, viewController2]];
-	NSString* title = @"Title";
-
-	[nav rnn_setBackButtonIcon:nil withColor:nil title:title];
-	XCTAssertEqual(title, viewController.navigationItem.backBarButtonItem.title);
-}
-
-
 @end

lib/ios/ReactNativeNavigationTests/UIViewController+LayoutProtocolTest.m

 #import <XCTest/XCTest.h>
 #import <OCMock/OCMock.h>
 #import "UIViewController+LayoutProtocol.h"
+#import "UIViewController+RNNOptions.h"
 #import "RNNViewControllerPresenter.h"
 #import "RCTConvert+Modal.h"
 
 	XCTAssertEqual(uut.viewControllers[1], child2);
 }
 
+- (void)testSetBackButtonIcon_withColor_shouldSetColor {
+	UIViewController* uut = [UIViewController new];
+	[[UINavigationController alloc] initWithRootViewController:uut];
+	UIColor* color = [UIColor blackColor];
+	
+	[uut rnn_setBackButtonIcon:nil withColor:color title:nil];
+	XCTAssertEqual(color, uut.navigationItem.backBarButtonItem.tintColor);
+}
+
+- (void)testSetBackButtonIcon_withColor_shouldSetTitle {
+	UIViewController* uut = [UIViewController new];
+	UINavigationController* nav = [[UINavigationController alloc] initWithRootViewController:uut];
+	NSString* title = @"Title";
+	
+	[uut rnn_setBackButtonIcon:nil withColor:nil title:title];
+	XCTAssertEqual(title, uut.navigationItem.backBarButtonItem.title);
+}
+
+- (void)testSetBackButtonIcon_withColor_shouldSetIcon {
+	UIViewController* uut = [UIViewController new];
+	UINavigationController* nav = [[UINavigationController alloc] initWithRootViewController:uut];
+	UIImage* icon = [UIImage new];
+	
+	[uut rnn_setBackButtonIcon:icon withColor:nil title:nil];
+	XCTAssertEqual(icon, uut.navigationItem.backBarButtonItem.image);
+}
+
+- (void)testSetBackButtonIcon_shouldSetTitleOnPreviousViewControllerIfExists {
+	UIViewController* uut = [UIViewController new];
+	UIViewController* viewController2 = [UIViewController new];
+	UINavigationController* nav = [[UINavigationController alloc] init];
+	[nav setViewControllers:@[uut, viewController2]];
+	NSString* title = @"Title";
+	
+	[uut rnn_setBackButtonIcon:nil withColor:nil title:title];
+	XCTAssertEqual(title, uut.navigationItem.backBarButtonItem.title);
+}
 
 @end

lib/ios/UINavigationController+RNNOptions.h

 
 - (void)rnn_setNavigationBarClipsToBounds:(BOOL)clipsToBounds;
 
-- (void)rnn_setBackButtonIcon:(UIImage *)icon withColor:(UIColor *)color title:(NSString *)title;
-
 - (void)rnn_setNavigationBarLargeTitleVisible:(BOOL)visible;
 
 - (void)rnn_setNavigationBarLargeTitleFontFamily:(NSString *)fontFamily fontSize:(NSNumber *)fontSize color:(UIColor *)color;

lib/ios/UINavigationController+RNNOptions.m

 #import "UINavigationController+RNNOptions.h"
 #import "RNNFontAttributesCreator.h"
-#import "UIImage+tint.h"
 
 const NSInteger BLUR_TOPBAR_TAG = 78264802;
 
 	}
 }
 
-- (void)rnn_setBackButtonIcon:(UIImage *)icon withColor:(UIColor *)color title:(NSString *)title {
-	UIBarButtonItem *backItem = [[UIBarButtonItem alloc] init];
-	if (icon) {
-		backItem.image = color
-		? [[icon withTintColor:color] imageWithRenderingMode:UIImageRenderingModeAlwaysOriginal]
-		: icon;
-		
-		[self.navigationBar setBackIndicatorImage:[UIImage new]];
-		[self.navigationBar setBackIndicatorTransitionMaskImage:[UIImage new]];
-	}
-	
-	UIViewController *lastViewControllerInStack = self.viewControllers.count > 1 ? [self.viewControllers objectAtIndex:self.viewControllers.count-2] : self.topViewController;
-	
-	backItem.title = title ? title : lastViewControllerInStack.navigationItem.title;
-	backItem.tintColor = color;
-	
-	lastViewControllerInStack.navigationItem.backBarButtonItem = backItem;
-}
-
 - (void)rnn_setBackButtonColor:(UIColor *)color {
 	self.navigationBar.tintColor = color;
 }

lib/ios/UIViewController+RNNOptions.h

 
 - (void)rnn_setInterceptTouchOutside:(BOOL)interceptTouchOutside;
 
+- (void)rnn_setBackButtonIcon:(UIImage *)icon withColor:(UIColor *)color title:(NSString *)title;
+
 - (BOOL)isModal;
 
 @end

lib/ios/UIViewController+RNNOptions.m

 #import "UIViewController+RNNOptions.h"
 #import <React/RCTRootView.h>
+#import "UIImage+tint.h"
 
 #define kStatusBarAnimationDuration 0.35
 const NSInteger BLUR_STATUS_TAG = 78264801;
 	}
 }
 
+- (void)rnn_setBackButtonIcon:(UIImage *)icon withColor:(UIColor *)color title:(NSString *)title {
+	UIBarButtonItem *backItem = [[UIBarButtonItem alloc] init];
+	if (icon) {
+		backItem.image = color
+		? [[icon withTintColor:color] imageWithRenderingMode:UIImageRenderingModeAlwaysOriginal]
+		: icon;
+		
+		[self.navigationController.navigationBar setBackIndicatorImage:[UIImage new]];
+		[self.navigationController.navigationBar setBackIndicatorTransitionMaskImage:[UIImage new]];
+	}
+	
+	UIViewController *lastViewControllerInStack = self.navigationController.viewControllers.count > 1 ? [self.navigationController.viewControllers objectAtIndex:self.navigationController.viewControllers.count-2] : self.navigationController.topViewController;
+	
+	backItem.title = title ? title : lastViewControllerInStack.navigationItem.title;
+	backItem.tintColor = color;
+	
+	lastViewControllerInStack.navigationItem.backBarButtonItem = backItem;
+}
+
 @end