Add support for ios_uirobot

build/android/pylib/base/test_run_factory.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
+import os
+import sys
+
+from pylib import constants
 from pylib.gtest import gtest_test_instance
 from pylib.gtest import local_device_gtest_run
 from pylib.local.device import local_device_environment
 from pylib.remote.device import remote_device_environment
 from pylib.remote.device import remote_device_gtest_run
-from pylib.remote.device import remote_device_uirobot_run
-from pylib.uirobot import uirobot_test_instance
+from pylib.remote.device import (
+    remote_device_uirobot_run as android_remote_device_uirobot_run)
+from pylib.uirobot import uirobot_test_instance as android_uirobot_test_instance
 
+sys.path.append(os.path.join(
+    constants.DIR_SOURCE_ROOT, 'build', 'mobile'))
+from bling.pylib.uirobot import (
+    uirobot_test_instance as ios_uirobot_test_instance)
+from bling.pylib.remote.device import (
+    remote_device_uirobot_run as ios_remote_device_uirobot_run)
 
-def CreateTestRun(_args, env, test_instance, error_func):
-  if isinstance(env, local_device_environment.LocalDeviceEnvironment):
-    if isinstance(test_instance, gtest_test_instance.GtestTestInstance):
-      return local_device_gtest_run.LocalDeviceGtestRun(env, test_instance)
+def CreateTestRun(args, env, test_instance, error_func):
+  if args.device_type == 'Android':
+    if isinstance(env, local_device_environment.LocalDeviceEnvironment):
+      if isinstance(test_instance, gtest_test_instance.GtestTestInstance):
+        return local_device_gtest_run.LocalDeviceGtestRun(env, test_instance)
 
-  if isinstance(env, remote_device_environment.RemoteDeviceEnvironment):
-    if isinstance(test_instance, gtest_test_instance.GtestTestInstance):
-      return remote_device_gtest_run.RemoteDeviceGtestRun(env, test_instance)
-    if isinstance(test_instance, uirobot_test_instance.UirobotTestInstance):
-      return remote_device_uirobot_run.RemoteDeviceUirobotRun(
-          env, test_instance)
+    if isinstance(env, remote_device_environment.RemoteDeviceEnvironment):
+      if isinstance(test_instance, gtest_test_instance.GtestTestInstance):
+        return remote_device_gtest_run.RemoteDeviceGtestRun(env, test_instance)
+      if isinstance(test_instance,
+                    android_uirobot_test_instance.UirobotTestInstance):
+        return android_remote_device_uirobot_run.RemoteDeviceUirobotRun(
+            env, test_instance)
+
+  if args.device_type == 'iOS':
+    if isinstance(env, remote_device_environment.RemoteDeviceEnvironment):
+      if isinstance(test_instance,
+                    ios_uirobot_test_instance.UirobotTestInstance):
+        return ios_remote_device_uirobot_run.RemoteDeviceUirobotRun(
+            env, test_instance)
+
 
   # TODO(jbudorick): Add local instrumentation test runs.
   # TODO(rnephew): Add remote_device instrumentation test runs.
 
   error_func('Unable to create test run for %s tests in %s environment'
              % (str(test_instance), str(env)))