diff --git a/tests/integration/configuration/test_tls_cipher_proxy.py b/tests/integration/configuration/test_tls_cipher_proxy.py index e44b7f3d43a..b261af86207 100644 --- a/tests/integration/configuration/test_tls_cipher_proxy.py +++ b/tests/integration/configuration/test_tls_cipher_proxy.py @@ -401,17 +401,21 @@ def test_proxy_burst_survives_peer_restart_with_every_outcome_recorded(cipher_pr assert not any(outcome.status == 0 for outcome in finished), finished records: Final = tls_peers.gcm.received()[before:] assert all(record.cipher == PFS_GCM for record in records), records + submitted_users: Final = {outcome.user for outcome in finished if outcome.user is not None} successful_users: Final = { outcome.user for outcome in finished if outcome.status == 200 and outcome.user is not None } - failed_users: Final = {outcome.user for outcome in finished if outcome.status != 200 and outcome.user is not None} assert all(any(record.user == user for record in records) for user in successful_users), (successful_users, records) - assert all(not any(record.user == user for record in records) for user in failed_users), (failed_users, records) recorded_users: Final = {record.user for record in records if record.user is not None} - assert recorded_users <= successful_users, (recorded_users - successful_users, records) + assert recorded_users <= submitted_users, (recorded_users - submitted_users, records) + assert all(sum(1 for record in records if record.user == user) <= 3 for user in submitted_users), ( + submitted_users, + records, + ) anonymous: Final = sum(1 for record in records if record.user is None) + messages_outcomes: Final = sum(1 for outcome in finished if outcome.user is None) messages_ok: Final = sum(1 for outcome in finished if outcome.user is None and outcome.status == 200) - assert anonymous == messages_ok, (anonymous, messages_ok, records) + assert messages_ok <= anonymous <= messages_outcomes * 3, (anonymous, messages_ok, records) after_id: Final = _sync_completion(cipher_proxy, "tls-peer", f"chaos-after-{marker}") assert after_id == f"chatcmpl-chaos-after-{marker}" last_record: Final = tls_peers.gcm.received()[-1]