mirror of https://github.com/JohnHammond/CTFd.git
34 lines
1000 B
Python
34 lines
1000 B
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from tests.helpers import create_ctfd, destroy_ctfd
|
|
from CTFd.config import TestingConfig
|
|
|
|
|
|
def test_reverse_proxy_config():
|
|
"""Test that REVERSE_PROXY configuration behaviors properly"""
|
|
|
|
class ReverseProxyConfig(TestingConfig):
|
|
REVERSE_PROXY = "1,2,3,4"
|
|
|
|
app = create_ctfd(config=ReverseProxyConfig)
|
|
with app.app_context():
|
|
assert app.wsgi_app.x_for == 1
|
|
assert app.wsgi_app.x_proto == 2
|
|
assert app.wsgi_app.x_host == 3
|
|
assert app.wsgi_app.x_port == 4
|
|
assert app.wsgi_app.x_prefix == 0
|
|
destroy_ctfd(app)
|
|
|
|
class ReverseProxyConfig(TestingConfig):
|
|
REVERSE_PROXY = "true"
|
|
|
|
app = create_ctfd(config=ReverseProxyConfig)
|
|
with app.app_context():
|
|
assert app.wsgi_app.x_for == 1
|
|
assert app.wsgi_app.x_proto == 1
|
|
assert app.wsgi_app.x_host == 1
|
|
assert app.wsgi_app.x_port == 1
|
|
assert app.wsgi_app.x_prefix == 1
|
|
destroy_ctfd(app)
|