Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(362)

Unified Diff: infra/services/sysmon/test/system_metrics_test.py

Issue 2106953006: adding os metrics (os name and version) being collected every hour (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: adding os metrics (os name and version) being collected every hour Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: infra/services/sysmon/test/system_metrics_test.py
diff --git a/infra/services/sysmon/test/system_metrics_test.py b/infra/services/sysmon/test/system_metrics_test.py
index ccce512b5d1e9b488f15947116c3d63671385699..007eaa06014912600599a726706cf47cd5d11952 100644
--- a/infra/services/sysmon/test/system_metrics_test.py
+++ b/infra/services/sysmon/test/system_metrics_test.py
@@ -157,3 +157,130 @@ class SystemMetricsTest(unittest.TestCase):
system_metrics.get_unix_time()
self.assertTrue(
1464000000000 < system_metrics.unix_time.get() < 9464000000000)
+
+ @mock.patch('platform.system')
+ @mock.patch('platform.mac_ver')
+ @mock.patch('platform.machine')
+ @mock.patch('sys')
+ def test_os_info_mac_10115_64_64(self,
+ platform_system_mock,
+ platform_mac_ver_mock,
+ platform_machine_mock,
+ sys_mock):
+ platform_system_mock.return_value = 'Darwin'
+ platform_mac_ver_mock.return_value = ('10.11.5', ('', '', ''), 'x86_64')
+ platform_machine_mock.return_value = 'x86_64'
+ sys_mock.maxsize = 9223372036854775807
+
+ system_metrics.get_os_info()
+
+ self.AssertTrue(platform_system_mock.called)
dsansome 2016/07/01 04:30:56 Does this work? The methods are named with a lowe
dsansome 2016/07/01 04:30:56 Also platform_system_mock.assert_called_once_with(
chrishall 2016/07/01 04:53:57 very nice!
chrishall 2016/07/01 04:53:57 Done.
chrishall 2016/07/01 04:53:57 Done.
chrishall 2016/07/01 04:53:57 Done.
+ self.AssertTrue(platform_mac_ver_mock.called)
+ self.AssertTrue(platform_machine_mock.called)
+
+ self.AssertEqual('mac', system_metrics.os_name.get())
+ self.AssertEqual('10.11.5', system_metrics.os_version.get())
+ self.AssertEqual('x86_64', system_metrics.os_arch.get())
+ self.AssertEqual('64', system_metrics.python_arch.get())
+
+ @mock.patch('platform.system')
+ @mock.patch('platform.release')
+ @mock.patch('platform.machine')
+ @mock.patch('sys')
+ def test_os_info_windows_7_32_32(self,
+ platform_system_mock,
+ platform_release_mock,
+ platform_machine_mock,
+ sys_mock):
+ platform_system_mock.return_value = 'Windows'
+ platform_release_mock.return_value = '7'
+ platform_machine_mock.return_value = 'x86'
+ sys_mock.maxsize = 2147483647
+
+ system_metrics.get_os_info()
+
+ self.AssertTrue(platform_system_mock.called)
+ self.AssertTrue(platform_release_mock.called)
+ self.AssertTrue(platform_machine_mock.called)
+
+ self.AssertEqual('windows', system_metrics.os_name.get())
+ self.AssertEqual('7', system_metrics.os_version.get())
+ self.AssertEqual('x86', system_metrics.os_arch.get())
+ self.AssertEqual('32', system_metrics.python_arch.get())
+
+ @mock.patch('platform.system')
+ @mock.patch('platform.dist')
+ @mock.patch('platform.machine')
+ @mock.patch('sys')
+ def test_os_info_ubuntu_1404_32_32(self,
+ platform_system_mock,
+ platform_dist_mock,
+ platform_machine_mock,
+ sys_mock):
+ platform_system_mock.return_value = 'Linux'
+ platform_dist_mock.return_value = ('Ubuntu', '14.04', 'trusty')
+ platform_machine_mock.return_value = 'i686'
+ sys_mock.maxsize_mock = 2147483647
+
+ system_metrics.get_os_info()
+
+ self.AssertTrue(platform_system_mock.called)
+ self.AssertTrue(platform_dist_mock.called)
+ self.AssertTrue(platform_machine_mock.called)
+
+ self.AssertEqual('ubuntu', system_metrics.os_name.get())
+ self.AssertEqual('14.04', system_metrics.os_version.get())
+ self.AssertEqual('i686', system_metrics.os_arch.get())
+ self.AssertEqual('32', system_metrics.python_arch.get())
+
+ @mock.patch('platform.system')
+ @mock.patch('platform.dist')
+ @mock.patch('platform.machine')
+ @mock.patch('sys')
+ def test_os_info_ubuntu_1404_64_32(self,
+ platform_system_mock,
+ platform_dist_mock,
+ platform_machine_mock,
+ sys_mock):
+ platform_system_mock.return_value = 'Linux'
+ platform_dist_mock.return_value = ('Ubuntu', '14.04', 'trusty')
+ platform_machine_mock.return_value = 'x86_64'
+ sys_mock.maxsize = 2147483647
+
+ system_metrics.get_os_info()
+
+ self.AssertTrue(platform_system_mock.called)
+ self.AssertTrue(platform_dist_mock.called)
+ self.AssertTrue(platform_machine_mock.called)
+
+ self.AssertEqual('ubuntu', system_metrics.os_name.get())
+ self.AssertEqual('14.04', system_metrics.os_version.get())
+ self.AssertEqual('x86_64', system_metrics.os_arch.get())
+ self.AssertEqual('32', system_metrics.python_arch.get())
+
+ @mock.patch('platform.system')
+ @mock.patch('platform.dist')
+ @mock.patch('platform.machine')
+ @mock.patch('sys')
+ def test_os_info_ubuntu_1404_64_64(self,
+ platform_system_mock,
+ platform_dist_mock,
+ platform_machine_mock,
+ sys_mock):
+ platform_system_mock.return_value = 'Linux'
+ platform_dist_mock.return_value = ('Ubuntu', '14.04', 'trusty')
+ platform_machine_mock.return_value = 'x86_64'
+ sys_mock.maxsize = 9223372036854775807
+
+ system_metrics.get_os_info()
+
+ self.AssertTrue(platform_system_mock.called)
+ self.AssertTrue(platform_dist_mock.called)
+ self.AssertTrue(platform_machine_mock.called)
+
+ self.AssertEqual('ubuntu', system_metrics.os_name.get())
+ self.AssertEqual('14.04', system_metrics.os_version.get())
+ self.AssertEqual('x86_64', system_metrics.os_arch.get())
+ self.AssertEqual('64', system_metrics.python_arch.get())
+
+
« infra/services/sysmon/system_metrics.py ('K') | « infra/services/sysmon/system_metrics.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698