blob: 243ea78a7bd4662c4752aa6f91bd065ab588d607 [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 puppet_metrics."""
# pylint: disable=protected-access
from __future__ import absolute_import
import io
import os
from unittest import mock
from chromite.lib import cros_test_lib
from chromite.scripts.sysmon import puppet_metrics
_SUMMARY = u"""\
---
version:
config: 1499979608
puppet: "3.4.3"
resources:
changed: 7
failed: 0
failed_to_restart: 0
out_of_sync: 7
restarted: 0
scheduled: 0
skipped: 1
total: 218
time:
config_retrieval: 2.862796974
cron: 0.004638468
exec: 11.494792536
file: 0.618018423
file_line: 0.003589435
filebucket: 0.000341392
group: 0.017957332
ini_subsetting: 0.001235189
mount: 0.001416499
package: 4.315027644000001
schedule: 0.001541641
service: 10.242378408
total: 52.958788377
user: 0.001673407
vcsrepo: 23.393381029
last_run: 1499979671
changes:
total: 7
events:
failure: 0
success: 7
total: 7%
"""
class TestPuppetRunSummary(cros_test_lib.TestCase):
"""Tests for _PuppetRunSummary."""
def test_config_version(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.config_version, 1499979608)
def test_puppet_version(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.puppet_version, '3.4.3')
def test_events(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.events, {
'failure': 0,
'success': 7
})
def test_resources(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.resources, {
'changed': 7,
'failed': 0,
'failed_to_restart': 0,
'out_of_sync': 7,
'restarted': 0,
'scheduled': 0,
'skipped': 1,
'other': 203,
})
def test_times(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.times, {
'config_retrieval': 2.862796974,
'cron': 0.004638468,
'exec': 11.494792536,
'file': 0.618018423,
'file_line': 0.003589435,
'filebucket': 0.000341392,
'group': 0.017957332,
'ini_subsetting': 0.001235189,
'mount': 0.001416499,
'other': 0,
'package': 4.315027644000001,
'schedule': 0.001541641,
'service': 10.242378408,
'user': 0.001673407,
'vcsrepo': 23.393381029,
})
def test_last_run_time(self):
summary = puppet_metrics._PuppetRunSummary(io.StringIO(_SUMMARY))
self.assertEqual(summary.last_run_time, 1499979671)
class TestPuppetMetrics(cros_test_lib.TempDirTestCase):
"""Tests for puppet_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)
self.tempfile = os.path.join(self.tempdir, 'last_run_summary.yaml')
def test_collect(self):
with open(self.tempfile, 'w') as f:
f.write(_SUMMARY)
with mock.patch('time.time', return_value=1500000000):
with mock.patch.object(puppet_metrics, 'LAST_RUN_FILE', self.tempfile):
puppet_metrics.collect_puppet_summary()
setter = self.store.set
calls = [
mock.call('puppet/version/config', (), None,
1499979608, enforce_ge=mock.ANY),
mock.call('puppet/version/puppet', (), None,
'3.4.3', enforce_ge=mock.ANY),
mock.call('puppet/events', ('failure',), None,
0, enforce_ge=mock.ANY),
mock.call('puppet/events', ('success',), None,
7, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('scheduled',), None,
0, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('skipped',), None,
1, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('restarted',), None,
0, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('changed',), None,
7, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('failed',), None,
0, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('other',), None,
203, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('failed_to_restart',), None,
0, enforce_ge=mock.ANY),
mock.call('puppet/resources', ('out_of_sync',), None,
7, enforce_ge=mock.ANY),
mock.call('puppet/times', ('vcsrepo',), None,
23.393381029, enforce_ge=mock.ANY),
mock.call('puppet/times', ('exec',), None,
11.494792536, enforce_ge=mock.ANY),
mock.call('puppet/times', ('cron',), None,
0.004638468, enforce_ge=mock.ANY),
mock.call('puppet/times', ('file_line',), None,
0.003589435, enforce_ge=mock.ANY),
mock.call('puppet/times', ('config_retrieval',), None,
2.862796974, enforce_ge=mock.ANY),
mock.call('puppet/times', ('user',), None,
0.001673407, enforce_ge=mock.ANY),
mock.call('puppet/times', ('file',), None,
0.618018423, enforce_ge=mock.ANY),
mock.call('puppet/times', ('group',), None,
0.017957332, enforce_ge=mock.ANY),
mock.call('puppet/times', ('service',), None,
10.242378408, enforce_ge=mock.ANY),
mock.call('puppet/times', ('package',), None,
4.315027644000001, enforce_ge=mock.ANY),
mock.call('puppet/times', ('mount',), None,
0.001416499, enforce_ge=mock.ANY),
mock.call('puppet/times', ('schedule',), None,
0.001541641, enforce_ge=mock.ANY),
mock.call('puppet/times', ('other',), None,
0.0, enforce_ge=mock.ANY),
mock.call('puppet/times', ('ini_subsetting',), None,
0.001235189, enforce_ge=mock.ANY),
mock.call('puppet/times', ('filebucket',), None,
0.000341392, enforce_ge=mock.ANY),
mock.call('puppet/age', (), None,
20329.0, enforce_ge=mock.ANY),
]
setter.assert_has_calls(calls, any_order=True)
self.assertEqual(len(setter.mock_calls), len(calls))