From 5d3aadadf1a39dd788a9c70f229ec8963fb0ad5a Mon Sep 17 00:00:00 2001
From: James Hush <james@daily.co>
Date: Thu, 5 Feb 2026 10:12:45 +0800
Subject: [PATCH] Improve timestamp test to check frame pts difference

---
 tests/test_smallwebrtc_transport.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/tests/test_smallwebrtc_transport.py b/tests/test_smallwebrtc_transport.py
index 1712ea1af..ad21e6ec1 100644
--- a/tests/test_smallwebrtc_transport.py
+++ b/tests/test_smallwebrtc_transport.py
@@ -115,15 +115,16 @@ class TestRawAudioTrack:
         num_chunks = 4  # 40ms
         track = RawAudioTrack(sample_rate=sample_rate, num_10ms_chunks=num_chunks)
 
-        # Initial timestamp should be 0
-        assert track._timestamp == 0
+        # Receive first frame and check its timestamp
+        frame1 = await track.recv()
+        first_pts = frame1.pts
 
-        # Receive one frame (silence is fine)
-        await track.recv()
+        # Receive second frame
+        frame2 = await track.recv()
 
-        # Timestamp should advance by samples_per_chunk
+        # Timestamp should advance by samples_per_chunk between frames
         expected_samples = int(sample_rate * 40 / 1000)  # 640 samples
-        assert track._timestamp == expected_samples
+        assert frame2.pts - first_pts == expected_samples
 
     def test_different_sample_rates(self):
         """Test chunk size calculation at different sample rates."""