PEP343는

Question

나는 비활성 사용자가 로그인 할 수 있는지 테스트 어디서 경우가 있고, 내가 그렇게

self.testuser.is_active = False 
//DO testing 
self.testuser.is_active = True 
//Proceed 

내 질문에 같이 할 장고 프레임 워크와 일부 응용 프로그램 테스트, 을하고있는 중이 야 PEP343 에 의해 제공되는 컨텍스트 매니저와 함께 사용하여, 이다 내가이 일을 시도하지만 난 다음

with self.testuser.is_active = False : 
//code 

실패 내가

을하려고 노력 이 주위에 방법이 16,

with self.settings(self.__set_attr(self.testuser.is_active = False)): 
//code 

그것은 또한

실패인가? 또는 is_active를 false로 설정하는 함수를 정의해야합니까?

Answer 1

이다는 contextlib에서 내장 더 일반적인 컨텍스트 관리자입니다.

from contextlib import contextmanager 

@contextmanager 
def temporary_changed_attr(object, attr, value): 
    if hasattr(object, attr): 
     old = getattr(object, attr) 
     setattr(object, attr, value) 
     yield 
     setattr(object, attr, old) 
    else: 
     setattr(object, attr, value) 
     yield 
     delattr(object, attr) 

# Example usage 
with temporary_changed_attr(self.testuser, 'is_active', False): 
    # self.testuser.is_active will be false in here 

Answer 2

자신 만의 컨텍스트 관리자를 작성해야합니다. 여기에 (contextlib 사용) 사건에 대한 하나의 :

더 나은

import contextlib 
@contextlib.contextmanager 
def toggle_active_user(user): 
    user.is_active = False 
    yield 
    user.is_active = True 

with toggle_active_user(self.testuser): 
    // Do testing 

, 복원 후 이전 상태를 저장하는 것입니다 :

여기

import contextlib 
@contextlib.contextmanager 
def toggle_active_user(user, new_value): 
    previous_is_active = user.is_active 
    user.is_active = new_value 
    yield 
    user.is_active = previous_is_active 

with toggle_active_user(self.testuser, False): 
    // Do testing