mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
264bb5a8a8
Serve fresh camera snapshots with explicit image responses and expose live crop controls without placing file or camera startup work on the UI thread. Co-authored-by: Prabhaav Pillai <prabhaav@msu.edu> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com>
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))
|