add mock imports to fakes

Signed-off-by: Alfredo Deza <alfredo.deza@inktank.com>

diff --git a/ceph_deploy/tests/fakes.py b/ceph_deploy/tests/fakes.py
index a96bcf7ac3f504a3d71e4a0eeeb2a432848729c5..486273b8f3f164767e4e351cac270494b5ccd2c0 100644 (file)
--- a/ceph_deploy/tests/fakes.py
+++ b/ceph_deploy/tests/fakes.py
@@ -1,5 +1,26 @@
+from mock import MagicMock
 
 
 def fake_getaddrinfo(*a, **kw):
     return_host = kw.get('return_host', 'host1')
     return [[0,0,0,0, return_host]]
+
+
+def mock_open(mock=None, data=None):
+    """
+    Fake the behavior of `open` when used as a context manager
+    """
+    if mock is None:
+        mock = MagicMock(spec=file)
+
+    handle = MagicMock(spec=file)
+    handle.write.return_value = None
+    if data is None:
+        handle.__enter__.return_value = handle
+    else:
+        handle.__enter__.return_value = data
+    mock.return_value = handle
+    return mock
+
+
+