workloads/googlemaps: Fix for alternative layouts.

Add additional check for text based directions button as id can be
missing on some devices and allow for skipping the view steps stage for
large screen devices which do not require this step.

2 files changed, 6 insertions, 1 deletions

diff --git a/wa/workloads/googlemaps/com.arm.wa.uiauto.googlemaps.apk b/wa/workloads/googlemaps/com.arm.wa.uiauto.googlemaps.apk
index ccf52ef1..16f92f3b 100644
--- a/wa/workloads/googlemaps/com.arm.wa.uiauto.googlemaps.apk
+++ b/wa/workloads/googlemaps/com.arm.wa.uiauto.googlemaps.apk
diff --git a/wa/workloads/googlemaps/uiauto/app/src/main/java/com/arm/wa/uiauto/UiAutomation.java b/wa/workloads/googlemaps/uiauto/app/src/main/java/com/arm/wa/uiauto/UiAutomation.java
index 2397775e..c1cc5aa7 100644
--- a/wa/workloads/googlemaps/uiauto/app/src/main/java/com/arm/wa/uiauto/UiAutomation.java
+++ b/wa/workloads/googlemaps/uiauto/app/src/main/java/com/arm/wa/uiauto/UiAutomation.java
@@ -126,6 +126,9 @@ public class UiAutomation extends BaseUiAutomation {
 
     public void getDirectionsFromLocation() throws Exception {
         UiObject directions = mDevice.findObject(new UiSelector().resourceId(packageID + "placepage_directions_button"));
+        if (!directions.exists()){
+            directions = mDevice.findObject(new UiSelector().textContains("DIRECTIONS"));
+        }
         directions.clickAndWaitForNewWindow(uiAutoTimeout);
     }
 
@@ -140,7 +143,9 @@ public class UiAutomation extends BaseUiAutomation {
     public void viewRouteSteps() throws Exception {
         UiObject steps = mDevice.findObject(new UiSelector().textContains("STEPS & MORE")
                                                             .className("android.widget.TextView"));
-        steps.clickAndWaitForNewWindow(uiAutoTimeout);
+        if (steps.exists()){
+            steps.clickAndWaitForNewWindow(uiAutoTimeout);
+        }
     }
 
     public void previewRoute() throws Exception {