import pytest
from flask import Flask
from flask import render_template_string

from flask_jwt_extended import create_access_token
from flask_jwt_extended import jwt_required
from flask_jwt_extended import JWTManager


@pytest.fixture(scope="function")
def app():
    app = Flask(__name__)
    app.config["JWT_SECRET_KEY"] = "foobarbaz"

    @app.route("/context_current_user", methods=["GET"])
    @jwt_required()
    def context_current_user():
        return render_template_string("{{ current_user }}")

    return app


def test_add_context_processor(app):
    jwt_manager = JWTManager(app, add_context_processor=True)

    @jwt_manager.user_lookup_loader
    def user_lookup_callback(_jwt_header, _jwt_data):
        return "test_user"

    test_client = app.test_client()

    with app.test_request_context():
        access_token = create_access_token("username")

    access_headers = {"Authorization": "Bearer {}".format(access_token)}
    response = test_client.get("/context_current_user", headers=access_headers)
    assert response.data == b"test_user"


def test_no_add_context_processor(app):
    jwt_manager = JWTManager(app)

    @jwt_manager.user_lookup_loader
    def user_lookup_callback(_jwt_header, _jwt_data):
        return "test_user"

    test_client = app.test_client()

    with app.test_request_context():
        access_token = create_access_token("username")

    access_headers = {"Authorization": "Bearer {}".format(access_token)}
    response = test_client.get("/context_current_user", headers=access_headers)
    assert response.data == b""