Merge pull request #622 from jurijsteblaj/fix/wait-time-func

Use time_func for timeout in wait_for_publish

src/paho/mqtt/client.py

@@ -349,10 +349,10 @@ def wait_for_publish(self, timeout=None):
         elif self.rc > 0:
             raise RuntimeError('Message publish failed: %s' % (error_string(self.rc)))
 
-        timeout_time = None if timeout is None else time.time() + timeout
+        timeout_time = None if timeout is None else time_func() + timeout
         timeout_tenth = None if timeout is None else timeout / 10.
         def timed_out():
-            return False if timeout is None else time.time() > timeout_time
+            return False if timeout is None else time_func() > timeout_time
 
         with self._condition:
             while not self._published and not timed_out():