mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
b6c81875b0
Own Connect credentials, registration and upload preferences through shared access providers. Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com>
81 lines
3.8 KiB
Python
81 lines
3.8 KiB
Python
"""Actual cloud consumers exclude recordings owned by another provider."""
|
|
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from queue import Queue
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from openpilot.starpilot.connect import provider
|
|
from openpilot.system.athena import athenad
|
|
from openpilot.system.loggerd import uploader
|
|
|
|
|
|
class TestCloudConsumers(unittest.TestCase):
|
|
def setUp(self):
|
|
temporary = tempfile.TemporaryDirectory()
|
|
self.addCleanup(temporary.cleanup)
|
|
self.root = Path(temporary.name)
|
|
self.enterContext(patch.object(provider, 'active_provider', return_value=provider.PROVIDERS['konik']))
|
|
self.enterContext(patch.object(athenad.Paths, 'log_root', return_value=str(self.root)))
|
|
self.enterContext(patch.object(athenad, 'CLOUD_PROVIDER', provider.PROVIDERS['konik']))
|
|
self.konik = self.root/'aaaaaaaa--bbbbbbbbbb--0'
|
|
self.comma = self.root/'cccccccc--dddddddddd--0'
|
|
for directory in (self.konik,self.comma):
|
|
directory.mkdir()
|
|
(directory/'qlog.zst').write_bytes(b'log')
|
|
(directory/'fcamera.hevc').write_bytes(b'camera')
|
|
provider.mark_recording(self.konik,'konik')
|
|
|
|
def test_actual_uploader_and_athena_listing_keep_legacy_comma_out(self):
|
|
with patch.object(uploader, 'Api', return_value=SimpleNamespace()):
|
|
instance = uploader.Uploader('0'*16, str(self.root))
|
|
with patch.object(uploader,'getxattr',return_value=None):
|
|
files = list(instance.list_upload_files(False))
|
|
self.assertTrue(files)
|
|
self.assertTrue(all(Path(path).parent == self.konik for _,_,path in files))
|
|
listed = athenad.scan_dir(str(self.root),'')
|
|
self.assertIn(self.konik.name+'/qlog.zst',listed)
|
|
self.assertNotIn(self.comma.name+'/qlog.zst',listed)
|
|
|
|
def test_actual_athena_upload_queue_rejects_foreign_source(self):
|
|
with (patch.object(athenad,'upload_queue',Queue()),
|
|
patch.object(athenad.UploadQueueCache,'cache'),
|
|
patch.object(athenad,'listUploadQueue',return_value=[])):
|
|
result=athenad.uploadFilesToUrls([
|
|
{'fn':self.comma.name+'/qlog.zst','url':'https://upload.invalid/comma','headers':{}},
|
|
{'fn':self.konik.name+'/qlog.zst','url':'https://upload.invalid/konik','headers':{}},
|
|
])
|
|
self.assertEqual(result['enqueued'],1)
|
|
self.assertEqual(result['failed'],[self.comma.name+'/qlog.zst'])
|
|
|
|
def clips(self):
|
|
with patch.object(athenad.threading.Thread,'start'):
|
|
return athenad.VideoClips()
|
|
|
|
def test_clip_source_and_derived_directory_are_provider_owned(self):
|
|
clips=self.clips()
|
|
self.assertEqual(Path(clips.clip_path),self.root/'clips-konik')
|
|
self.assertEqual(clips._available_ranges('cccccccc--dddddddddd'),{})
|
|
own=clips._source_inputs('aaaaaaaa--bbbbbbbbbb','fcamera.hevc',0,1)
|
|
self.assertEqual(own,[str(self.konik/'fcamera.hevc')])
|
|
with self.assertRaises(ValueError):
|
|
clips._source_inputs('cccccccc--dddddddddd','fcamera.hevc',0,1)
|
|
provider.mark_recording(clips.clip_path,'konik')
|
|
(Path(clips.clip_path)/'test.mp4').write_bytes(b'video')
|
|
self.assertEqual(clips.getClipChunk('test.mp4',0)['size'],5)
|
|
with patch.object(provider,'active_provider',return_value=provider.PROVIDERS['comma']):
|
|
self.assertFalse(provider.owns_recording(Path(clips.clip_path)/'test.mp4',self.root))
|
|
|
|
def test_existing_comma_clips_keep_legacy_path_and_foreign_marker_denied(self):
|
|
with patch.object(athenad,'CLOUD_PROVIDER',provider.PROVIDERS['comma']):
|
|
clips=self.clips()
|
|
self.assertEqual(Path(clips.clip_path),self.root/'clips')
|
|
self.assertEqual(clips._source_inputs('cccccccc--dddddddddd','fcamera.hevc',0,1),
|
|
[str(self.comma/'fcamera.hevc')])
|
|
directory=self.root/'clips-konik'
|
|
provider.mark_recording(directory,'konik')
|
|
with self.assertRaises(ValueError):
|
|
provider.mark_recording(directory,'comma')
|