如何用自定義裝飾器測試視圖與django中的用戶相關？

Question

我有一個看起來像這樣的觀點：

@login_required 
@active_required() 
def myView(request): 
     print 'in my view' 

的active_required客戶裝飾看起來是這樣的：

def active_required(): 
    def decorator(func): 
     def inner_decorator(request, *args, **kwargs): 
      my_user = request.user 
      if my_user.active: 
       return func(request, *args, **kwargs) 
      else: 
        return HttpResponseRedirect(reverse('activate')) 

     return wraps(func)(inner_decorator) 

    return decorator 

我的測試是這樣的：

def test_my_view(self): 
    self.client.login(username='user', password='11111111') 
    response = self.client.post(reverse('my-view'), data, follow=True) 
    self.assertEqual(response.status_code, 200) 

我得到以下錯誤：

@active_required 
TypeError: active_required() takes no arguments (1 given) 

如果在測試中創建的數據庫不包含此用戶處於活動狀態，那麼如何添加它們？或者我因爲另一個原因而收到此錯誤？

Answer 1

那麼看Django login_reguiredsourcelogin_required需要作爲第一個參數function認爲這將是一個裝飾。

所以我認爲此代碼應工作：

def active_required(func): 
     @wraps(func) 
     def inner_decorator(request, *args, **kwargs): 
      my_user = request.user 
      if my_user.active: 
       return func(request, *args, **kwargs) 
      else: 
        return HttpResponseRedirect(reverse('activate')) 

     return inner_decorator 

    return active_required 

如果此代碼不能正常工作（尚未測試），你可以使用user_passes_test裝飾：

def user_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME): 
    """ 
    Decorator for views that checks that the user passes the given test, 
    redirecting to the log-in page if necessary. The test should be a callable 
    that takes the user object and returns True if the user passes. 
    """ 

    def decorator(view_func): 
     @wraps(view_func, assigned=available_attrs(view_func)) 
     def _wrapped_view(request, *args, **kwargs): 
      if test_func(request.user): 
       return view_func(request, *args, **kwargs) 
      path = request.build_absolute_uri() 
      # urlparse chokes on lazy objects in Python 3, force to str 
      resolved_login_url = force_str(
       resolve_url(login_url or settings.LOGIN_URL)) 
      # If the login url is the same scheme and net location then just 
      # use the path as the "next" url. 
      login_scheme, login_netloc = urlparse(resolved_login_url)[:2] 
      current_scheme, current_netloc = urlparse(path)[:2] 
      if ((not login_scheme or login_scheme == current_scheme) and 
       (not login_netloc or login_netloc == current_netloc)): 
       path = request.get_full_path() 
      from django.contrib.auth.views import redirect_to_login 
      return redirect_to_login(
       path, resolved_login_url, redirect_field_name) 
     return _wrapped_view 
    return decorator 

這一個是在鏈接上面和第一個參數這一個需要它是一個function和function必須接受一個參數，參數是user

這樣算下來，我敢肯定，你的代碼必須工作：

from django.contrib.auth.decorators import user_passes_test, login_required 
@login_required 
@user_passes_test(lambda user: user.is_active()) 
def myView(request): 
     print 'in my view'