blob: 7d1539602b0ef366ed19f019be8af6e5e9d0a335 [file] [log] [blame]
Tudor Brindus3e03eba2018-07-18 11:27:13 -07001# Copyright 2018 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5"""Unit tests for cros_generate_update_payload."""
6
Tudor Brindus3e03eba2018-07-18 11:27:13 -07007from chromite.lib import cros_test_lib
Amin Hassani6bc73a12018-11-29 21:07:12 -08008from chromite.lib import partial_mock
Amin Hassani6bc73a12018-11-29 21:07:12 -08009from chromite.lib.paygen import paygen_payload_lib
Tudor Brindus3e03eba2018-07-18 11:27:13 -070010from chromite.scripts import cros_generate_update_payload
11
Mike Frysinger807d8282022-04-28 22:45:17 -040012
Greg Edelstona4c9b3b2020-01-07 17:51:13 -070013pytestmark = cros_test_lib.pytestmark_inside_only
14
Amin Hassani6bc73a12018-11-29 21:07:12 -080015
16class CrOSGenerateUpdatePayloadTest(cros_test_lib.MockTestCase):
Alex Klein1699fab2022-09-08 08:46:06 -060017 """Test correct arguments passed to delta_generator."""
Amin Hassani6bc73a12018-11-29 21:07:12 -080018
Alex Klein1699fab2022-09-08 08:46:06 -060019 def testGenerateUpdatePayload(self):
20 """Test correct arguments propagated to delta_generator call."""
Tudor Brindus3e03eba2018-07-18 11:27:13 -070021
Alex Klein1699fab2022-09-08 08:46:06 -060022 paygen_mock = self.PatchObject(
23 paygen_payload_lib, "GenerateUpdatePayload"
24 )
Amin Hassani6bc73a12018-11-29 21:07:12 -080025
Alex Klein1699fab2022-09-08 08:46:06 -060026 cros_generate_update_payload.main(
27 [
28 "--tgt-image",
29 "foo-tgt-image",
30 "--src-image",
31 "foo-src-image",
32 "--output",
33 "foo-output",
34 "--check",
35 "--minios",
36 "--private-key",
37 "foo-private-key",
38 "--work-dir",
39 "foo-work-dir",
40 ]
41 )
Tudor Brindus3e03eba2018-07-18 11:27:13 -070042
Alex Klein1699fab2022-09-08 08:46:06 -060043 paygen_mock.assert_called_once_with(
44 partial_mock.HasString("foo-tgt-image"),
45 partial_mock.HasString("foo-output"),
46 src_image=partial_mock.HasString("foo-src-image"),
47 work_dir=partial_mock.HasString("foo-work-dir"),
48 private_key=partial_mock.HasString("foo-private-key"),
49 check=True,
50 minios=True,
51 )