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