1 files changed, 17 insertions, 0 deletions
diff --git a/testsuite/ext/ssl_protocols.py b/testsuite/ext/ssl_protocols.py
new file mode 100644
index 000000000..66068d2a9
--- /dev/null
+++ b/testsuite/ext/ssl_protocols.py
@@ -0,0 +1,17 @@
+try:
+    from logilab.astng import MANAGER, scoped_nodes, node_classes
+    PYLINT=0
+except ImportError:
+    from astroid import MANAGER, scoped_nodes, node_classes
+    PYLINT=1
+
+def ssl_transform(module):
+    if module.name == 'ssl':
+        for proto in ('SSLv23', 'TLSv1'):
+            module.locals['PROTOCOL_%s' % proto] = [node_classes.Const()]
+
+def register(linter):
+    if PYLINT == 0:
+        MANAGER.register_transformer(ssl_transform)
+    else:
+        MANAGER.register_transform(scoped_nodes.Module, ssl_transform)