From a7d273f13f89170dc4bc8c314a544c1915b3bcf5 Mon Sep 17 00:00:00 2001
From: Sebastian Pipping <sebastian@pipping.org>
Date: Sat, 8 Feb 2020 00:58:09 +0100
Subject: [PATCH] tests: Prepare version-specific code for extension

---
 tests/test_servefile.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tests/test_servefile.py b/tests/test_servefile.py
index 4eb2976..e6f3bd0 100644
--- a/tests/test_servefile.py
+++ b/tests/test_servefile.py
@@ -4,16 +4,17 @@ import pytest
 import requests
 import socket
 import subprocess
+import sys
 import tarfile
 import time
 import urllib3
 
 # crudly written to learn more about pytest and to have a base for refactoring
 
-try:
-    ConnectionRefusedError
+
+if sys.version_info.major >= 3:
     connrefused_exc = ConnectionRefusedError
-except NameError:
+else:
     connrefused_exc = socket.error