sysmon: Add osinfo metrics tests
BUG=chromium:736911
TEST=None
Change-Id: I18c733a97088a91aad3dbcc657d2e54ffda053c0
Reviewed-on: https://chromium-review.googlesource.com/570827
Commit-Ready: Allen Li <ayatane@chromium.org>
Tested-by: Allen Li <ayatane@chromium.org>
Reviewed-by: Aviv Keshet <akeshet@chromium.org>
diff --git a/scripts/sysmon/osinfo_metrics_unittest.py b/scripts/sysmon/osinfo_metrics_unittest.py
new file mode 100644
index 0000000..00df495
--- /dev/null
+++ b/scripts/sysmon/osinfo_metrics_unittest.py
@@ -0,0 +1,43 @@
+# 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 __future__ import print_function
+
+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('infra_libs.ts_mon.common.interface.state.store',
+ autospec=True)
+ self.store = patcher.start()
+ self.addCleanup(patcher.stop)
+
+ def test_collect(self):
+ with mock.patch('platform.system', autospec=True) as system, \
+ mock.patch('platform.dist', autospec=True) as dist, \
+ mock.patch('sys.maxsize', 2**64):
+ system.return_value = 'Linux'
+ dist.return_value = ('Ubuntu', '14.04', 'trusty')
+ osinfo_metrics.collect_os_info()
+
+ setter = self.store.set
+ calls = [
+ mock.call('proc/os/name', (), None, 'ubuntu', enforce_ge=mock.ANY),
+ mock.call('proc/os/version', (), None, '14.04', 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))