def test_post_returns_token_after_regenerating(self, models,
pyramid_request):
"""After regenerating a token it should return its new value."""
data = views.DeveloperController(pyramid_request).post()
assert data[
'token'] == models.Token.get_dev_token_by_userid.return_value.value
def test_post_inits_new_token_for_authenticated_userid(self, models, pyramid_request):
"""If the user doesn't have a token yet it should generate one."""
models.Token.get_by_userid.return_value = None
views.DeveloperController(pyramid_request).post()
models.Token.assert_called_once_with(userid=pyramid_request.authenticated_userid)
def test_post_adds_new_token_to_db(self, models, pyramid_request):
"""If the user doesn't have a token yet it should add one to the db."""
models.Token.get_by_userid.return_value = None
views.DeveloperController(pyramid_request).post()
assert models.Token.return_value in pyramid_request.db.added
models.Token.assert_called_once_with(userid=pyramid_request.authenticated_userid)
def test_post_calls_regenerate(self, models, pyramid_request):
"""If the user already has a token it should regenerate it."""
views.DeveloperController(pyramid_request).post()
models.Token.get_dev_token_by_userid.return_value.regenerate.assert_called_with(
)
def test_post_gets_token_for_authenticated_userid(self, models,
pyramid_request):
views.DeveloperController(pyramid_request).post()
models.Token.get_dev_token_by_userid.assert_called_once_with(
pyramid_request.db, pyramid_request.authenticated_userid)
def test_get_with_no_token(self, models, pyramid_request):
models.Token.get_dev_token_by_userid.return_value = None
result = views.DeveloperController(pyramid_request).get()
assert result == {}
def test_get_returns_token(self, models, pyramid_request):
models.Token.get_dev_token_by_userid.return_value.value = u'abc123'
data = views.DeveloperController(pyramid_request).get()
assert data.get('token') == u'abc123'
def controller(self, pyramid_request):
return views.DeveloperController(pyramid_request)
