mirror of
https://github.com/frappe/gunicorn.git
synced 2026-01-14 11:09:11 +08:00
85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
import datetime
|
|
|
|
from gunicorn.config import Config
|
|
from gunicorn.glogging import Logger
|
|
|
|
from support import SimpleNamespace
|
|
|
|
|
|
def test_atoms_defaults():
|
|
response = SimpleNamespace(
|
|
status='200', response_length=1024,
|
|
headers=(('Content-Type', 'application/json'),), sent=1024,
|
|
)
|
|
request = SimpleNamespace(headers=(('Accept', 'application/json'),))
|
|
environ = {
|
|
'REQUEST_METHOD': 'GET', 'RAW_URI': '/my/path?foo=bar',
|
|
'PATH_INFO': '/my/path', 'QUERY_STRING': 'foo=bar',
|
|
'SERVER_PROTOCOL': 'HTTP/1.1',
|
|
}
|
|
logger = Logger(Config())
|
|
atoms = logger.atoms(response, request, environ, datetime.timedelta(seconds=1))
|
|
assert isinstance(atoms, dict)
|
|
assert atoms['r'] == 'GET /my/path?foo=bar HTTP/1.1'
|
|
assert atoms['m'] == 'GET'
|
|
assert atoms['U'] == '/my/path'
|
|
assert atoms['q'] == 'foo=bar'
|
|
assert atoms['H'] == 'HTTP/1.1'
|
|
assert atoms['b'] == '1024'
|
|
assert atoms['B'] == 1024
|
|
assert atoms['{accept}i'] == 'application/json'
|
|
assert atoms['{content-type}o'] == 'application/json'
|
|
|
|
|
|
def test_atoms_zero_bytes():
|
|
response = SimpleNamespace(
|
|
status='200', response_length=0,
|
|
headers=(('Content-Type', 'application/json'),), sent=0,
|
|
)
|
|
request = SimpleNamespace(headers=(('Accept', 'application/json'),))
|
|
environ = {
|
|
'REQUEST_METHOD': 'GET', 'RAW_URI': '/my/path?foo=bar',
|
|
'PATH_INFO': '/my/path', 'QUERY_STRING': 'foo=bar',
|
|
'SERVER_PROTOCOL': 'HTTP/1.1',
|
|
}
|
|
logger = Logger(Config())
|
|
atoms = logger.atoms(response, request, environ, datetime.timedelta(seconds=1))
|
|
assert atoms['b'] == '0'
|
|
assert atoms['B'] == 0
|
|
|
|
|
|
def test_get_username_from_basic_auth_header():
|
|
request = SimpleNamespace(headers=())
|
|
response = SimpleNamespace(
|
|
status='200', response_length=1024, sent=1024,
|
|
headers=(('Content-Type', 'text/plain'),),
|
|
)
|
|
environ = {
|
|
'REQUEST_METHOD': 'GET', 'RAW_URI': '/my/path?foo=bar',
|
|
'PATH_INFO': '/my/path', 'QUERY_STRING': 'foo=bar',
|
|
'SERVER_PROTOCOL': 'HTTP/1.1',
|
|
'HTTP_AUTHORIZATION': 'Basic YnJrMHY6',
|
|
}
|
|
logger = Logger(Config())
|
|
atoms = logger.atoms(response, request, environ, datetime.timedelta(seconds=1))
|
|
assert atoms['u'] == 'brk0v'
|
|
|
|
|
|
def test_get_username_handles_malformed_basic_auth_header():
|
|
"""Should catch a malformed auth header"""
|
|
request = SimpleNamespace(headers=())
|
|
response = SimpleNamespace(
|
|
status='200', response_length=1024, sent=1024,
|
|
headers=(('Content-Type', 'text/plain'),),
|
|
)
|
|
environ = {
|
|
'REQUEST_METHOD': 'GET', 'RAW_URI': '/my/path?foo=bar',
|
|
'PATH_INFO': '/my/path', 'QUERY_STRING': 'foo=bar',
|
|
'SERVER_PROTOCOL': 'HTTP/1.1',
|
|
'HTTP_AUTHORIZATION': 'Basic ixsTtkKzIpVTncfQjbBcnoRNoDfbnaXG',
|
|
}
|
|
logger = Logger(Config())
|
|
|
|
atoms = logger.atoms(response, request, environ, datetime.timedelta(seconds=1))
|
|
assert atoms['u'] == '-'
|