Beispiel #1
0
    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
Beispiel #2
0
    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