Add support for ios_uirobot

build/android/pylib/base/test_instance_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.utils import isolator
-from pylib.uirobot import uirobot_test_instance
+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)
 
 def CreateTestInstance(args, error_func):
 
-  if args.command == 'gtest':
-    return gtest_test_instance.GtestTestInstance(
-        args, isolator.Isolator(constants.ISOLATE_DEPS_DIR), error_func)
-  if args.command == 'uirobot':
-    return uirobot_test_instance.UirobotTestInstance(args)
-  # TODO(jbudorick) Add instrumentation test instance.
+  if args.device_type == 'Android':
+    if args.command == 'gtest':
+      return gtest_test_instance.GtestTestInstance(
+          args, isolator.Isolator(constants.ISOLATE_DEPS_DIR), error_func)
+    if args.command == 'uirobot':
+      return android_uirobot_test_instance.UirobotTestInstance(args)
+    # TODO(jbudorick) Add instrumentation test instance.
+
+  if args.device_type == 'iOS':
+    if args.command == 'uirobot':
+      return ios_uirobot_test_instance.UirobotTestInstance(args)
 
   error_func('Unable to create %s test instance.' % args.command)