Added the pytest test example to README.rst.

This commit is contained in:
依瑪貓 2022-12-06 07:54:49 +08:00
parent 2028cb1362
commit cbbd2248f0

View File

@ -359,10 +359,13 @@ Writing Tests
============= =============
You can write tests with our test client that handles HTTP Digest You can write tests with our test client that handles HTTP Digest
Authentication. Example for a unittest testcase: Authentication.
Example for a unittest_ test case:
:: ::
from flask import Flask
from flask_digest_auth import Client from flask_digest_auth import Client
from flask_testing import TestCase from flask_testing import TestCase
from my_app import create_app from my_app import create_app
@ -385,6 +388,41 @@ Authentication. Example for a unittest testcase:
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
Example for a pytest_ test:
::
import pytest
from flask import Flask
from flask_digest_auth import Client
from my_app import create_app
@pytest.fixture()
def app():
app: Flask = create_app({
"SECRET_KEY": token_urlsafe(32),
"TESTING": True
})
app.test_client_class = Client
yield app
@pytest.fixture()
def client(app):
return app.test_client()
def test_admin(app: Flask, client: Client):
with app.app_context():
response = self.client.get("/admin")
assert response.status_code == 401
response = self.client.get(
"/admin", digest_auth=("my_name", "my_pass"))
assert response.status_code == 200
.. _unittest: https://docs.python.org/3/library/unittest.html
.. _pytest: https://pytest.org
Copyright Copyright
========= =========