Function create_test_server

tests/test_kaggle_module_resolver.py:18–37

(handler_class)

@contextmanager
def create_test_server(handler_class):
    hostname = 'localhost'
    port = 8080
    addr = f"http://{hostname}:{port}"

    # Simulates we are inside a Kaggle environment.
    env = EnvironmentVarGuard()
    env.set('KAGGLE_KERNEL_RUN_TYPE', 'Interactive')
    env.set('KAGGLE_USER_SECRETS_TOKEN', 'foo jwt token')
    env.set('KAGGLE_DATA_PROXY_TOKEN', 'foo proxy token')
    env.set('KAGGLE_DATA_PROXY_URL', addr)

    with env:
        with HTTPServer((hostname, port), handler_class) as test_server:
            threading.Thread(target=test_server.serve_forever).start()

            try:
                yield addr
            finally:
                test_server.shutdown()

class HubHTTPHandler(BaseHTTPRequestHandler):
    def do_GET(self):