WebTest: тестирование с помощью декораторов + вызовы хранилища данных

Question

У меня есть приложение приложения Google App Engine, и мой запрос hasnler имеет декоратор, который выполняет аутентификацию. С помощью WebTest я обнаружил yesterday, как вы можете установить зарегистрированного пользователя и администратора.

Теперь сегодня мой декоратор аутентификации стал немного более сложным. Он также проверяет, имеет ли пользователь профиль в базе данных, и если он этого не сделает, он будет перенаправлен на страницу «новый пользователь».

def authenticated(method): 
    @functools.wraps(method) 
    def wrapper(self, *args, **kwargs): 
     user = users.get_current_user() 
     if not user: 
      self.redirect(users.create_login_url(self.request.uri)) 
      return 

     profile = Profile.get_by_key_name(str(user.user_id)) 
     if not profile: 
      self.redirect('/newuser') 

     return method(self, *args, **kwargs) 

    return wrapper 

Теперь добавление профиля части ломает мое модульное тестирование, который проверяет, если пользователь вошел в системе и получает код состояния 200 (assertOK).

def user_ok(self): 
    os.environ['USER_EMAIL'] = 'info@example.com' 
    os.environ['USER_IS_ADMIN'] = '' 
    response = self.get('/appindex') 
    self.assertOK(response) 

Так что теперь мне нужно, чтобы иметь возможность каким-то образом придать функциональность профиля в декоратора, так что я могу установить его в моих тестах. Кто-нибудь понял, как это сделать, я пытался придумать способ, но я все время застреваю.

Answer 1

Вы должны создать профиль во время теста, который будет использоваться декоратора:

def user_ok(self): 
    key_name = 'info@example.com' 
    new_user = Profile(key_name=key_name) 
    new_user.put() 

    os.environ['USER_EMAIL'] = key_name 
    os.environ['USER_ID'] = key_name 
    os.environ['USER_IS_ADMIN'] = '' 
    response = self.get('/appindex') 
    self.assertOK(response) 

    # Now let's reset it to check that the user will be redirected. 
    new_user.delete() 
    response = self.get('/appindex') 
    self.assertEqual(response.headers['Location'], 'http://localhost/newuser')