From 21a33e1f8c22e38a1cb5e48f102393490f718dad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebasti=C3=A1n=20Rajo?= <elecay@gmail.com>
Date: Fri, 6 Dec 2019 14:02:44 +0100
Subject: [PATCH] Fix pymongo version issue. Closes #45

---
 mongoaudit/testers/tls.py | 4 ++--
 requirements.txt          | 2 +-
 setup.py                  | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/mongoaudit/testers/tls.py b/mongoaudit/testers/tls.py
index 4de83a7..d1b4eb8 100644
--- a/mongoaudit/testers/tls.py
+++ b/mongoaudit/testers/tls.py
@@ -22,7 +22,7 @@ def enabled(test):
         return TestResult(success=True, severity=OMITTED)
 
     try:
-        with test.tester.conn._socket_for_writes() as socket_info:
+        with test.tester.conn._socket_for_writes(None) as socket_info:
             socket = socket_info.sock
             return TestResult(success=isinstance(socket, ssl.SSLSocket))
     except (KeyError, AttributeError):
@@ -36,7 +36,7 @@ def valid(test):
     if not enabled(test) is True:
         return TestResult(success=True, severity=OMITTED)
 
-    with test.tester.conn._socket_for_writes() as socket_info:
+    with test.tester.conn._socket_for_writes(None) as socket_info:
         cert = socket_info.sock.getpeercert()
         if not cert:
             return TestResult(success=True,
diff --git a/requirements.txt b/requirements.txt
index 722e679..382d78b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 pyinstaller==3.2.1
-pymongo==3.3.1
+pymongo==3.9.0
 pytest==3.0.6
 urwid==1.3.1
diff --git a/setup.py b/setup.py
index f5d0c41..3ba026c 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
         ]
     },
     install_requires=[
-        'pymongo>=3.3.1',
+        'pymongo>=3.9.0',
         'urwid>=1.3.1'
     ],
     classifiers=[