1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
# Copyright 2012 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Tests for :class:`User`-related helpers."""
from __future__ import (
absolute_import,
print_function,
unicode_literals,
)
__metaclass__ = type
__all__ = []
from maasserver.models.user import (
create_auth_token,
get_auth_tokens,
)
from maasserver.testing.factory import factory
from maasserver.testing.testcase import TestCase
from piston.models import (
KEY_SIZE,
SECRET_SIZE,
)
class AuthTokensTest(TestCase):
"""Test creation and retrieval of auth tokens."""
def assertTokenValid(self, token):
self.assertIsInstance(token.key, basestring)
self.assertEqual(KEY_SIZE, len(token.key))
self.assertIsInstance(token.secret, basestring)
self.assertEqual(SECRET_SIZE, len(token.secret))
def assertConsumerValid(self, consumer):
self.assertIsInstance(consumer.key, basestring)
self.assertEqual(KEY_SIZE, len(consumer.key))
self.assertEqual('', consumer.secret)
def test_create_auth_token(self):
user = factory.make_user()
token = create_auth_token(user)
self.assertEqual(user, token.user)
self.assertEqual(user, token.consumer.user)
self.assertTrue(token.is_approved)
self.assertConsumerValid(token.consumer)
self.assertTokenValid(token)
def test_get_auth_tokens_finds_tokens_for_user(self):
user = factory.make_user()
token = create_auth_token(user)
self.assertIn(token, get_auth_tokens(user))
def test_get_auth_tokens_ignores_other_users(self):
user, other_user = factory.make_user(), factory.make_user()
unrelated_token = create_auth_token(other_user)
self.assertNotIn(unrelated_token, get_auth_tokens(user))
def test_get_auth_tokens_ignores_unapproved_tokens(self):
user = factory.make_user()
token = create_auth_token(user)
token.is_approved = False
token.save()
self.assertNotIn(token, get_auth_tokens(user))
|