A ViewAction (espresso) that changes the orientation of the screen

OrientationChangeAction.java

/**
   * An Espresso ViewAction that changes the orientation of the screen
   */
  public static class OrientationChangeAction implements ViewAction {
    private final int orientation;

    private OrientationChangeAction(int orientation) {
      this.orientation = orientation;
    }

    @Override public Matcher<View> getConstraints() {
      return isRoot();
    }

    @Override public String getDescription() {
      return "change orientation to " + orientation;
    }

    @Override public void perform(UiController uiController, View view) {
      uiController.loopMainThreadUntilIdle();
      final Activity activity = (Activity) view.getContext();
      activity.setRequestedOrientation(orientation);

      Collection<Activity> resumedActivities =
          ActivityLifecycleMonitorRegistry.getInstance().getActivitiesInStage(Stage.RESUMED);
      if (resumedActivities.isEmpty()) {
        throw new RuntimeException("Could not change orientation");
      }
    }

    public static ViewAction orientationLandscape() {
      return new OrientationChangeAction(ActivityInfo.SCREEN_ORIENTATION_LANDSCAPE);
    }

    public static ViewAction orientationPortrait() {
      return new OrientationChangeAction(ActivityInfo.SCREEN_ORIENTATION_PORTRAIT);
    }
  }