Make tests less flaky

zigpy · May 2, 2024 · 087d682 · 087d682
1 parent 20378f6
commit 087d682
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_ash.py b/tests/test_ash.py
@@ -106,8 +106,7 @@ def write(self, data: bytes) -> None:
  if random.random() < 0.20:
  return
 
- for byte in data:
- super().write(bytes([byte]))
+ super().write(data)
 
 
 class FakeTransportWithDelays(FakeTransport):
@@ -586,6 +585,9 @@ async def test_ash_end_to_end(transport_cls: type[FakeTransport]) -> None:
  await host.send_data(b"test")
 
  # Trigger a failure caused by excessive NAKs
+ ncp._t_rx_ack = ash.T_RX_ACK_INIT / 1000
+ host._t_rx_ack = ash.T_RX_ACK_INIT / 1000
+
  with patch.object(ncp, "nak_state", True):
  with pytest.raises(ash.NotAcked):
  await host.send_data(b"ncp NAKing until failure")