scripts/sysmon/osinfo_metrics_unittest.py

# Copyright 2017 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 osinfo_metrics."""

# pylint: disable=protected-access

from __future__ import absolute_import

from unittest import mock

from chromite.lib import cros_test_lib
from chromite.scripts.sysmon import osinfo_metrics


class TestOSInfoMetrics(cros_test_lib.TestCase):
    """Tests for osinfo_metrics."""

    def setUp(self):
        patcher = mock.patch(
            "chromite.third_party.infra_libs.ts_mon.common.interface.state.store",
            autospec=True,
        )
        self.store = patcher.start()
        self.addCleanup(patcher.stop)

    def test_collect(self):
        distro = ("Ubuntu", "14.04", "trusty")
        # This is removed in Python 3.8+.
        try:
            dist = mock.patch(
                "platform.dist", autospec=True, return_value=distro
            )
            dist.start()
        except AttributeError:
            distro = ("", "", "")
            dist = None

        with mock.patch("platform.system", autospec=True) as system, mock.patch(
            "sys.maxsize", 2**64
        ):
            system.return_value = "Linux"
            if dist is not None:
                dist.return_value = distro
            osinfo_metrics.collect_os_info()

        if dist is not None:
            dist.stop()

        setter = self.store.set
        print(setter.mock_calls)
        calls = [
            mock.call(
                "proc/os/name", (), None, distro[0].lower(), enforce_ge=mock.ANY
            ),
            mock.call(
                "proc/os/version", (), None, distro[1], enforce_ge=mock.ANY
            ),
            mock.call("proc/os/arch", (), None, "x86_64", enforce_ge=mock.ANY),
            mock.call("proc/python/arch", (), None, "64", enforce_ge=mock.ANY),
        ]
        setter.assert_has_calls(calls)
        self.assertEqual(len(setter.mock_calls), len(calls))