def test_report_sending(self, move_mock, mail_mock):
with freeze_time('2016-01-05'):
move()
since = datetime(2016, 1, 4).date()
until = datetime(2016, 1, 5).date()
send_move_report("2016-01-04", "2016-01-05")
mail_mock.send_move_report.assert_called_once_with((since, until))
def test_returns_checkpoint(self, move_creds_mock):
move()
checkpoint = domain.get_last_move_checkpoint()
assert checkpoint.year == 2016
assert checkpoint.month == 2
assert checkpoint.day == 1
assert checkpoint.hour == 1
assert checkpoint.minute == 2
assert checkpoint.second == 3
def test_logs_start_of_successful_move(self, move_creds_mock):
move()
checkpoint = get_last_move_checkpoint()
assert checkpoint.year == 2016
assert checkpoint.month == 2
assert checkpoint.day == 1
assert checkpoint.hour == 1
assert checkpoint.minute == 2
assert checkpoint.second == 3
def test_command_send_email_for_given_dates(self, move_mock, mail_mock):
move()
since = datetime(2016, 1, 2).date()
until = datetime(2016, 1, 3).date()
call_command('send_report', since=since, until=until)
wait_for_ping()
mail_mock.send_move_report.assert_called_once_with((since, until))
def test_command_send_email_for_yesterday(self, move_mock, mail_mock):
move()
since = datetime(2016, 1, 4).date()
until = datetime(2016, 1, 5).date()
with capture_logs(logger) as stream:
call_command('send_report')
wait_for_ping()
mail_mock.send_move_report.assert_called_once_with((since, until))
assert 'INFO - Success sending report for' in stream[0]
def test_retries_if_checkpoint_older_than_report_end_date(
self, move_mock, mail_mock):
with freeze_time('2016-01-05'):
move()
since = datetime(2016, 1, 4).date()
until = datetime(2016, 1, 6).date()
call_command('send_report', since=since, until=until)
wait_for_ping()
assert not mail_mock.called
def test_filters_credentials_by_creation_date(self, move_creds_mock,
account):
with freeze_time('2016-01-01'):
cred1 = collector_domain.create_credential(account, 'zxcv.jpg',
'normal')
cred1.mark_as_found()
with freeze_time('2016-02-01'):
cred2 = collector_domain.create_credential(account, 'zxcv.jpg',
'normal')
cred2.mark_as_found()
with freeze_time('2016-02-01'):
move()
args, kwargs = move_creds_mock.call_args
assert move_creds_mock.call_count == 1
assert len(args) == 1
assert len(args[0]) == 1
assert args[0][0] == cred1
def test_retries_if_move_didnt_finish_on_time(self, move_mock, mail_mock,
settings):
settings.SEND_REPORT_RETRY_TIME = 1
with freeze_time('2016-01-05'):
move()
since = datetime(2016, 1, 4).date()
until = datetime(2016, 1, 6).date()
call_command('send_report', since=since, until=until)
wait_for_ping()
with freeze_time('2016-01-06'):
move()
wait_for_ping()
mail_mock.send_move_report.assert_called_once_with((since, until))
def move():
mover_commands.move()
def test_does_not_log_start_of_failed_move(self, move_creds_mock):
move_creds_mock.side_effect = ValueError()
with pytest.raises(ValueError):
move()
assert get_last_move_checkpoint() is None
