test_commands.py 文件源码

def test_call_must_have_first_and_last_message_correct(self):
        scheduler = Scheduler(report='my_report_class',
                              periodicity=Scheduler.PER_MON_SUN)
        scheduler.save()

        out = StringIO()
        management.call_command('scheduler_process', stdout=out)

        first_line, last_line, blank_line = (
            out.getvalue().split('\n')[0],
            out.getvalue().split('\n')[-2],
            out.getvalue().split('\n')[-1]
        )

        first_message = 'Starting scheduler process'
        last_message = 'Scheduler #{} processed'.format(scheduler.id)

        self.assertIn(first_message, first_line)
        self.assertIn(last_message, last_line)
        self.assertEqual(blank_line, '')