mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-03-11 19:07:35 +03:00
aed3793ce1
Introduce a new RetainPolicyEnum with 'time' (default, existing behavior) and 'continuous_rollover' (fill disk, overwrite oldest) options. Add the retain_policy field to RecordConfig and include unit tests for config validation.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import unittest
|
|
|
|
from frigate.config import FrigateConfig
|
|
|
|
|
|
class TestRetainPolicyConfig(unittest.TestCase):
|
|
def setUp(self):
|
|
self.base_config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"front_door": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"detect": {"height": 1080, "width": 1920, "fps": 5},
|
|
}
|
|
},
|
|
}
|
|
|
|
def test_default_retain_policy_is_time(self):
|
|
config = FrigateConfig(**self.base_config)
|
|
assert config.record.retain_policy.value == "time"
|
|
|
|
def test_continuous_rollover_policy(self):
|
|
self.base_config["record"] = {
|
|
"enabled": True,
|
|
"retain_policy": "continuous_rollover",
|
|
}
|
|
config = FrigateConfig(**self.base_config)
|
|
assert config.record.retain_policy.value == "continuous_rollover"
|
|
|
|
def test_continuous_rollover_ignores_continuous_days(self):
|
|
self.base_config["record"] = {
|
|
"enabled": True,
|
|
"retain_policy": "continuous_rollover",
|
|
"continuous": {"days": 30},
|
|
}
|
|
config = FrigateConfig(**self.base_config)
|
|
assert config.record.retain_policy.value == "continuous_rollover"
|
|
assert config.record.continuous.days == 30
|
|
|
|
def test_invalid_retain_policy_rejected(self):
|
|
self.base_config["record"] = {
|
|
"retain_policy": "invalid_value",
|
|
}
|
|
with self.assertRaises(Exception):
|
|
FrigateConfig(**self.base_config)
|