from flask import _request_ctx_stack

from flask_atlas_auth import models
from flask_atlas_auth.utils import current_user


class TestCurrentUser:
    def test_current_user_authenticated(self, app, faker):
        test_user = models.User(
            faker.pystr(), faker.email(), faker.pystr(), []
        )
        with app.test_request_context():
            ctx = _request_ctx_stack.top
            ctx.user = test_user
            assert current_user == test_user
            assert current_user.is_authenticated

    def test_current_user_anonymous(self, app, mocker):
        with app.test_request_context():
            assert not current_user.is_authenticated