def test_proxy_base_default_secure_require_certs(self):
""" If secure (default), will ask for the certfile and keyfile """
p = utils.proxy_parser_base(port=80)
# Since we are secure, we need --certfile and --keyfile
with self.assertRaises(SystemExit) as e:
args, left = p.parse_known_args()
self.assertEqual(e.exception.code, 2) # exit status must be 2
def test_proxy_base_non_secure_no_certfile(self):
""" If not using TLS, we don't suggest TLS related arguments. """
p = utils.proxy_parser_base(port=80, secure=False)
args, left = p.parse_known_args()
# The values for cerfile and keyfile must be empty
self.assertIsNone(args.certfile)
self.assertIsNone(args.keyfile)
def test_proxy_base_non_secure_no_certfile(self):
""" If not using TLS, we don't suggest TLS related arguments. """
p = utils.proxy_parser_base(port=80, secure=False)
args, left = p.parse_known_args()
# This would raise if certfile is not in Namespace
with self.assertRaises(AttributeError):
args.certfile
def parse_args(args=None):
parser = utils.proxy_parser_base(port=80, secure=False)
parser.add_argument(
"--trusted",
nargs="*",
default=["::1", "127.0.0.1"],
help="Trusted reverse proxy list separated by space %(default)s. \
If you do not want to add a trusted trusted reverse proxy, \
just specify this flag with empty parameters.",
)
return parser, parser.parse_args(args=args)
def parse_args(args=None):
parser = utils.proxy_parser_base(port=80, secure=False)
parser.add_argument(
'--trusted',
nargs='*',
default=['::1', '127.0.0.1'],
help='Trusted reverse proxy list separated by space %(default)s. \
If you do not want to add a trusted trusted reverse proxy, \
just specify this flag with empty parameters.',
)
return parser, parser.parse_args(args=args)
def test_proxy_base_parser_noargs(self):
""" We must provide a port parameter to proxy_parser_base. """
with self.assertRaises(TypeError):
utils.proxy_parser_base()
def parse_args():
parser = utils.proxy_parser_base(port=443, secure=True)
return parser.parse_args()
def test_proxy_base_default_secure(self):
""" If we don't specify `secure`, it will default to secure. """
p = utils.proxy_parser_base(port=80)
args, left = p.parse_known_args()
# This would raise if certfile is not in Namespace
args.certfile
def parse_args():
parser = utils.proxy_parser_base(port=80, secure=False)
return parser.parse_args()
