def test_generate(self):
testuser = create_user('testuser')
unique_user_token = GenericTokenWithMetadata.objects.generate(
content_object=testuser, app='testapp',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEqual(unique_user_token.content_object, testuser)
self.assertEqual(unique_user_token.app, 'testapp')
self.assertEqual(len(unique_user_token.token), 73)
def test_generate(self):
testuser = create_user('testuser')
unique_user_token = GenericTokenWithMetadata.objects.generate(
content_object=testuser,
app='testapp',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEqual(unique_user_token.content_object, testuser)
self.assertEqual(unique_user_token.app, 'testapp')
self.assertEqual(len(unique_user_token.token), 73)
def test_generate_handle_not_unique(self):
self._create_generic_token_with_metadata(user=create_user('testuser1'), app='testapp1', token='taken')
tokens = iter(['taken', 'free'])
with mock.patch('django_cradmin.apps.cradmin_generic_token_with_metadata.models.generate_token',
lambda: next(tokens)):
unique_user_token = GenericTokenWithMetadata.objects.generate(
content_object=create_user('testuser2'), app='testapp2',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEqual(unique_user_token.token, 'free')
def test_filter_by_content_object(self):
testobject1 = create_user('testobject1')
testobject2 = create_user('testobject2')
generictoken = GenericTokenWithMetadata.objects.generate(
content_object=testobject1, app='testapp',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEquals(
GenericTokenWithMetadata.objects.filter_by_content_object(testobject1).get(),
generictoken)
self.assertEquals(GenericTokenWithMetadata.objects.filter_by_content_object(testobject2).count(), 0)
def test_filter_by_content_object(self):
testobject1 = create_user('testobject1')
testobject2 = create_user('testobject2')
generictoken = GenericTokenWithMetadata.objects.generate(
content_object=testobject1,
app='testapp',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEqual(
GenericTokenWithMetadata.objects.filter_by_content_object(
testobject1).get(), generictoken)
self.assertEqual(
GenericTokenWithMetadata.objects.filter_by_content_object(
testobject2).count(), 0)
def test_generate_handle_not_unique(self):
self._create_generic_token_with_metadata(user=create_user('testuser1'),
app='testapp1',
token='taken')
tokens = iter(['taken', 'free'])
with mock.patch(
'django_cradmin.apps.cradmin_generic_token_with_metadata.models.generate_token',
lambda: next(tokens)):
unique_user_token = GenericTokenWithMetadata.objects.generate(
content_object=create_user('testuser2'),
app='testapp2',
expiration_datetime=get_expiration_datetime_for_app('testapp'))
self.assertEqual(unique_user_token.token, 'free')
def get_expiration_datetime(self):
"""
Get the value to use for the ``expiration_datetime`` attribute of
:class:`~django_cradmin.apps.cradmin_generic_token_with_metadata.models.GenericTokenWithMetadata`.
Defaults to the expiration_datetime provided via the constructor,
and falls back to getting the configured expiration datetime for
the app.
"""
if hasattr(self, 'expiration_datetime'):
return self.expiration_datetime
else:
return get_expiration_datetime_for_app(self.get_appname())
def get_expiration_datetime(self):
"""
Get the value to use for the ``expiration_datetime`` attribute of
:class:`~django_cradmin.apps.cradmin_generic_token_with_metadata.models.GenericTokenWithMetadata`.
Defaults to the expiration_datetime provided via the constructor,
and falls back to getting the configured expiration datetime for
the app.
"""
if hasattr(self, 'expiration_datetime'):
return self.expiration_datetime
else:
return get_expiration_datetime_for_app(self.get_appname())
def get_expiration_datetime(self):
"""
Get the value to use for the ``expiration_datetime`` attribute of
:class:`~django_cradmin.apps.cradmin_generic_token_with_metadata.models.GenericTokenWithMetadata`.
"""
return get_expiration_datetime_for_app(self.appname)
def _generate_token(self, user):
return GenericTokenWithMetadata.objects.generate(
app='cradmin_resetpassword',
content_object=user,
expiration_datetime=get_expiration_datetime_for_app(
'cradmin_resetpassword')).token
def get_expiration_datetime(self):
"""
Get the value to use for the ``expiration_datetime`` attribute of
:class:`~django_cradmin.apps.cradmin_generic_token_with_metadata.models.GenericTokenWithMetadata`.
"""
return get_expiration_datetime_for_app(self.appname)
