vh/NetAlertX
1
0
Fork 0
mirror of https://github.com/jokob-sk/NetAlertX.git synced 2026-04-04 09:11:34 -07:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Coderabbit requested changes

Browse Source
This commit is contained in:
Adam Outler
2026-01-31 16:52:30 +00:00
parent 53962bc38b
commit 77fd017d90
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
test/api_endpoints/test_docs.py
View File

@@ -6,19 +6,18 @@ from api_server import api_server_start
def test_index_redirect_logic(): def test_index_redirect_logic():
"""Test the redirect function logic directly.""" """Test the redirect function logic directly."""
with api_server_start.app.test_request_context(): with api_server_start.app.test_client() as client:
response = api_server_start.index_redirect() response = client.get("/", follow_redirects=False)
# In Flask, a redirect return object is a Response
assert response.status_code == 302 assert response.status_code == 302
assert response.headers['Location'] == '/docs' assert response.location == '/docs'
def test_api_docs_logic(): def test_api_docs_logic():
"""Test that api_docs attempts to serve the correct file.""" """Test that api_docs attempts to serve the correct file."""
with api_server_start.app.test_request_context(): with api_server_start.app.test_client() as client:
response = api_server_start.api_docs() response = client.get("/docs")
assert response.status_code == 200 assert response.status_code == 200
assert response.mimetype == 'text/html' assert response.mimetype == "text/html"
# Manually join the chunks from the generator response.direct_passthrough = False
data = b"".join(response.response) data = response.get_data()
assert b'<!DOCTYPE html>' in data or b'<html>' in data assert b"<!DOCTYPE html>" in data or b"<html>" in data