def test_periodic_func(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
# We need to define the function in here to make sure the first
# call is scheduled to the frozen time
@periodic(timedelta(minutes=5), "arg")
def periodic_function(arg):
CronTest.run_count += 1
CronTest.received_arg = arg
self.assertEqual(CronTest.run_count, 0)
run_scheduled_tasks()
# Check the function has run and got the correct argument
self.assertEqual(CronTest.run_count, 1)
self.assertEqual(CronTest.received_arg, "arg")
# Now check that it is called at correct intervals
for _ in range(0, 32):
frozen_time.tick(delta=timedelta(minutes=1))
run_scheduled_tasks()
self.assertEqual(CronTest.run_count, 7)
# Do some cleanup! Else this will get called in other tests..
cron.periodic_functions.remove(periodic_function)
def test_periodic_func(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
# We need to define the function in here to make sure the first
# call is scheduled to the frozen time
@periodic(timedelta(minutes=5), "arg")
def periodic_function(arg):
CronTest.run_count += 1
CronTest.received_arg = arg
self.assertEqual(CronTest.run_count, 0)
run_scheduled_tasks()
# Check the function has run and got the correct argument
self.assertEqual(CronTest.run_count, 1)
self.assertEqual(CronTest.received_arg, "arg")
# Now check that it is called at correct intervals
for _ in range(0, 32):
frozen_time.tick(delta=timedelta(minutes=1))
run_scheduled_tasks()
self.assertEqual(CronTest.run_count, 7)
# Do some cleanup! Else this will get called in other tests..
cron.periodic_functions.remove(periodic_function)
def test_schedule_once_soon_works(self):
with self.app.app_context():
CronTest.run_count = 0
@schedulable
def inc():
CronTest.run_count += 1
schedule_once_soon(inc)
print("nmow")
schedule_once_soon(inc)
run_scheduled_tasks()
self.assertEqual(CronTest.run_count, 1)
def test_schedule_once_soon_works(self):
with self.app.app_context():
CronTest.run_count = 0
@schedulable
def inc():
CronTest.run_count += 1
schedule_once_soon(inc)
print("nmow")
schedule_once_soon(inc)
run_scheduled_tasks()
self.assertEqual(CronTest.run_count, 1)
def test_session_expiry(self):
with self.app.app_context(), freeze_time() as frozen_time:
self.new_object("users", nethz="pablo", password="******")
self.api.post('/sessions',
data={"username": "******", "password": "******"},
status_code=201)
frozen_time.tick(delta=self.app.config['SESSION_TIMEOUT'] -
timedelta(days=1))
run_scheduled_tasks()
self.assertEqual(self.app.data.driver.db['sessions'].find().count(),
1)
frozen_time.tick(delta=timedelta(days=2))
run_scheduled_tasks()
self.assertEqual(self.app.data.driver.db['sessions'].find().count(),
0)
def test_session_expiry(self):
with self.app.app_context(), freeze_time() as frozen_time:
self.new_object("users", nethz="pablo", password="******")
self.api.post('/sessions',
data={"username": "******", "password": "******"},
status_code=201)
frozen_time.tick(delta=self.app.config['SESSION_TIMEOUT'] -
timedelta(days=1))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['sessions'].count_documents({}), 1)
frozen_time.tick(delta=timedelta(days=2))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['sessions'].count_documents({}), 0)
def test_scheduled_email_on_delete(self):
"""Test that a user receives only one an email if an entry is deleted"""
with self.app.app_context(), freeze_time(
"2017-01-01 00:00:00") as frozen_time:
user_id = 24 * '0'
blacklist_id = 24 * '1'
# Create user and blacklist entry
self.load_fixture(
{'users': [{
'_id': user_id,
'email': "*****@*****.**"
}]})
r = self.load_fixture({
'blacklist': [{
'_id': blacklist_id,
'user': user_id,
'reason': "Test1",
'end_time': "2017-01-02T00:00:00Z",
}]
})
etag = r[0]['_etag']
header = {'If-Match': etag}
with freeze_time(datetime(2017, 1, 1)):
r = self.api.delete("/blacklist/%s" % blacklist_id,
headers=header,
token=self.get_root_token(),
status_code=204)
run_scheduled_tasks()
# Only the creation email should be sent
self.assertEqual(len(self.app.test_mails), 2)
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
# Since the entry was deleted no mail should be sent
self.assertEqual(len(self.app.test_mails), 2)
def test_scheduled_function_gets_called(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
@schedulable
def scheduled_function(arg):
CronTest.has_run = True
CronTest.received_arg = arg
schedule_task(datetime(2016, 1, 1, 1, 0, 0),
scheduled_function,
"arg")
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
frozen_time.tick(delta=timedelta(hours=1))
run_scheduled_tasks()
self.assertTrue(CronTest.has_run)
self.assertEqual(CronTest.received_arg, "arg")
def test_scheduled_function_gets_called(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
@schedulable
def scheduled_function(arg):
CronTest.has_run = True
CronTest.received_arg = arg
schedule_task(datetime(2016, 1, 1, 1, 0, 0), scheduled_function,
"arg")
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
frozen_time.tick(delta=timedelta(hours=1))
run_scheduled_tasks()
self.assertTrue(CronTest.has_run)
self.assertEqual(CronTest.received_arg, "arg")
def test_update_scheduled_task(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
@schedulable
def tester(arg):
CronTest.has_run = True
CronTest.received_arg = arg
schedule_task(datetime(2016, 1, 1, 1, 0, 0),
tester,
"arg")
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
update_scheduled_task(datetime(2016, 1, 1, 3, 20, 0),
tester,
"new-arg")
frozen_time.tick(delta=timedelta(hours=2))
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
frozen_time.tick(delta=timedelta(hours=3))
run_scheduled_tasks()
self.assertTrue(CronTest.has_run)
self.assertEqual(CronTest.received_arg, "new-arg")
def test_update_scheduled_task(self):
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
@schedulable
def tester(arg):
CronTest.has_run = True
CronTest.received_arg = arg
schedule_task(datetime(2016, 1, 1, 1, 0, 0), tester, "arg")
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
update_scheduled_task(datetime(2016, 1, 1, 3, 20, 0), tester,
"new-arg")
frozen_time.tick(delta=timedelta(hours=2))
run_scheduled_tasks()
self.assertFalse(CronTest.has_run)
frozen_time.tick(delta=timedelta(hours=3))
run_scheduled_tasks()
self.assertTrue(CronTest.has_run)
self.assertEqual(CronTest.received_arg, "new-arg")
def test_expired_groupmembership_gets_removed(self):
user = self.new_object('users')
group = self.new_object('groups')
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
self.new_object('groupmemberships',
user=str(user['_id']),
group=str(group['_id']),
expiry='2016-01-03T00:00:00Z')
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['groupmemberships'].find().count(), 1)
frozen_time.tick(delta=timedelta(days=2))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['sessions'].find().count(), 0)
def test_expired_groupmembership_gets_removed(self):
user = self.new_object('users')
group = self.new_object('groups')
with self.app.app_context(), freeze_time(
"2016-01-01 00:00:00") as frozen_time:
self.new_object('groupmemberships',
user=str(user['_id']),
group=str(group['_id']),
expiry='2016-01-03T00:00:00Z')
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['groupmemberships'].count_documents({}),
1)
frozen_time.tick(delta=timedelta(days=2))
run_scheduled_tasks()
self.assertEqual(
self.app.data.driver.db['sessions'].count_documents({}), 0)
def test_receive_scheduled_email_on_create(self):
"""Test if a user receives an email if the end_time is reached"""
with self.app.app_context(), freeze_time(
"2017-01-01 00:00:00") as frozen_time:
user_id = 24 * '0'
blacklist_id = 24 * '1'
# Create user and blacklist entry
self.load_fixture(
{'users': [{
'_id': user_id,
'email': "*****@*****.**"
}]})
self.load_fixture({
'blacklist': [{
'_id': blacklist_id,
'user': user_id,
'reason': "Test1",
'end_time': datetime(2017, 1, 2)
}]
})
run_scheduled_tasks()
self.assertEqual(len(self.app.test_mails), 1)
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
# Check mail
mail = self.app.test_mails[1]
self.assertEqual(mail['receivers'], '*****@*****.**')
expected_text = (
"Congratulations, your blacklist entry with the following "
"reason has been removed:\n\nTest1\n\nBest Regards,\nAMIV")
self.assertEqual(mail['text'], expected_text)
def test_receive_scheduled_email_on_patch(self):
"""Test if a user receives an email if the end_time is reached"""
with self.app.app_context(), freeze_time(
"2017-01-01 00:00:00") as frozen_time:
user_id = 24 * '0'
blacklist_id = 24 * '1'
# Create user and blacklist entry
self.load_fixture(
{'users': [{
'_id': user_id,
'email': "*****@*****.**"
}]})
r = self.load_fixture({
'blacklist': [{
'_id': blacklist_id,
'user': user_id,
'reason': "Test1",
'end_time': "2017-01-02T00:00:00Z",
}]
})
etag = r[0]['_etag']
patch = {
'user': user_id,
'reason': "Test1",
'end_time': '2017-01-03T00:00:00Z'
}
header = {'If-Match': etag}
with freeze_time(datetime(2017, 1, 1)):
r = self.api.patch("/blacklist/%s" % blacklist_id,
data=patch,
headers=header,
token=self.get_root_token(),
status_code=200)
run_scheduled_tasks()
# Only the creation email should be sent
self.assertEqual(len(self.app.test_mails), 1)
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
# Since the date was changed to the future, no email should be sent
self.assertEqual(len(self.app.test_mails), 1)
frozen_time.tick(delta=timedelta(days=1))
run_scheduled_tasks()
# Check mail
mail = self.app.test_mails[1]
self.assertEqual(mail['receivers'], '*****@*****.**')
expected_text = (
"Congratulations, your blacklist entry with the following "
"reason has been removed:\n\nTest1\n\nBest Regards,\nAMIV")
self.assertEqual(mail['text'], expected_text)
def run_cron(app):
"""Run scheduled tasks with the given app."""
echo("Executing scheduled tasks...")
with app.app_context():
run_scheduled_tasks()
def run_cron(app):
"""Run scheduled tasks with the given app."""
echo("Executing scheduled tasks...")
with app.app_context():
run_scheduled_tasks()
def cron(config):
"""Run scheduled tasks."""
app = create_app(config_file=config)
with app.app_context():
run_scheduled_tasks()