add e2e test for "setTabBadge"

e2e/ScreenStyle.test.js

@@ -44,4 +44,10 @@ describe('screen style', () => {
     await element(by.id('scrollView')).swipe('down', 'fast');
     await expect(element(by.type('UINavigationBar'))).toBeVisible();
   });
+
+  it('set Tab Bar badge on a current Tab', async () => {
+    await elementByLabel('Switch to tab based app').tap();
+    await elementByLabel('Set Tab Badge').tap();
+    await expect(element(by.text('EnCyClOpEdIa'))).toBeVisible();
+  });
 });

lib/ios/RNNNavigationOptions.m

@@ -21,7 +21,7 @@
 	self.topBarButtonColor = [navigationOptions objectForKey:@"topBarButtonColor"];
 	self.topBarTranslucent = [navigationOptions objectForKey:@"topBarTranslucent"];
 	self.setTabBadge = [navigationOptions objectForKey:@"setTabBadge"];
-
+	
 	return self;
 }
 
@@ -81,24 +81,23 @@
 		viewController.navigationController.navigationBar.tintColor = buttonColor;
 	} else {
 		viewController.navigationController.navigationBar.tintColor = nil;
-
-	if (self.setTabBadge) {
-		NSString *badge = [RCTConvert NSString:self.setTabBadge];
-		if (viewController.navigationController) {
-			viewController.navigationController.tabBarItem.badgeValue = badge;
-    } else {
-			viewController.tabBarItem.badgeValue = badge;
-	}
-
-	}
-	
-	if (self.topBarTranslucent) {
-		if ([self.topBarTranslucent boolValue]) {
-			viewController.navigationController.navigationBar.translucent = YES;
-		} else {
-			viewController.navigationController.navigationBar.translucent = NO;
+		
+		if (self.setTabBadge) {
+			NSString *badge = [RCTConvert NSString:self.setTabBadge];
+			if (viewController.navigationController) {
+				viewController.navigationController.tabBarItem.badgeValue = badge;
+			} else {
+				viewController.tabBarItem.badgeValue = badge;
+			}
+		}
+		
+		if (self.topBarTranslucent) {
+			if ([self.topBarTranslucent boolValue]) {
+				viewController.navigationController.navigationBar.translucent = YES;
+			} else {
+				viewController.navigationController.navigationBar.translucent = NO;
+			}
 		}
-	}
 	}
 }
 @end

playground/src/containers/TextScreen.js

@@ -14,7 +14,7 @@ class TextScreen extends Component {
         <Text style={styles.h1}>{this.props.text || 'Text Screen'}</Text>
         {this.renderTextFromFunctionInProps()}
         <Text style={styles.footer}>{`this.props.containerId = ${this.props.containerId}`}</Text>
-        <Button title={'setTabBadge'} onPress={() => this.onButtonPress()} />
+        <Button title={'Set Tab Badge'} onPress={() => this.onButtonPress()} />
       </View>
     );
   }
@@ -30,7 +30,7 @@ class TextScreen extends Component {
 
   onButtonPress() {
     Navigation.setOptions(this.props.containerId, {
-      setTabBadge: `${_.floor(Math.random() * 10)}`
+      setTabBadge: `EnCyClOpEdIa`
     });
   }
 }