Fix xmlrpc implementation

Signed-off-by: Dan Ryan <dan@danryan.co> fix xmlrpc implementation Signed-off-by: Dan Ryan <dan@danryan.co>
pypa · Jul 6, 2019 · e559b00 · e559b00
1 parent 1630da8
commit e559b00
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/tests/pytest-pypi/pytest_pypi/app.py b/tests/pytest-pypi/pytest_pypi/app.py
@@ -10,13 +10,10 @@
 from zipfile import is_zipfile
 
 import requests
+from six.moves import xmlrpc_client
 
 from flask import Flask, redirect, abort, render_template, send_file, jsonify
 
-if sys.version_info[:2] >= (3, 0):
-    from xmlrpc.client import ServerProxy
-else:
-    from xmlrpclib import ServerProxy
 
 app = Flask(__name__)
 session = requests.Session()
@@ -27,12 +24,12 @@
 
 @contextlib.contextmanager
 def xml_pypi_server(server):
-    session = requests.Session()
-    client = ServerProxy(server, session)
+    transport = xmlrpc_client.Transport()
+    client = xmlrpc_client.ServerProxy(server, transport)
     try:
         yield client
     finally:
-        session.close()
+        transport.close()
 
 
 def get_pypi_package_names():