def test_file_not_found(self): self.mock_lastrunfile.return_value = os.path.join(self.tempdir, 'missing') puppet_metrics.get_puppet_summary() self.assertIs(None, puppet_metrics.config_version.get()) self.assertIs(None, puppet_metrics.puppet_version.get())
def test_invalid_file(self): self.fh.write('"') self.fh.close() puppet_metrics.get_puppet_summary() self.assertIs(None, puppet_metrics.config_version.get()) self.assertIs(None, puppet_metrics.puppet_version.get())
def test_file_contains_array(self): self.fh.write("""\ - one - two """) self.fh.close() puppet_metrics.get_puppet_summary() self.assertIs(None, puppet_metrics.config_version.get()) self.assertIs(None, puppet_metrics.puppet_version.get())
def single_iteration(): try: system_metrics.get_cpu_info() system_metrics.get_disk_info() system_metrics.get_mem_info() system_metrics.get_net_info() system_metrics.get_proc_info() puppet_metrics.get_puppet_summary() finally: ts_mon.flush() return True
def test_summary_missing_values(self): self.fh.write("""\ --- version: config: puppet: """) self.fh.close() mock_time = mock.create_autospec(time.time, spec_set=True) mock_time.return_value = 1440132466 + 123 puppet_metrics.get_puppet_summary(time_fn=mock_time) self.assertEqual(None, puppet_metrics.config_version.get()) self.assertEqual(None, puppet_metrics.puppet_version.get())
def test_summary_missing_sections(self): self.fh.write("""\ --- something_else: foo: 123 bar: 456 """) self.fh.close() mock_time = mock.Mock() mock_time.return_value = 1440132466 + 123 puppet_metrics.get_puppet_summary(time_fn=mock_time) self.assertEqual(None, puppet_metrics.config_version.get()) self.assertEqual(None, puppet_metrics.puppet_version.get())
def test_summary(self): self.fh.write("""\ --- version: config: 1440131220 puppet: "3.6.2" resources: changed: 1 failed: 2 failed_to_restart: 3 out_of_sync: 4 restarted: 5 scheduled: 6 skipped: 7 total: 51 time: anchor: 0.01 apt_key: 0.02 config_retrieval: 0.03 exec: 0.04 file: 0.05 filebucket: 0.06 package: 0.07 schedule: 0.08 service: 0.08 total: 0.09 last_run: 1440132466 changes: total: 4 events: failure: 1 success: 2 total: 3 """) self.fh.close() mock_time = mock.create_autospec(time.time, spec_set=True) mock_time.return_value = 1440132466 + 123 puppet_metrics.get_puppet_summary(time_fn=mock_time) self.assertEqual(1440131220, puppet_metrics.config_version.get()) self.assertEqual('3.6.2', puppet_metrics.puppet_version.get()) self.assertEqual(1, puppet_metrics.events.get({'result': 'failure'})) self.assertEqual(2, puppet_metrics.events.get({'result': 'success'})) self.assertEqual(None, puppet_metrics.events.get({'result': 'total'})) self.assertEqual(1, puppet_metrics.resources.get({'action': 'changed'})) self.assertEqual(2, puppet_metrics.resources.get({'action': 'failed'})) self.assertEqual(3, puppet_metrics.resources.get({ 'action': 'failed_to_restart'})) self.assertEqual(4, puppet_metrics.resources.get({'action': 'out_of_sync'})) self.assertEqual(5, puppet_metrics.resources.get({'action': 'restarted'})) self.assertEqual(6, puppet_metrics.resources.get({'action': 'scheduled'})) self.assertEqual(7, puppet_metrics.resources.get({'action': 'skipped'})) self.assertEqual(51, puppet_metrics.resources.get({'action': 'total'})) self.assertEqual(0.01, puppet_metrics.times.get({'step': 'anchor'})) self.assertEqual(0.02, puppet_metrics.times.get({'step': 'apt_key'})) self.assertEqual(0.03, puppet_metrics.times.get({ 'step': 'config_retrieval'})) self.assertEqual(0.04, puppet_metrics.times.get({'step': 'exec'})) self.assertEqual(0.05, puppet_metrics.times.get({'step': 'file'})) self.assertEqual(0.06, puppet_metrics.times.get({'step': 'filebucket'})) self.assertEqual(0.07, puppet_metrics.times.get({'step': 'package'})) self.assertEqual(0.08, puppet_metrics.times.get({'step': 'schedule'})) self.assertEqual(0.08, puppet_metrics.times.get({'step': 'service'})) self.assertEqual(None, puppet_metrics.times.get({'step': 'total'})) self.assertEqual(123, puppet_metrics.age.get())
def test_empty_file(self): puppet_metrics.get_puppet_summary() self.assertIs(None, puppet_metrics.config_version.get()) self.assertIs(None, puppet_metrics.puppet_version.get())