实时通信的端到端测试与压测

import puppeteer from "puppeteer";

const browser = await puppeteer.launch({
  headless: "new",
  args: [
    "--use-fake-device-for-media-stream",
    "--use-fake-ui-for-media-stream",
    "--use-file-for-fake-video-capture=/tmp/fixtures/color-bars.y4m",
    "--use-file-for-fake-audio-capture=/tmp/fixtures/tone-1khz.wav",
    "--autoplay-policy=no-user-gesture-required", "--no-sandbox",
  ],
});

const page = await browser.newPage();
await page.goto("https://localhost:8443/test-room.html", { waitUntil: "domcontentloaded" });

ffmpeg -f lavfi -i testsrc2=size=1280x720:rate=30 -t 30 -pix_fmt yuv420p /tmp/fixtures/motion.y4m

ffmpeg -f lavfi -i "sine=frequency=1000:sample_rate=48000:duration=30" \
  -af "volume='if(lt(mod(t,4),2),0.5,0)':eval=frame" \
  -c:a pcm_s16le /tmp/fixtures/tone-gated.wav

async function waitForIceConnected(page, timeout = 15000) {
  await page.waitForFunction(() => window.__pc?.iceConnectionState === "connected",
    { timeout, polling: 100 });
}

async function waitForFirstFrame(page, timeout = 10000) {
  await page.waitForFunction(() => {
    const v = document.querySelector("#remote");
    return v && v.readyState >= 3 && v.videoWidth > 0;
  }, { timeout, polling: 100 });
}

async function assertHealthy(page, thresholds) {
  const stats = await page.evaluate(async () => {
    const report = await window.__pc.getStats();
    const out = { inbound: null, outbound: null, pair: null };
    report.forEach((r) => {
      if (r.type === "inbound-rtp" && r.kind === "video") out.inbound = r;
      if (r.type === "outbound-rtp" && r.kind === "video") out.outbound = r;
      if (r.type === "candidate-pair" && r.state === "succeeded") out.pair = r;
    });
    return {
      packetsReceived: out.inbound?.packetsReceived ?? 0,
      packetsLost: out.inbound?.packetsLost ?? 0,
      framesDecoded: out.inbound?.framesDecoded ?? 0,
      freezeCount: out.inbound?.freezeCount ?? 0,
      rtt: out.pair?.currentRoundTripTime ?? -1,
      qualityLimitation: out.outbound?.qualityLimitationReason ?? "unknown",
    };
  });

  const lossRate = stats.packetsReceived > 0
    ? stats.packetsLost / (stats.packetsReceived + stats.packetsLost) : 0;

  expect(stats.framesDecoded).toBeGreaterThan(thresholds.minFrames);
  expect(lossRate).toBeLessThan(thresholds.maxLossRate);
  expect(stats.rtt).toBeGreaterThan(0);
  expect(stats.rtt).toBeLessThan(thresholds.maxRtt);
  return stats;
}

async function loadTest({ rooms, peersPerRoom, browser }) {
  const handles = [];
  for (let r = 0; r < rooms; r++) {
    const roomId = `load-${r}`;
    for (let p = 0; p < peersPerRoom; p++) {
      const page = await browser.newPage();
      await page.goto(`${BASE}/room.html?room=${roomId}&role=peer${p}`);
      handles.push({ page, roomId, peer: p });
    }
  }
  await new Promise((res) => setTimeout(res, 60000));   // 等全部进入稳定态
  const samples = await Promise.all(handles.map((h) => collectStats(h.page)));
  return { handles, samples };
}

// 用 Pion 模拟大量只订阅的观众（省略信令细节）
func spawnViewer(sfuURL, roomID string, wg *sync.WaitGroup) {
	defer wg.Done()
	pc, err := webrtc.NewPeerConnection(webrtc.Configuration{
		ICEServers: []webrtc.ICEServer{{URLs: []string{"stun:stun.example.com:3478"}}},
	})
	if err != nil {
		log.Println("create pc failed:", err)
		return
	}
	defer pc.Close()

	// 只接收，不发送，模拟直播观众
	_, err = pc.AddTransceiverFromKind(webrtc.RTPCodecTypeVideo,
		webrtc.RTPTransceiverInit{Direction: webrtc.RTPTransceiverDirectionRecvonly})
	if err != nil {
		log.Println("add transceiver failed:", err)
		return
	}
	offer, _ := pc.CreateOffer(nil)
	_ = pc.SetLocalDescription(offer)
}

sudo tc qdisc add dev eth0 root netem loss 5% delay 20ms 10ms distribution normal rate 1mbit
tc qdisc show dev eth0        # 查看当前规则
sudo tc qdisc del dev eth0 root   # 清理

baseline:
  version: "2.14.0"
  two_party_call: { ttff_ms: 780, mos: 4.2, loss_rate: 0.004, freeze_rate: 0.002 }
  room_6_peers:   { ttff_ms: 1200, loss_rate: 0.011, cpu_percent: 42 }
  weak_network_5pct_loss: { mos: 3.6, freeze_rate: 0.031 }

function checkRegression(current, baseline, tolerance) {
  const failures = [];
  for (const [key, base] of Object.entries(baseline)) {
    const cur = current[key];
    if (cur === undefined) continue;
    const delta = (cur - base) / base;
    if (delta > tolerance[key]) failures.push({ metric: key, base, cur, delta: `${(delta * 100).toFixed(1)}%` });
  }
  return failures;
}

services:
  chrome-test:
    image: browserless/chrome:latest
    shm_size: 1gb
    environment:
      CHROME_FLAGS: "--use-fake-device-for-media-stream --use-fake-ui-for-media-stream --no-sandbox"
    volumes: ["./fixtures:/tmp/fixtures:ro"]

单台 SFU（8 核 / 1 Gbps 出口）：
  2 人房间    ：约 400 间（800 路下行）
  6 人房间    ：约 90 间（约 540 路下行，下行 810 Mbps）
  20 人房间   ：约 12 间（约 240 路下行，下行 720 Mbps）
  直播观众    ：约 330 路（1.5 Mbps 单路，50% 余量）