Andal.LND/edx-platform
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Properly clean up a wrapped function

Browse Source
This commit is contained in:
Ned Batchelder
2018-04-26 15:50:05 -04:00
parent 52e72c636d
commit 9b63a06bfc
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
openedx/core/djangoapps/oauth_dispatch/management/commands/tests/test_clear_expired_tokens.py
View File

@@ -25,7 +25,7 @@ def counter(fn):
"""
def _counted(*largs, **kargs):
_counted.invocations += 1
fn(*largs, **kargs)
return fn(*largs, **kargs)
_counted.invocations = 0
return _counted
@@ -90,8 +90,11 @@ class EdxClearExpiredTokensTests(TestCase):
AccessToken.objects.filter(refresh_token__isnull=True, expires__lt=now).count(),
initial_count
)
original_delete = QuerySet.delete
QuerySet.delete = counter(QuerySet.delete)
call_command('edx_clear_expired_tokens', batch_size=1, sleep_time=0)
self.assertEqual(QuerySet.delete.invocations, initial_count)
self.assertEqual(AccessToken.objects.filter(refresh_token__isnull=True, expires__lt=now).count(), 0)
try:
call_command('edx_clear_expired_tokens', batch_size=1, sleep_time=0)
self.assertEqual(QuerySet.delete.invocations, initial_count)
self.assertEqual(AccessToken.objects.filter(refresh_token__isnull=True, expires__lt=now).count(), 0)
finally:
QuerySet.delete = original_delete