Don't fail tests if pycurl is not installed

jupyterhub · Dec 7, 2024 · 67606fd · 67606fd
1 parent 3703273
commit 67606fd
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/binderhub/tests/utils.py b/binderhub/tests/utils.py
@@ -7,12 +7,21 @@
 
 import requests
 from tornado import gen
-from tornado.curl_httpclient import CurlAsyncHTTPClient
 from tornado.httpclient import HTTPError, HTTPRequest, HTTPResponse
 from tornado.httputil import HTTPHeaders
 
+try:
+    from tornado.curl_httpclient import CurlAsyncHTTPClient
 
-class MockAsyncHTTPClient(CurlAsyncHTTPClient):
+    BASE_HTTP_CLIENT = CurlAsyncHTTPClient
+except ModuleNotFoundError:
+    # pycurl is not installed, use regular asynchttpclient
+    from tornado.httpclient import AsyncHTTPClient
+
+    BASE_HTTP_CLIENT = AsyncHTTPClient
+
+
+class MockAsyncHTTPClient(BASE_HTTP_CLIENT):
     mocks = {}
     records = {}