def test_next_run_no_schedules(self):
# Setup
strategy = mock.Mock()
strategy.retrieve_schedules.return_value = Response(200, {})
next_command = commands.NextRunCommand(self.context, strategy, 'next',
'next')
self.cli.add_command(next_command)
# Test
self.cli.run('next'.split())
# Verify
self.assertTrue('no schedules' in self.recorder.lines[0])
def test_next_run_quiet(self):
# Setup
strategy = mock.Mock()
strategy.retrieve_schedules.return_value = Response(
200, copy.copy(EXAMPLE_SCHEDULE_LIST))
next_command = commands.NextRunCommand(self.context, strategy, 'next',
'next')
self.cli.add_command(next_command)
# Test
self.cli.run('next --quiet'.split())
# Verify
self.assertEqual(1, len(self.prompt.get_write_tags()))
self.assertEqual(TAG_PARAGRAPH, self.prompt.get_write_tags()[0])
self.assertEqual('2012-05-22T00:00:00Z\n', self.recorder.lines[0])
def test_next_run(self):
# Setup
strategy = mock.Mock()
strategy.retrieve_schedules.return_value = Response(
200, copy.copy(EXAMPLE_SCHEDULE_LIST))
next_command = commands.NextRunCommand(self.context, strategy, 'next',
'next')
next_command.create_option('--extra', 'extra')
self.cli.add_command(next_command)
# Test
self.cli.run('next --extra foo'.split())
# Verify
self.assertEqual(1, strategy.retrieve_schedules.call_count)
self.assertEqual(strategy.retrieve_schedules.call_args[0][0]['extra'],
'foo')
self.assertEqual(1, len(self.prompt.get_write_tags()))
self.assertEqual(TAG_PARAGRAPH, self.prompt.get_write_tags()[0])
self.assertTrue('2012-05-22T00:00:00Z' in self.recorder.lines[0])
self.assertTrue('2012-05-15T00:00:00Z/P1W' in self.recorder.lines[0])