コード例 #1
0
ファイル: test_controller.py プロジェクト: Enome/active_app
    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
コード例 #2
0
ファイル: test_controller.py プロジェクト: Enome/active_app
    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