mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-09 22:13:43 +08:00
firestar5683
7b1cc1f8e2
Serve fresh camera snapshots with explicit image responses and expose live crop controls without placing file or camera startup work on the UI thread.
33 lines
1.6 KiB
Python
33 lines
1.6 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
|
import os
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from openpilot.starpilot.galaxy import camera_request
|
|
|
|
|
|
class TestCameraRequest(unittest.TestCase):
|
|
def test_concurrent_latest_expiry_and_finite_lease(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = Path(directory) / 'lease'
|
|
with ThreadPoolExecutor(max_workers=8) as pool:
|
|
list(pool.map(lambda n: camera_request.request(clock=lambda: 100 + n, path=path), range(128)))
|
|
self.assertEqual(int(path.read_text()), 227 + camera_request.TTL_NS)
|
|
self.assertTrue(camera_request.requested(clock=lambda: 228, path=path))
|
|
self.assertFalse(camera_request.requested(clock=lambda: 227 + camera_request.TTL_NS, path=path))
|
|
self.assertFalse(camera_request.requested(clock=lambda: 1, path=path))
|
|
|
|
def test_prefix_isolation_and_frame_receipt_aging(self):
|
|
with patch.dict(os.environ, OPENPILOT_PREFIX='camera-a'):
|
|
first = camera_request._path('lease')
|
|
with patch.dict(os.environ, OPENPILOT_PREFIX='camera-b'):
|
|
self.assertNotEqual(first, camera_request._path('lease'))
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = Path(directory) / 'frame'
|
|
camera_request.mark_frame(clock=lambda: 100, path=path)
|
|
self.assertTrue(camera_request.frame_available(clock=lambda: 101, path=path))
|
|
self.assertFalse(camera_request.frame_available(clock=lambda: 101 + camera_request.FRAME_TTL_NS, path=path))
|
|
self.assertFalse(camera_request.frame_available(clock=lambda: 99, path=path))
|