mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-21 19:07:46 +01:00
87144cd572
* Add hwaccel presets * Use hwaccel presets * Add input arg presets * Use input arg presets * Make util to clean up redundant code * Add support for output arg presets * Add tests * Update camera specific to use presets * Update hwaccel to use presets * Format files and fix tests * Rewrite tests to test record correctly * Move presets from string to list to avoid manually separating into a list * Add mjpeg cuvid decoder preset * Fix tests * Fix comment
153 lines
6.0 KiB
Python
153 lines
6.0 KiB
Python
import unittest
|
|
from frigate.config import FrigateConfig
|
|
from frigate.ffmpeg_presets import parse_preset_input
|
|
|
|
|
|
class TestFfmpegPresets(unittest.TestCase):
|
|
def setUp(self):
|
|
self.default_ffmpeg = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{
|
|
"path": "rtsp://10.0.0.1:554/video",
|
|
"roles": ["detect", "rtmp"],
|
|
}
|
|
],
|
|
"output_args": {
|
|
"detect": "-f rawvideo -pix_fmt yuv420p",
|
|
"record": "-f segment -segment_time 10 -segment_format mp4 -reset_timestamps 1 -strftime 1 -c copy -an",
|
|
"rtmp": "-c copy -f flv",
|
|
},
|
|
},
|
|
"detect": {
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"fps": 5,
|
|
},
|
|
"record": {
|
|
"enabled": True,
|
|
},
|
|
"rtmp": {
|
|
"enabled": True,
|
|
},
|
|
"name": "back",
|
|
}
|
|
},
|
|
}
|
|
|
|
def test_default_ffmpeg(self):
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert self.default_ffmpeg == frigate_config.dict(exclude_unset=True)
|
|
|
|
def test_ffmpeg_hwaccel_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"][
|
|
"hwaccel_args"
|
|
] = "preset-rpi-64-h264"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "preset-rpi-64-h264" not in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
assert "-c:v h264_v4l2m2m" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_hwaccel_not_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"][
|
|
"hwaccel_args"
|
|
] = "-other-hwaccel args"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "-other-hwaccel args" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_default_ffmpeg_input_arg_preset(self):
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"][
|
|
"input_args"
|
|
] = "preset-rtsp-generic"
|
|
frigate_preset_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
frigate_preset_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert (
|
|
frigate_preset_config.cameras["back"].ffmpeg_cmds[0]["cmd"]
|
|
== frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"]
|
|
)
|
|
|
|
def test_ffmpeg_input_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"][
|
|
"input_args"
|
|
] = "preset-rtmp-generic"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "preset-rtmp-generic" not in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
assert (" ".join(parse_preset_input("preset-rtmp-generic", 5))) in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_input_not_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"]["input_args"] = "-some inputs"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "-some inputs" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_output_record_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"]["output_args"][
|
|
"record"
|
|
] = "preset-record-generic-audio"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "preset-record-generic-audio" not in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
assert "-c:v copy -c:a aac" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_output_record_not_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"]["output_args"][
|
|
"record"
|
|
] = "-some output"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "-some output" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_output_rtmp_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"]["output_args"][
|
|
"rtmp"
|
|
] = "preset-rtmp-jpeg"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "preset-rtmp-jpeg" not in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
assert "-c:v libx264" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
def test_ffmpeg_output_rtmp_not_preset(self):
|
|
self.default_ffmpeg["cameras"]["back"]["ffmpeg"]["output_args"][
|
|
"rtmp"
|
|
] = "-some output"
|
|
frigate_config = FrigateConfig(**self.default_ffmpeg)
|
|
frigate_config.cameras["back"].create_ffmpeg_cmds()
|
|
assert "-some output" in (
|
|
" ".join(frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"])
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(verbosity=2)
|