| # -*- coding: utf-8 -*- |
| # Copyright 2018 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. |
| |
| """Unit tests for cros_update.""" |
| |
| from __future__ import print_function |
| |
| import mock |
| |
| from chromite.lib import auto_updater |
| from chromite.lib import auto_updater_transfer |
| from chromite.lib import cros_test_lib |
| from chromite.scripts import cros_update |
| |
| |
| # pylint: disable=protected-access |
| |
| class CrosUpdateTest(cros_test_lib.RunCommandTestCase): |
| """Tests cros_update functions.""" |
| |
| def setUp(self): |
| """Setup an instance of CrOSUpdateTrigger.""" |
| self._cros_update_trigger = cros_update.CrOSUpdateTrigger( |
| 'foo-host-name', 'foo-build-name', 'foo-static-dir', |
| static_url='foo-static', devserver_url='foo-devserver-url') |
| self._cros_updater = auto_updater.ChromiumOSUpdater( |
| mock.MagicMock(work_dir='foo-dir'), 'foo-build-name', 'foo-payload-dir', |
| transfer_class=auto_updater_transfer.LocalTransfer) |