mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 14:04:20 +01:00
6833378c36
* Format Javascript and CSS files with `prettier`: `prettier --write 'CTFd/themes/**/*'` * Format Python with `black`: `black CTFd` & `black tests` * Travis now uses xenial instead of trusty.
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)
|