blob: 050bf0eae9af5077f687d730843973f7e491cb19 [file] [log] [blame]
#!/usr/bin/python
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
# pylint: disable-msg=C0111
import unittest
import common
from autotest_lib.site_utils import test_that
class TestThatUnittests(unittest.TestCase):
def test_validate_arguments(self):
# Deferred until validate_arguments allows for lab runs.
pass
def test_parse_arguments(self):
args = test_that.parse_arguments(
['-b', 'some_board', '-i', 'some_image', '--args', 'some_args',
'some_remote', 'test1', 'test2'])
self.assertEqual('some_board', args.board)
self.assertEqual('some_image', args.build)
self.assertEqual('some_args', args.args)
self.assertEqual('some_remote', args.remote)
self.assertEqual(['test1', 'test2'], args.tests)
def test_parse_arguments_internal(self):
args, remote_argv = test_that._parse_arguments_internal(
['-b', 'some_board', '-i', 'some_image', '--args', 'some_args',
'some_remote', 'test1', 'test2'])
self.assertEqual('some_board', args.board)
self.assertEqual('some_image', args.build)
self.assertEqual('some_args', args.args)
self.assertEqual('some_remote', args.remote)
self.assertEqual(['test1', 'test2'], args.tests)
self.assertEqual(remote_argv,
['-b', 'some_board', '-i', 'some_image', '--args',
'some_args', 'some_remote', 'test1', 'test2'])
def test_parse_arguments_internal_with_local_argument(self):
args, remote_argv = test_that._parse_arguments_internal(
['-b', 'some_board', '-i', 'some_image', '-w', 'server:port',
'--args', 'some_args', 'some_remote', 'test1', 'test2'])
self.assertEqual('server:port', args.web)
self.assertEqual('some_board', args.board)
self.assertEqual('some_image', args.build)
self.assertEqual('some_args', args.args)
self.assertEqual('some_remote', args.remote)
self.assertEqual(['test1', 'test2'], args.tests)
self.assertEqual(remote_argv,
['-b', 'some_board', '-i', 'some_image', '--args',
'some_args', 'some_remote', 'test1', 'test2'])
def test_parse_arguments_with_local_argument(self):
args = test_that.parse_arguments(
['-b', 'some_board', '-i', 'some_image', '-w', 'server:port',
'--args', 'some_args', 'some_remote', 'test1', 'test2'])
self.assertEqual('server:port', args.web)
self.assertEqual('some_board', args.board)
self.assertEqual('some_image', args.build)
self.assertEqual('some_args', args.args)
self.assertEqual('some_remote', args.remote)
self.assertEqual(['test1', 'test2'], args.tests)
def test_parse_arguments_lab_run_requires_build(self):
"""Running against :lab: requires certain extra arguments."""
args = test_that.parse_arguments(['-b', 'some_board', ':lab:', 'test1'])
with self.assertRaises(ValueError):
test_that.validate_arguments(args)
def test_parse_arguments_lab_run_disallows_suite_args(self):
"""Running against :lab: requires certain extra arguments."""
args = test_that.parse_arguments([
'-b', 'some_board', '-i', 'some_image', '--args', 'some_args',
':lab:', 'test1'])
with self.assertRaises(ValueError):
test_that.validate_arguments(args)
if __name__ == '__main__':
unittest.main()