source: mod_gnutls/test/https-test-client.py

#!/usr/bin/python3
# PYTHON_ARGCOMPLETE_OK

# Copyright 2019 Fiona Klute
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import contextlib
import os
import yaml

from mgstest.tests import run_test_conf

if __name__ == "__main__":
    import argparse
    parser = argparse.ArgumentParser(
        description='Send HTTP requests through gnutls-cli',
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('host', nargs='?', default=None,
                        help='Access this host. Overrides TEST_TARGET, '
                        'but not the test configuration file.')
    parser.add_argument('-p', '--port', default=None,
                        help='Access this port. Overrides TEST_PORT, '
                        'but not the test configuration file.')
    parser.add_argument('--timeout', type=float,
                        help='Timeout for HTTP requests', default='5.0')
    parser.add_argument('--test-config', type=argparse.FileType('r'),
                        required=True, help='load YAML test configuration')

    # enable bash completion if argcomplete is available
    try:
        import argcomplete
        argcomplete.autocomplete(parser)
    except ImportError:
        pass

    args = parser.parse_args()

    if args.host:
        os.environ['TEST_TARGET'] = args.host
    if args.port:
        os.environ['TEST_PORT'] = args.port

    with contextlib.closing(args.test_config):
        test_conf = yaml.load(args.test_config, Loader=yaml.Loader)
        run_test_conf(test_conf, args.timeout)