class TestBaseJobSummary(object):
def setup(self):
self.cls = Job('jname', 'schedname')
def test_summary_one_line(self):
self.cls._output = 'foo'
assert self.cls.summary() == 'foo'
def test_summary(self):
self.cls._output = "foo\n\n \nbar\n"
assert self.cls.summary() == 'bar'
def test_none(self):
self.cls._output = None
assert self.cls.summary() == ''
def test_short(self):
self.cls._output = "\n \n"
assert self.cls.summary() == ''
def test_regex_one(self):
self.cls._summary_regex = '^f.*$'
self.cls._output = "foo\n"
assert self.cls.summary() == 'foo'
def test_regex_multiple(self):
self.cls._summary_regex = '^f.*$'
self.cls._output = "foo\nfie\nfoe\nbar\nfit\nbaz"
assert self.cls.summary() == 'fit'
def test_regex_no_match(self):
self.cls._summary_regex = '^f.*$'
self.cls._output = "\nbar\nbaz\nblam"
assert self.cls.summary() == 'blam'
def test_init_regex(self):
cls = Job('jname', 'schedname', summary_regex='foobar')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex == 'foobar'
assert cls._skip_reason is None
assert cls._cron_expression is None
def test_init_cronex_skip(self):
m_cronex = Mock()
m_cronex.check_trigger.return_value = False
with patch('%s.CronExpression' % pbm) as mock_cronex:
mock_cronex.return_value = m_cronex
cls = Job('jname', 'schedname', cron_expression='foo')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex is None
assert cls._cron_expression == m_cronex
assert mock_cronex.mock_calls == [
call('foo'),
call().check_trigger((2017, 11, 26, 13, 45))
]
assert m_cronex.mock_calls == [
call.check_trigger((2017, 11, 26, 13, 45))
]
assert cls._skip_reason == 'cronex: "foo"'
def get_job_classes():
jobclasses = {}
for cls in Job.__subclasses__():
jobclasses[cls.__name__] = cls
return jobclasses
def setup(self):
self.cls = Job('jname', 'schedname')
class TestBaseJob(object):
def setup(self):
self.cls = Job('jname', 'schedname')
def test_init(self):
cls = Job('jname', 'schedname')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex is None
assert cls._skip_reason is None
assert cls._cron_expression is None
def test_init_regex(self):
cls = Job('jname', 'schedname', summary_regex='foobar')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex == 'foobar'
assert cls._skip_reason is None
assert cls._cron_expression is None
@freeze_time('2017-11-26 13:45:52')
def test_init_cronex_skip(self):
m_cronex = Mock()
m_cronex.check_trigger.return_value = False
with patch('%s.CronExpression' % pbm) as mock_cronex:
mock_cronex.return_value = m_cronex
cls = Job('jname', 'schedname', cron_expression='foo')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex is None
assert cls._cron_expression == m_cronex
assert mock_cronex.mock_calls == [
call('foo'),
call().check_trigger((2017, 11, 26, 13, 45))
]
assert m_cronex.mock_calls == [
call.check_trigger((2017, 11, 26, 13, 45))
]
assert cls._skip_reason == 'cronex: "foo"'
@freeze_time('2017-11-26 13:45:52')
def test_init_cronex_no_skip(self):
m_cronex = Mock()
m_cronex.check_trigger.return_value = True
with patch('%s.CronExpression' % pbm) as mock_cronex:
mock_cronex.return_value = m_cronex
cls = Job('jname', 'schedname', cron_expression='foo')
assert cls._name == 'jname'
assert cls._schedule_name == 'schedname'
assert cls._started is False
assert cls._finished is False
assert cls._exit_code is None
assert cls._output is None
assert cls._start_time is None
assert cls._finish_time is None
assert cls._summary_regex is None
assert cls._cron_expression == m_cronex
assert mock_cronex.mock_calls == [
call('foo'),
call().check_trigger((2017, 11, 26, 13, 45))
]
assert m_cronex.mock_calls == [
call.check_trigger((2017, 11, 26, 13, 45))
]
assert cls._skip_reason is None
def test_name(self):
assert self.cls.name == 'jname'
def test_skip(self):
assert self.cls.skip is None
def test_skip_true(self):
self.cls._skip_reason = 'foo'
assert self.cls.skip == 'foo'
def test_schedule_name(self):
assert self.cls.schedule_name == 'schedname'
def test_is_started(self):
self.cls._started = 2
assert self.cls.is_started == 2
def test_is_finished(self):
self.cls._finished = 7
assert self.cls.is_finished == 7
def test_run(self):
with pytest.raises(NotImplementedError):
self.cls.run()
def test_poll(self):
self.cls._finished = 6
assert self.cls.poll() == 6
def test_exit_code(self):
self.cls._exit_code = 23
assert self.cls.exitcode == 23
def test_output(self):
self.cls._output = 'foo'
assert self.cls.output == 'foo'
def test_repr(self):
assert self.cls.__repr__() == '<Job name="jname">'
def test_duration(self):
self.cls._start_time = datetime(2017, 11, 23, 14, 52, 34)
td = timedelta(seconds=3668)
self.cls._finish_time = self.cls._start_time + td
assert self.cls.duration == td
def test_error_repr(self):
self.cls._started = True
self.cls._output = 'foobar'
self.cls._exit_code = 9
expected = "<Job name=\"%s\">\nSchedule Name: %s\nStarted: %s\n" \
"Finished: %s\nDuration: %s\nExit Code: %s\nOutput: %s\n" % (
'jname', 'schedname', True, False, None, 9, 'foobar'
)
assert self.cls.error_repr == expected
def test_error_repr_no_ecode(self):
self.cls._started = True
self.cls._output = 'foobar'
expected = "<Job name=\"%s\">\nSchedule Name: %s\nStarted: %s\n" \
"Finished: %s\nDuration: %s\nOutput: %s\n" % (
'jname', 'schedname', True, False, None, 'foobar'
)
assert self.cls.error_repr == expected