Merge branch 'feature/travis-container' of https://github.com/AlexanderS/bcfg2

1 files changed, 11 insertions, 3 deletions

diff --git a/testsuite/ext/ssl_protocols.py b/testsuite/ext/ssl_protocols.py
index 66068d2a9..f92e3e355 100644
--- a/testsuite/ext/ssl_protocols.py
+++ b/testsuite/ext/ssl_protocols.py
@@ -1,5 +1,5 @@
 try:
-    from logilab.astng import MANAGER, scoped_nodes, node_classes
+    from logilab.astng import MANAGER, builder, scoped_nodes, node_classes
     PYLINT=0
 except ImportError:
     from astroid import MANAGER, scoped_nodes, node_classes
@@ -8,10 +8,18 @@ except ImportError:
 def ssl_transform(module):
     if module.name == 'ssl':
         for proto in ('SSLv23', 'TLSv1'):
-            module.locals['PROTOCOL_%s' % proto] = [node_classes.Const()]
+            module.locals['PROTOCOL_%s' % proto] = [node_classes.Const(0)]
 
 def register(linter):
     if PYLINT == 0:
-        MANAGER.register_transformer(ssl_transform)
+        if hasattr(MANAGER, 'register_transformer'):
+            MANAGER.register_transformer(ssl_transform)
+        else:
+            safe = builder.ASTNGBuilder.string_build
+            def _string_build(self, data, modname='', path=None):
+                if modname == 'ssl':
+                    data += '\n\nPROTOCOL_SSLv23 = 0\nPROTOCOL_TLSv1 = 0'
+                return safe(self, data, modname, path)
+            builder.ASTNGBuilder.string_build = _string_build
     else:
         MANAGER.register_transform(scoped_nodes.Module, ssl_transform)