tidaldb/tidal-net/tests/large_payload.rs
jx12n b55ad70141 fix: M0-M10 third-pass remediation — durability, replication, and CLI hardening
Resolves the 142 findings from tidal/docs/reviews/CODE_REVIEW_m0-m10.md across
the engine, server, net, and CLI surfaces:

- WAL/session-journal durability, checkpoint format, and crash-recovery hardening
- Replication shipper/receiver, tenant isolation, and migration paths
- Cluster scatter-gather, router, standalone server + health/offload endpoints
- tidalctl refactored into command modules with JSON output and WAL-state tooling
- Cohort, governance, signal-ledger, and vector-registry correctness fixes
- Expanded UAT/integration/durability test coverage across all milestones
2026-06-08 10:28:34 -06:00

97 lines
3.6 KiB
Rust

// Integration-test exemptions (same posture as the tidaldb integration tests):
// unwrap on known-good fixtures and short-lived read guards are idiomatic here.
#![allow(clippy::unwrap_used, clippy::significant_drop_tightening)]
//! Regression test for the tonic default 4 MiB codec message-size limit.
//!
//! tonic 0.12 defaults BOTH the encoder and decoder message-size limits to
//! 4 MiB. `GrpcTransportConfig::max_payload_bytes` defaults to 64 MiB and the
//! WAL default sealed segment is 16 MiB, so a full-size segment is 4x the codec
//! default. If the limits are not raised on both ends, the FIRST full-size
//! segment a leader ships fails inside the codec, maps to `TransportError::
//! Closed`, and the shipper retries the same seqno forever (silent replication
//! stall). This test ships a payload strictly between 4 MiB and the configured
//! max and asserts it round-trips byte-for-byte, so the regression can never
//! return undetected.
use std::{collections::HashMap, net::SocketAddr, thread, time::Duration};
use tidal_net::{GrpcTransport, config::GrpcTransportConfig};
use tidaldb::replication::{
WalSegmentId,
shard::{RegionId, ShardId},
transport::{Transport, WalSegmentPayload},
};
/// Get a unique listen address using port 0 (OS-assigned).
fn free_addr() -> SocketAddr {
let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap();
listener.local_addr().unwrap()
}
fn make_config(
shard: ShardId,
listen: SocketAddr,
peers: HashMap<ShardId, SocketAddr>,
) -> GrpcTransportConfig {
GrpcTransportConfig {
local_shard: shard,
listen_addr: listen,
peers,
insecure: true,
..Default::default()
}
}
#[test]
fn ships_payload_larger_than_tonic_default_codec_limit() {
// Sanity: the test is only meaningful if the payload exceeds tonic's 4 MiB
// default AND fits inside the configured ceiling. The payload-exceeds-default
// half is a compile-time invariant; the fits-in-ceiling half is runtime.
const FOUR_MIB: usize = 4 * 1024 * 1024;
const EIGHT_MIB: usize = 8 * 1024 * 1024;
const _: () = assert!(EIGHT_MIB > FOUR_MIB, "payload must exceed tonic default");
let addr0 = free_addr();
let addr1 = free_addr();
let config0 = make_config(ShardId(0), addr0, HashMap::from([(ShardId(1), addr1)]));
let config1 = make_config(ShardId(1), addr1, HashMap::from([(ShardId(0), addr0)]));
assert!(
EIGHT_MIB <= config0.max_payload_bytes,
"payload must fit configured max"
);
let t0 = GrpcTransport::new(config0).expect("transport 0");
let t1 = GrpcTransport::new(config1).expect("transport 1");
// Give the servers a moment to start.
thread::sleep(Duration::from_millis(150));
// A deterministic, verifiable 8 MiB body (a repeating byte pattern keyed on
// index so a silent truncation or corruption is caught, not just length).
let body: Vec<u8> = (0..EIGHT_MIB).map(|i| (i % 251) as u8).collect();
let payload = WalSegmentPayload {
id: WalSegmentId::new(RegionId::SINGLE, ShardId(0), 7),
bytes: body.clone(),
event_count: 3,
leader_last_seq: 7,
};
t0.send_segment(ShardId(1), payload)
.expect("8 MiB segment must ship past the raised codec limit");
let received = t1
.recv_segment()
.expect("follower must receive the 8 MiB segment");
assert_eq!(received.id.seqno, 7);
assert_eq!(received.event_count, 3);
assert_eq!(
received.bytes.len(),
EIGHT_MIB,
"body length must round-trip"
);
assert_eq!(received.bytes, body, "body bytes must round-trip exactly");
}