robot_classes.kt

class ScreenRobot {
   fun enterTextIntoView(@IdRes viewId: Int, text: String): T {
       onView(withId(viewId)).perform(typeText(text))
       closeKeyboard()
       return this as T
   }
}
class LoginRobot: ScreenRobot() {
   fun inputPassword(pass: String): LoginRobot {
       return enterTextIntoView(LOGIN_PASSWORD, pass)
   }
   companion object {
       private val LOGIN_PASSWORD = R.id.login_password
   }
}