1 files changed, 18 insertions, 2 deletions
diff --git a/ipatests/test_integration/host.py b/ipatests/test_integration/host.py
index b4c7ffd6..1614b5fd 100644
--- a/ipatests/test_integration/host.py
+++ b/ipatests/test_integration/host.py
@@ -23,6 +23,7 @@ import os
 import socket
 import threading
 import subprocess
+from contextlib import contextmanager
 import errno
 
 import paramiko
@@ -118,6 +119,21 @@ class RemoteCommand(object):
         return thread
 
 
+@contextmanager
+def sftp_open(sftp, filename, mode='r'):
+    """Context manager that provides a file-like object over a SFTP channel
+
+    This provides compatibility with older Paramiko versions.
+    (In Paramiko 1.10+, file objects from `sftp.open` are directly usable as
+    context managers).
+    """
+    file = sftp.open(filename, mode)
+    try:
+        yield file
+    finally:
+        file.close()
+
+
 class Host(object):
     """Representation of a remote IPA host"""
     def __init__(self, domain, hostname, role, index, ip=None):
@@ -314,13 +330,13 @@ class Host(object):
     def get_file_contents(self, filename):
         """Read the named remote file and return the contents as a string"""
         self.log.debug('READ %s', filename)
-        with self.sftp.open(filename) as f:
+        with sftp_open(self.sftp, filename) as f:
             return f.read()
 
     def put_file_contents(self, filename, contents):
         """Write the given string to the named remote file"""
         self.log.info('WRITE %s', filename)
-        with self.sftp.open(filename, 'w') as f:
+        with sftp_open(self.sftp, filename, 'w') as f:
             f.write(contents)
 
     def file_exists(self, filename):