From 52d2c309f5ccc3af9bb6500a9deb3a0684c21378 Mon Sep 17 00:00:00 2001 From: Jake Date: Fri, 13 Dec 2024 11:19:52 -0500 Subject: [PATCH] Fix formatting --- frigate/config/camera/record.py | 1 + frigate/test/test_storage.py | 14 +++++++++----- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/frigate/config/camera/record.py b/frigate/config/camera/record.py index d984b6387..2be076f8e 100644 --- a/frigate/config/camera/record.py +++ b/frigate/config/camera/record.py @@ -32,6 +32,7 @@ class RecordRetainConfig(FrigateBaseModel): days: float = Field(default=0, title="Default retention period.") mode: RetainModeEnum = Field(default=RetainModeEnum.all, title="Retain mode.") + class RecordCleanupConfig(FrigateBaseModel): trigger_minutes: int = Field( default=60, diff --git a/frigate/test/test_storage.py b/frigate/test/test_storage.py index 55f5873c6..bdec99ca9 100644 --- a/frigate/test/test_storage.py +++ b/frigate/test/test_storage.py @@ -140,13 +140,15 @@ class TestHttp(unittest.TestCase): assert storage.camera_storage_stats == { "front_door": {"bandwidth": 0, "needs_refresh": True}, } - - @patch('frigate.storage.shutil.disk_usage') + + @patch("frigate.storage.shutil.disk_usage") def test_storage_cleanup(self, mock_disk_usage): """Ensure that all recordings are cleaned up when necessary.""" config = FrigateConfig(**self.minimal_config) storage = StorageMaintainer(config, MagicMock()) - mock_disk_usage.return_value = shutil._ntuple_diskusage(total=10000, used=5000, free=5000) + mock_disk_usage.return_value = shutil._ntuple_diskusage( + total=10000, used=5000, free=5000 + ) id = "123456.keep" time_keep = datetime.datetime.now().timestamp() @@ -213,12 +215,14 @@ class TestHttp(unittest.TestCase): Recordings.get(Recordings.id == rec_d2_id) Recordings.get(Recordings.id == rec_d3_id) - @patch('frigate.storage.shutil.disk_usage') + @patch("frigate.storage.shutil.disk_usage") def test_storage_cleanup_keeps_retained(self, mock_disk_usage): """Ensure that all recordings are cleaned up when necessary.""" config = FrigateConfig(**self.minimal_config) storage = StorageMaintainer(config, MagicMock()) - mock_disk_usage.return_value = shutil._ntuple_diskusage(total=10000, used=5000, free=5000) + mock_disk_usage.return_value = shutil._ntuple_diskusage( + total=10000, used=5000, free=5000 + ) id = "123456.keep" time_keep = datetime.datetime.now().timestamp()