blob: 66913b616f9016b5842cc8a8c8ddc7a6c4cccc5f [file] [log] [blame]
# Copyright 2017 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 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))