# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.


from pylib.gtest import gtest_test_instance
from pylib.hostside import hostside_test_instance
from pylib.instrumentation import instrumentation_test_instance
from pylib.junit import junit_test_instance
from pylib.monkey import monkey_test_instance
from pylib.utils import device_dependencies


def CreateTestInstance(args, error_func):

  if args.command == 'gtest':
    return gtest_test_instance.GtestTestInstance(
        args, device_dependencies.GetDataDependencies, error_func)
  if args.command == 'hostside':
    return hostside_test_instance.HostsideTestInstance(
        args, error_func)
  if args.command == 'instrumentation':
    return instrumentation_test_instance.InstrumentationTestInstance(
        args, device_dependencies.GetDataDependencies, error_func)
  if args.command == 'junit':
    return junit_test_instance.JunitTestInstance(args, error_func)
  if args.command == 'monkey':
    return monkey_test_instance.MonkeyTestInstance(args, error_func)

  error_func('Unable to create %s test instance.' % args.command)
  raise RuntimeError('error_func must call exit inside.')