fix rate limits test

Author: arvidn

chia/_tests/core/server/test_rate_limits.py

@@ -24,7 +24,6 @@
 rl_v2 = [Capability.BASE, Capability.BLOCK_HEADERS, Capability.RATE_LIMITS_V2]
 rl_v1 = [Capability.BASE]
 node_with_params_b = node_with_params
-test_different_versions_results: list[int] = []
 
 
 @dataclass
@@ -418,11 +417,14 @@ async def test_different_versions(
     # The following code checks whether all of the runs resulted in the same number of items in "rate_limits_tx",
     # which would mean the same rate limits are always used. This should not happen, since two nodes with V2
     # will use V2.
-    total_tx_msg_count = len(get_rate_limits_to_use(a_con.local_capabilities, a_con.peer_capabilities))
+    rate_limits = get_rate_limits_to_use(a_con.local_capabilities, a_con.peer_capabilities)[0]
+    limit = rate_limits[ProtocolMessageTypes.request_header_blocks]
+    assert isinstance(limit, RLSettings)
 
-    test_different_versions_results.append(total_tx_msg_count)
-    if len(test_different_versions_results) >= 4:
-        assert len(set(test_different_versions_results)) >= 2
+    if Capability.RATE_LIMITS_V2 in a_con.local_capabilities and Capability.RATE_LIMITS_V2 in a_con.peer_capabilities:
+        assert limit.frequency == 5000
+    else:
+        assert limit.frequency == 500
 
 
 @pytest.mark.anyio