scripts/cros_generate_update_payload_unittest.py

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

"""Unit tests for cros_generate_update_payload."""

from chromite.lib import chroot_lib
from chromite.lib import cros_test_lib
from chromite.lib import partial_mock
from chromite.lib.paygen import paygen_payload_lib
from chromite.scripts import cros_generate_update_payload


pytestmark = cros_test_lib.pytestmark_inside_only


class CrOSGenerateUpdatePayloadTest(cros_test_lib.MockTestCase):
    """Test correct arguments passed to delta_generator."""

    def testGenerateUpdatePayload(self) -> None:
        """Test correct arguments propagated to delta_generator call."""

        paygen_mock = self.PatchObject(
            paygen_payload_lib, "GenerateUpdatePayload"
        )

        cros_generate_update_payload.main(
            [
                "--tgt-image",
                "foo-tgt-image",
                "--src-image",
                "foo-src-image",
                "--output",
                "foo-output",
                "--check",
                "--minios",
                "--private-key",
                "foo-private-key",
                "--work-dir",
                "foo-work-dir",
            ]
        )

        paygen_mock.assert_called_once_with(
            chroot_lib.Chroot(),
            partial_mock.HasString("foo-tgt-image"),
            partial_mock.HasString("foo-output"),
            src_image=partial_mock.HasString("foo-src-image"),
            work_dir=partial_mock.HasString("foo-work-dir"),
            private_key=partial_mock.HasString("foo-private-key"),
            check=True,
            minios=True,
        )