sysmon: Add sysmon.net_metrics tests
BUG=chromium:736911
TEST=None
Change-Id: I57aaf1fd6e22bcd86c91c0bd6045d9b8ff864a95
Reviewed-on: https://chromium-review.googlesource.com/570506
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/net_metrics_unittest.py b/scripts/sysmon/net_metrics_unittest.py
index 8752c98..1b422f8 100644
--- a/scripts/sysmon/net_metrics_unittest.py
+++ b/scripts/sysmon/net_metrics_unittest.py
@@ -2,19 +2,87 @@
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
-"""Unit tests for git_metrics."""
+"""Unit tests for net_metrics."""
+
+# pylint: disable=protected-access
from __future__ import absolute_import
from __future__ import print_function
+import mock
+
+import psutil
+
from chromite.lib import cros_test_lib
from chromite.scripts.sysmon import net_metrics
-class TestNetMetrics(cros_test_lib.TestCase):
- """Tests for network metrics."""
+snetio = psutil._common.snetio
+snicstats = psutil._common.snicstats
- def test_collect_net_info(self):
- """Tests that collecting the git hash doesn't have type conflicts."""
- # This has the side-effect of checking the types are correct.
- net_metrics.collect_net_info()
+
+class TestNetMetrics(cros_test_lib.TestCase):
+ """Tests for net_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('psutil.net_io_counters', autospec=True) \
+ as net_io_counters, \
+ mock.patch('psutil.net_if_stats', autospec=True) as net_if_stats:
+ net_io_counters.return_value = {
+ 'lo': snetio(
+ bytes_sent=17247495681, bytes_recv=172474956,
+ packets_sent=109564455, packets_recv=1095644,
+ errin=10, errout=1, dropin=20, dropout=2),
+ }
+ net_if_stats.return_value = {
+ 'lo': snicstats(isup=True, duplex=0, speed=0, mtu=65536),
+ }
+ net_metrics.collect_net_info()
+
+ setter = self.store.set
+ calls = [
+ mock.call('dev/net/bytes/up', ('lo',), None,
+ 17247495681, enforce_ge=mock.ANY),
+ mock.call('dev/net/bytes/down', ('lo',), None,
+ 172474956, enforce_ge=mock.ANY),
+ mock.call('dev/net/err/up', ('lo',), None,
+ 1, enforce_ge=mock.ANY),
+ mock.call('dev/net/err/down', ('lo',), None,
+ 10, enforce_ge=mock.ANY),
+ mock.call('dev/net/drop/up', ('lo',), None,
+ 2, enforce_ge=mock.ANY),
+ mock.call('dev/net/drop/down', ('lo',), None,
+ 20, enforce_ge=mock.ANY),
+ mock.call('dev/net/bytes', ('up', 'lo'), None,
+ 17247495681, enforce_ge=mock.ANY),
+ mock.call('dev/net/bytes', ('down', 'lo'), None,
+ 172474956, enforce_ge=mock.ANY),
+ mock.call('dev/net/packets', ('up', 'lo'), None,
+ 109564455, enforce_ge=mock.ANY),
+ mock.call('dev/net/packets', ('down', 'lo'), None,
+ 1095644, enforce_ge=mock.ANY),
+ mock.call('dev/net/errors', ('up', 'lo'), None,
+ 1, enforce_ge=mock.ANY),
+ mock.call('dev/net/errors', ('down', 'lo'), None,
+ 10, enforce_ge=mock.ANY),
+ mock.call('dev/net/dropped', ('up', 'lo'), None,
+ 2, enforce_ge=mock.ANY),
+ mock.call('dev/net/dropped', ('down', 'lo'), None,
+ 20, enforce_ge=mock.ANY),
+ mock.call('dev/net/isup', ('lo',), None,
+ True, enforce_ge=mock.ANY),
+ mock.call('dev/net/duplex', ('lo',), None,
+ 0, enforce_ge=mock.ANY),
+ mock.call('dev/net/speed', ('lo',), None,
+ 0, enforce_ge=mock.ANY),
+ mock.call('dev/net/mtu', ('lo',), None,
+ 65536, enforce_ge=mock.ANY),
+ ]
+ setter.assert_has_calls(calls)
+ self.assertEqual(len(setter.mock_calls), len(calls))