24 lines
689 B
Python
24 lines
689 B
Python
import mock
|
|
from test_manager import login
|
|
|
|
|
|
def test_maintenance_middleware(app, admin_user, db, monkeypatch, settings):
|
|
app = login(app)
|
|
resp = app.get('/')
|
|
assert resp.status_code == 200
|
|
|
|
settings.MAINTENANCE_MODE = True
|
|
resp = app.get('/', status=503)
|
|
assert 'The site is under maintenance' in resp.text
|
|
|
|
settings.MAINTENANCE_PASS_THROUGH_IPS = ['127.0.0.1']
|
|
resp = app.get('/')
|
|
assert resp.status_code == 200
|
|
|
|
settings.MAINTENANCE_PASS_THROUGH_IPS = []
|
|
resp = app.get('/', status=503)
|
|
|
|
settings.MAINTENANCE_PASS_THROUGH_HEADER = 'X-Entrouvert'
|
|
resp = app.get('/', headers={'X-Entrouvert': 'yes'})
|
|
assert resp.status_code == 200
|