mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-11 03:38:38 +00:00
test(integration): bound chaos peer records instead of assuming no lost responses
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
4ca2493f73
commit
495418966a
1 changed files with 8 additions and 4 deletions
|
|
@ -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]
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue