django-webtest, аутентификация пользователей и просмотр декораторов

Question

Я использую django-webtest (v1.5.6), чтобы проверить, что декоратор ограничивает доступ к представлению для аутентифицированных пользователей.

Мой взгляд просто:

@active_account_required
def homepage(request):
    return render(request, 'accounts/account_homepage.html', {
        'user': request.user,
        })  

active_account_required Декоратор это:

def active_account_required(function = None):
    """ 
    Check the user is both logged in and has a valid, activated user account.

    If a user tries to access a view protected by this decorator, they will be
    redirected accordingly.

    See http://passingcuriosity.com/2009/writing-view-decorators-for-django/
    """

    def _dec(view_func):
        def _view(request, *args, **kwargs):
            if request.user.is_anonymous():
                return HttpResponseRedirect(reverse_lazy('auth_login'))
            if not request.user.get_profile().is_activated():
                return HttpResponseRedirect(reverse_lazy('registration_activation_incomplete'))
            return view_func(request, *args, **kwargs)

        _view.__name__ = view_func.__name__
        _view.__dict__ = view_func.__dict__
        _view.__doc__ = view_func.__doc__

        return _view

    if function is None:
        return _dec
    else:
        return _dec(function)

Мой метод испытаний

class AccountViewsTests(WebTest):
    def test_activated_user_can_access_account_homepage(self):
        """ 
        Test an activated user can access the account homepage
        """
        user = G(User)
        user.get_profile().datetime_activated = timezone.now()
        res = self.app.get(reverse('account_homepage'), user = user)
        pdb.set_trace()
        self.assertTemplateUsed(res, 'accounts/account_homepage.html',
                'Activated account did not access account homepage correctly')

(Пользовательский объект создается с использованием G функция от https://github.com/paulocheque/django-dynamic-fixture)

При запуске теста декоратор препятствует доступу к homepage Посмотреть.

Вы можете видеть, что я использую pdb осмотреть объекты. Пользователь - действительный объект пользователя, который должен пройти все тесты в active_account_required декоратор:

(Pdb) user
<User: 2>
(Pdb) user.is_anonymous()
False
(Pdb) user.get_profile().is_activated()
True

Несмотря на то, что пользователь прав, ответ от self.app.get(reverse('account_homepage'), user = user) 302 перенаправление на registration_activation_incomplete URL согласно коду декоратора:

(Pdb) res
<302 FOUND text/html location: http://localhost:80/accounts/registration/activate/incomplete/ no body>

Похоже, что пользовательский объект неправильно отправляется в запросе WebTest, но это соответствует документации django-webtest. Я также попытался передать пользователя по имени пользователя, как user='2' но получить тот же результат.

Есть идеи?

Answer 1

К сожалению, проблема в том, что я просто забыл сохранить свой профиль пользователя после установки отметки времени активации!

Изменение кода теста на:

user = G(User)
user.get_profile().datetime_activated = timezone.now()
user.get_profile().save()
res = self.app.get(reverse('account_homepage'), user = user)

т.е. добавление user.get_profile().save() получил это работает:)

Извините за шум.