Improve test for device retry and timeout

miio/tests/test_device.py

@@ -17,7 +17,14 @@ def test_get_properties_splitting(mocker, max_properties):
     if max_properties is None:
         max_properties = len(properties)
     assert send.call_count == math.ceil(len(properties) / max_properties)
+
+
+def test_default_timeout_and_retry(mocker):
+    send = mocker.patch("miio.miioprotocol.MiIOProtocol.send")
+    d = Device("127.0.0.1", "68ffffffffffffffffffffffffffffff")
     assert 5 == d._protocol._timeout
+    d.send(command="fake_command", parameters=[])
+    send.assert_called_with("fake_command", [], 3, extra_parameters=None)
 
 
 def test_timeout_retry(mocker):
@@ -29,6 +36,15 @@ def test_timeout_retry(mocker):
     d.send("fake_command", [])
     send.assert_called_with("fake_command", [], 3, extra_parameters=None)
 
+    class CustomDevice(Device):
+        retry_count = 5
+        timeout = 1
+
+    d2 = CustomDevice("127.0.0.1", "68ffffffffffffffffffffffffffffff")
+    assert 1 == d2._protocol._timeout
+    d2.send("fake_command", [])
+    send.assert_called_with("fake_command", [], 5, extra_parameters=None)
+
 
 def test_unavailable_device_info_raises(mocker):
     send = mocker.patch("miio.Device.send", side_effect=PayloadDecodeException)