tests for the new TimeoutError behavior

diff --git a/remoto/tests/test_log.py b/remoto/tests/test_log.py
index 5757da62fd2db55648277f4546e788fc0965503e..8ef23f04bc366b0df5b51eebeead171ca3fb5738 100644 (file)
--- a/remoto/tests/test_log.py
+++ b/remoto/tests/test_log.py
@@ -1,4 +1,5 @@
 from remoto import log
+from remoto.exc import TimeoutError
 from mock import Mock
 
 
@@ -34,3 +35,11 @@ class TestReporting(object):
         message = conn.logger.error.call_args[0][0]
         assert message == 'an error message'
 
+    def test_timeout_error(self):
+        conn = Mock()
+        result = Mock()
+        result.receive.side_effect = TimeoutError
+        log.reporting(conn, result)
+        message = conn.logger.warning.call_args[0][0]
+        assert 'No data was received after ' in message
+