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입니다. 당신이 user_passes_test 장식을 사용할 수 있습니다

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 

이 코드가 작동하지 않는 경우 (아직 테스트되지 않았습니다) :

그래서이 코드해야 작품이 생각이 하나가에

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이고

의 매개 변수를 허용해야합니다.

그래서 나는 확신이 일을 코드해야 작품 :

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'