# -*- coding: utf-8 -*-
# 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
from __future__ import print_function

import io
import os
import sys

import mock

from chromite.lib import cros_test_lib
from chromite.scripts.sysmon import puppet_metrics


assert sys.version_info >= (3, 6), 'This module requires Python 3.6+'


_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('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))