фляга: при юнит-тестировании request.authorization всегда None - PullRequest
Новая
       12

фляга: при юнит-тестировании request.authorization всегда None

0 голосов
/ 26 июня 2018

Я надеюсь, что кто-нибудь может мне помочь.

Я должен написать юнит-тест с юнит-тестом Python в API фляги. У меня есть маршрут входа в систему, который прекрасно работает при доступе к нему через приложение с помощью интерфейса React, но всякий раз, когда я пытался отправить сообщение из теста, request.authorization - None ... Это сводит меня с ума

Я просмотрел весь интернет и попробовал много разных подходов, но что бы я ни делал, request.authorization всегда отсутствует при выполнении теста

Тестирование: 

import unittest
import base64

from backend.peace_api import app


class TestLogin(unittest.TestCase):

    # Assert login() with correct authentication
    def test_login(self):
        with app.app_context():
            tester = app.test_client(self)

            auth = 'seo@hotmail.com:password'

            authheader = base64.b64encode(bytes(auth, 'UTF-8'))
            headers = {"HTTP_AUTHORIZATION": "Bearer " + str(authheader), "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8"}

            response = tester.post('/api/login/', headers=dict(headers))
            print(response.json)

            self.assertEqual(response.status_code, 200)


if __name__ == '__main__':
    unittest.main()

Маршрут: 

import jwt
import datetime
from flask import Blueprint, request, jsonify
from backend.peace_api import database, secret_key
from backend.peace_api.flat.models.flat import Flat


login_blueprint = Blueprint("login", __name__)


@login_blueprint.route("/", methods=["POST"])
def login():
    auth = request.authorization # This here is always None
    print("Hello World")
    print(request)
    print(request.authorization)
    if auth is None:
        return jsonify({"success": False}, 401)

    email = auth.username
    password = auth.password

    if email is None or email is None or password is None:
        return jsonify({"success": False}, 500)

    mongo_flat = database.flats.find_one({"email": email})
    if mongo_flat is not None:
        flat = Flat(
            mongo_flat["_id"],
            mongo_flat["name"],
            mongo_flat["email"],
            mongo_flat["password"],
            mongo_flat["tasks"],
            mongo_flat["members"],
        )

        if password == flat.password and email == flat.email:
            token = jwt.encode(
                {
                    "id": str(flat.id),
                    "exp": datetime.datetime.utcnow() + datetime.timedelta(minutes=30),
                },
                secret_key,
            )
            return jsonify({"token": token.decode("UTF-8")})

        else:
            return jsonify({"success": False}, 401)
    else:
        return jsonify({"success": False}, 401)

Печатное сообщение: 

Testing started at 19:15 ...
Launching unittests with arguments python -m unittest test_login.TestLogin in [...]\tests
Hello World
<Request 'http://localhost/api/login/' [POST]>
None


Ran 1 test in 0.017s

OK
[{'success': False}, 401]

Честно говоря, я понятия не имею, что мне делать ... Спасибо за помощь

1 Ответ

0 голосов
/ 26 июня 2018

Итак, есть несколько проблем с вашей настройкой, которые приводят к тому, что заголовок не отправляется или отправляется, а искажается.

  1. Имя заголовка - «Авторизация», а не «HTTP_AUTHORIZATION».
  2. Значение учетных данных для заголовка авторизации должно быть в кодировке base64 для спецификации .
  3. По умолчанию промежуточное ПО авторизации для Werkzeug поддерживает только Basic auth , поэтому ваш токен Bearer не будет работать, если вы не используете расширение, добавляющее поддержку Bearer в Werkzeug (не зная больше о вашей настройке, трудно понять, что там происходит).

Вот очень упрощенное приложение Flask, которое демонстрирует работающий тестовый клиент с работающим заголовком Authorization: 

import flask
import base64

app = flask.Flask("app")

@app.route("/")
def test():
    print(flask.request.authorization)
    return "Worked"

with app.test_client() as c:
    c.get("/", headers={"Authorization": "Basic {}".format(base64.b64encode(b"useo@hotmail.com:pass").decode("utf8"))})

Какие отпечатки: 

{'password': 'pass', 'username': 'seo@hotmail.com'}
<Response streamed [200 OK]>

Подобный вопрос был задан здесь:

Flask werkzeug request.authorization отсутствует, кроме заголовков авторизации

...