def test_authenticated_user(self): #Arrange f = Mock() this = Mock() this.users.get_current_user.return_value = True #Act decorator = authenticated(f) decorator(this) #Assert assert this.users.get_current_user.called assert f.called
def test_authenticated_no_user(self): #Arrange f = Mock() this = Mock() this.users.get_current_user.return_value = None this.users.create_login_url.return_value = '/login' #Act decorator = authenticated(f) result = decorator(this) #Assert assert not f.called assert this.users.get_current_user.called assert type(result) == Redirect