| #!/usr/bin/python2 |
| # 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. |
| |
| import StringIO |
| import mox |
| import unittest |
| import urllib2 |
| |
| import mock |
| |
| import common |
| from autotest_lib.client.common_lib.cros import retry |
| from autotest_lib.server import site_utils |
| |
| |
| # Mock retry.retry used in test_push before importing test_push. |
| retry.retry = mock.create_autospec(retry.retry, return_value=lambda func: func) |
| from autotest_lib.site_utils import test_push |
| |
| class TestPushUnittests(mox.MoxTestBase): |
| """Unittest for test_push script.""" |
| |
| _ARGV = [ |
| 'command', |
| '--build', 'stumpy-release/R36-5881-0.0', |
| '--shard_build', 'quawks-release/R36-5881-0.0' |
| ] |
| |
| def setUp(self): |
| """Initialize the unittest.""" |
| super(TestPushUnittests, self).setUp() |
| # Overwrite expected test results. |
| test_push.EXPECTED_TEST_RESULTS = { |
| '^SERVER_JOB$': ['GOOD'], |
| '.*control.dependency$': ['TEST_NA'], |
| '.*dummy_Fail.RetryFail$': ['FAIL', 'FAIL'], |
| } |
| test_push.TKO = None |
| |
| |
| def stub_out_methods(self, test_views): |
| """Stub out methods in test_push module with given test results. |
| |
| @param test_views: Desired test result views. |
| |
| """ |
| self.mox.UnsetStubs() |
| response = StringIO.StringIO('some_value') |
| self.mox.StubOutWithMock(urllib2, 'urlopen') |
| urllib2.urlopen(mox.IgnoreArg()).AndReturn(response) |
| |
| self.mox.StubOutWithMock(test_push, 'check_dut_image') |
| test_push.check_dut_image(mox.IgnoreArg(), mox.IgnoreArg()).AndReturn( |
| None) |
| |
| self.mox.StubOutWithMock(test_push, 'do_run_suite') |
| test_push.do_run_suite( |
| test_push.PUSH_TO_PROD_SUITE, mox.IgnoreArg(), mox.IgnoreArg(), |
| mox.IgnoreArg()).AndReturn((1)) |
| |
| self.mox.StubOutWithMock(site_utils, 'get_test_views_from_tko') |
| site_utils.get_test_views_from_tko(1, None).AndReturn(test_views) |
| |
| |
| def test_suite_success(self): |
| """Test test_suite method with matching results.""" |
| test_views = {'SERVER_JOB': ['GOOD'], |
| 'dummy_fail/control.dependency': ['TEST_NA'], |
| 'dummy_Fail.RetryFail': ['FAIL', 'FAIL'], |
| } |
| |
| self.stub_out_methods(test_views) |
| self.mox.ReplayAll() |
| test_push.test_suite(test_push.PUSH_TO_PROD_SUITE, test_views, |
| arguments=test_push.parse_arguments(self._ARGV)) |
| self.mox.VerifyAll() |
| |
| |
| def test_suite_fail_with_missing_test(self): |
| """Test test_suite method that should fail with missing test.""" |
| test_views = {'SERVER_JOB': ['GOOD'], |
| 'dummy_fail/control.dependency': ['TEST_NA'], |
| } |
| |
| self.stub_out_methods(test_views) |
| self.mox.ReplayAll() |
| test_push.test_suite(test_push.PUSH_TO_PROD_SUITE, test_views, |
| arguments=test_push.parse_arguments(self._ARGV)) |
| self.mox.VerifyAll() |
| |
| |
| def test_suite_fail_with_unexpected_test_results(self): |
| """Test test_suite method that should fail with unexpected test results. |
| """ |
| test_views = {'SERVER_JOB': ['FAIL'], |
| 'dummy_fail/control.dependency': ['TEST_NA'], |
| 'dummy_Fail.RetryFail': ['FAIL', 'FAIL'], |
| } |
| |
| self.stub_out_methods(test_views) |
| self.mox.ReplayAll() |
| test_push.test_suite(test_push.PUSH_TO_PROD_SUITE, test_views, |
| arguments=test_push.parse_arguments(self._ARGV)) |
| self.mox.VerifyAll() |
| |
| |
| def test_suite_fail_with_extra_test(self): |
| """Test test_suite method that should fail with extra test.""" |
| test_views = {'SERVER_JOB': ['GOOD'], |
| 'dummy_fail/control.dependency': ['TEST_NA'], |
| 'dummy_Fail.RetryFail': ['FAIL', 'FAIL'], |
| 'dummy_Fail.ExtraTest': ['GOOD'], |
| } |
| |
| self.stub_out_methods(test_views) |
| self.mox.ReplayAll() |
| test_push.test_suite(test_push.PUSH_TO_PROD_SUITE, test_views, |
| arguments=test_push.parse_arguments(self._ARGV)) |
| self.mox.VerifyAll() |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |