fix: 拒绝ACT episode编号冲突

xr_rm_teleop/test/test_act_episode_recorder.py

@@ -744,13 +744,7 @@ def _push_recording_frames(recorder, control_ns, frame_number):
     )
 
 
-@requires_h5py
-def test_end_to_end_fake_episode_saves_and_returns_idle(tmp_path):
-    recorder = _recorder_for_test(tmp_path)
-    recorder._handle_right_b(grip=False)
-
-    assert recorder.state is RecordingState.ARMED
-
+def _finish_fake_episode(recorder):
     start_ns = recorder._now_ns()
     frame_number = 1001
     for offset, seq in enumerate(range(100, 107)):
@@ -768,12 +762,39 @@ def test_end_to_end_fake_episode_saves_and_returns_idle(tmp_path):
     _push_recording_frames(recorder, final_ns, frame_number)
     recorder._on_control_sample(_control_message(108, final_ns, grip=False))
 
+
+@requires_h5py
+def test_end_to_end_fake_episode_saves_and_returns_idle(tmp_path):
+    recorder = _recorder_for_test(tmp_path)
+    recorder._handle_right_b(grip=False)
+
+    assert recorder.state is RecordingState.ARMED
+
+    _finish_fake_episode(recorder)
+
     assert (tmp_path / "tomato_pick" / "episode_0.hdf5").is_file()
     assert recorder.state is RecordingState.IDLE
     assert "SAVING" in recorder._status_pub.messages
     assert recorder._status_pub.messages[-2:] == ["SAVED", "IDLE"]
 
 
+@requires_h5py
+def test_final_publish_rejects_allocated_episode_number_conflict(tmp_path):
+    recorder = _recorder_for_test(tmp_path)
+    recorder._handle_right_b(grip=False)
+    existing = recorder._task_dir / "episode_0.hdf5"
+    existing.write_bytes(b"existing")
+
+    _finish_fake_episode(recorder)
+
+    assert existing.read_bytes() == b"existing"
+    assert not (recorder._task_dir / "episode_1.hdf5").exists()
+    rejected = list((recorder._task_dir / "rejected").glob("*.hdf5"))
+    assert len(rejected) == 1
+    with h5py.File(rejected[0], "r") as root:
+        assert root.attrs["reject_reason"] == "episode_number_conflict"
+
+
 @requires_h5py
 def test_discard_and_interrupt_only_process_current_partial(tmp_path):
     recorder = _recorder_for_test(tmp_path)