def update_current_prisons(**kwargs):
job = ScheduledCommand(
name='update_current_prisons',
arg_string='',
cron_entry='*/10 * * * *',
delete_after_next=True
)
job.save()
def update_current_prisons(*args, **kwargs):
job = ScheduledCommand(
name='update_current_prisons',
arg_string='',
cron_entry='*/10 * * * *',
delete_after_next=True
)
job.save()
def test_not_due_command_is_not_scheduled(self):
command = ScheduledCommand(
name='load_test_data',
arg_string='',
cron_entry='*/10 * * * *',
next_execution=timezone.now() + timedelta(minutes=5)
)
command.save()
self.assertFalse(command.is_scheduled())
def test_cron_entry_validation_fails(self):
try:
command = ScheduledCommand(
name='load-test_data', arg_string='', cron_entry='* F F * *'
)
command.clean_fields()
self.fail('No ValidationError raised by invalid cron entry')
except ValidationError:
pass
def test_command_validation_fails(self):
try:
command = ScheduledCommand(
name='non_existent_command', arg_string='', cron_entry='* * * * *'
)
command.clean_fields()
self.fail('No ValidationError raised by invalid command')
except ValidationError:
pass
def test_command_deleted_when_flag_set(self):
command = ScheduledCommand(
name='load_test_data',
arg_string='--number-of-prisoners 60 --number-of-transactions 70',
cron_entry='* * * * *',
next_execution=timezone.now(),
delete_after_next=True
)
command.save()
run_commands = run_scheduled_commands.Command()
with captured_stdout(), silence_logger():
run_commands.handle()
self.assertEqual(ScheduledCommand.objects.all().count(), 0)
def save(self):
from performance.models import DigitalTakeup
for nomis_id, credit_by_prison in self.credits_by_prison.items():
DigitalTakeup.objects.update_or_create(
defaults=credit_by_prison,
date=self.date,
prison_id=nomis_id,
)
datestr = self.date.strftime('%Y-%m-%dT00:00:00')
job = ScheduledCommand(
name='update_performance_platform',
arg_string='--resources transactions-by-channel-type --timestamp %s' % datestr,
cron_entry='*/10 * * * *',
delete_after_next=True
)
job.save()
def save(self):
from performance.models import DigitalTakeup
for nomis_id, credit_by_prison in self.credits_by_prison.items():
DigitalTakeup.objects.update_or_create(
defaults=credit_by_prison,
date=self.date,
prison_id=nomis_id,
)
datestr = self.date.strftime('%Y-%m-%dT00:00:00')
job = ScheduledCommand(
name='update_performance_platform',
arg_string='--resources transactions-by-channel-type --timestamp %s'
% datestr,
cron_entry='*/10 * * * *',
delete_after_next=True)
job.save()
def test_command_running(self):
command = ScheduledCommand(
name='load_test_data',
arg_string='--number-of-prisoners 60 --number-of-transactions 70',
cron_entry='* * * * *',
next_execution=timezone.now()
)
command.save()
run_commands = run_scheduled_commands.Command()
with captured_stdout() as stdout, silence_logger():
run_commands.handle()
stdout = stdout.getvalue()
self.assertIn('Making test users', stdout)
self.assertIn('random credits', stdout)
from prison.models import PrisonerLocation
self.assertEqual(PrisonerLocation.objects.count(), 60)
from transaction.models import Transaction
self.assertEqual(Transaction.objects.count(), 70)
def test_newly_created_command_is_scheduled(self):
command = ScheduledCommand(
name='load_test_data', arg_string='', cron_entry='* * * * *'
)
command.save()
self.assertTrue(timezone.now() - command.next_execution < timedelta(minutes=1))
def test_command_validation(self):
command = ScheduledCommand(
name='load_test_data', arg_string='', cron_entry='* * * * *'
)
command.clean_fields()