@@ -34,15 +34,19 @@ public void setUp(String config_file, String environment) throws Exception {
3434 Iterator it = envCapabilities .entrySet ().iterator ();
3535 while (it .hasNext ()) {
3636 Map .Entry pair = (Map .Entry ) it .next ();
37- capabilities .setCapability (pair .getKey ().toString (), pair .getValue (). toString () );
37+ capabilities .setCapability (pair .getKey ().toString (), pair .getValue ());
3838 }
3939
4040 Map <String , String > commonCapabilities = (Map <String , String >) config .get ("capabilities" );
4141 it = commonCapabilities .entrySet ().iterator ();
4242 while (it .hasNext ()) {
4343 Map .Entry pair = (Map .Entry ) it .next ();
4444 if (capabilities .getCapability (pair .getKey ().toString ()) == null ) {
45- capabilities .setCapability (pair .getKey ().toString (), pair .getValue ().toString ());
45+ capabilities .setCapability (pair .getKey ().toString (), pair .getValue ());
46+ } else if (pair .getKey ().toString ().equalsIgnoreCase ("bstack:options" )) {
47+ HashMap bstackOptionsMap = (HashMap ) pair .getValue ();
48+ bstackOptionsMap .putAll ((HashMap ) capabilities .getCapability ("bstack:options" ));
49+ capabilities .setCapability (pair .getKey ().toString (), bstackOptionsMap );
4650 }
4751 }
4852
@@ -56,12 +60,14 @@ public void setUp(String config_file, String environment) throws Exception {
5660 accessKey = (String ) config .get ("key" );
5761 }
5862
59- if (capabilities .getCapability ("browserstack.local" ) != null
60- && capabilities .getCapability ("browserstack.local" ) == "true" ) {
61- l = new Local ();
62- Map <String , String > options = new HashMap <String , String >();
63- options .put ("key" , accessKey );
64- l .start (options );
63+ if (capabilities .getCapability ("bstack:options" ) != null ) {
64+ HashMap bstackOptionsMap = (HashMap ) capabilities .getCapability ("bstack:options" );
65+ if (bstackOptionsMap .get ("local" ).toString ().equalsIgnoreCase ("true" )) {
66+ l = new Local ();
67+ Map <String , String > options = new HashMap <String , String >();
68+ options .put ("key" , accessKey );
69+ l .start (options );
70+ }
6571 }
6672
6773 driver = new RemoteWebDriver (
0 commit comments