def testcover_main_list(self, mock_rrd_list, mock_parse_args): fake_tstamp = os.path.join(self._testdata_dir, 'fake.tstamp') mlab_export.EXPORT_LOCKFILE = fake_tstamp mock_options = disable_show_options(mock.Mock()) mock_options.show_nagios = True # Enable one show option. mock_parse_args.return_value = mock_options mlab_export.main() self.assertTrue(mock_parse_args.called) self.assertTrue(mock_rrd_list.called)
def testcover_main_export(self, mock_rrd_export, mock_parse_args): fake_tstamp = os.path.join(self._testdata_dir, 'fake.tstamp') mlab_export.LAST_EXPORT_FILENAME = fake_tstamp mock_options = disable_show_options(mock.Mock()) mock_options.update = True mock_options.rrddir_prefix = mlab_export.RRD_PREFIX mock_options.ts_start = FAKE_TIMESTAMP - 60 * 60 mock_options.ts_end = FAKE_TIMESTAMP mock_parse_args.return_value = mock_options mlab_export.main() self.assertTrue(mock_parse_args.called) self.assertTrue(mock_rrd_export.called)