mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-29 00:06:19 +01:00
434 lines
13 KiB
Python
434 lines
13 KiB
Python
import json
|
|
from unittest import TestCase, main
|
|
import voluptuous as vol
|
|
from frigate.config import FRIGATE_CONFIG_SCHEMA, FrigateConfig
|
|
|
|
class TestConfig(TestCase):
|
|
def setUp(self):
|
|
self.minimal = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
def test_empty(self):
|
|
FRIGATE_CONFIG_SCHEMA({})
|
|
|
|
def test_minimal(self):
|
|
FRIGATE_CONFIG_SCHEMA(self.minimal)
|
|
|
|
def test_config_class(self):
|
|
FrigateConfig(config=self.minimal)
|
|
|
|
def test_inherit_tracked_objects(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog']
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('dog' in frigate_config.cameras['back'].objects.track)
|
|
|
|
def test_override_tracked_objects(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog']
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'track': ['cat']
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('cat' in frigate_config.cameras['back'].objects.track)
|
|
|
|
def test_default_object_filters(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog']
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('dog' in frigate_config.cameras['back'].objects.filters)
|
|
|
|
def test_inherit_object_filters(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog'],
|
|
'filters': {
|
|
'dog': {
|
|
'threshold': 0.7
|
|
}
|
|
}
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('dog' in frigate_config.cameras['back'].objects.filters)
|
|
assert(frigate_config.cameras['back'].objects.filters['dog'].threshold == 0.7)
|
|
|
|
def test_override_object_filters(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'track': ['person', 'dog'],
|
|
'filters': {
|
|
'dog': {
|
|
'threshold': 0.7
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('dog' in frigate_config.cameras['back'].objects.filters)
|
|
assert(frigate_config.cameras['back'].objects.filters['dog'].threshold == 0.7)
|
|
|
|
def test_global_object_mask(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog']
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'mask': '0,0,1,1,0,1',
|
|
'filters': {
|
|
'dog': {
|
|
'mask': '1,1,1,1,1,1'
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('dog' in frigate_config.cameras['back'].objects.filters)
|
|
assert(len(frigate_config.cameras['back'].objects.filters['dog']._raw_mask) == 2)
|
|
assert(len(frigate_config.cameras['back'].objects.filters['person']._raw_mask) == 1)
|
|
|
|
def test_ffmpeg_params_global(self):
|
|
config = {
|
|
'ffmpeg': {
|
|
'input_args': ['-re']
|
|
},
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'track': ['person', 'dog'],
|
|
'filters': {
|
|
'dog': {
|
|
'threshold': 0.7
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('-re' in frigate_config.cameras['back'].ffmpeg_cmds[0]['cmd'])
|
|
|
|
def test_ffmpeg_params_camera(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
],
|
|
'input_args': ['-re']
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'track': ['person', 'dog'],
|
|
'filters': {
|
|
'dog': {
|
|
'threshold': 0.7
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('-re' in frigate_config.cameras['back'].ffmpeg_cmds[0]['cmd'])
|
|
|
|
def test_ffmpeg_params_input(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'], 'input_args': ['-re'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'objects': {
|
|
'track': ['person', 'dog'],
|
|
'filters': {
|
|
'dog': {
|
|
'threshold': 0.7
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert('-re' in frigate_config.cameras['back'].ffmpeg_cmds[0]['cmd'])
|
|
|
|
|
|
def test_inherit_clips_retention(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'clips': {
|
|
'retain': {
|
|
'default': 20,
|
|
'objects': {
|
|
'person': 30
|
|
}
|
|
}
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert(frigate_config.cameras['back'].clips.retain.objects['person'] == 30)
|
|
|
|
def test_roles_listed_twice_throws_error(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'clips': {
|
|
'retain': {
|
|
'default': 20,
|
|
'objects': {
|
|
'person': 30
|
|
}
|
|
}
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] },
|
|
{ 'path': 'rtsp://10.0.0.1:554/video2', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
self.assertRaises(vol.MultipleInvalid, lambda: FrigateConfig(config=config))
|
|
|
|
def test_zone_matching_camera_name_throws_error(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'clips': {
|
|
'retain': {
|
|
'default': 20,
|
|
'objects': {
|
|
'person': 30
|
|
}
|
|
}
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'zones': {
|
|
'back': {
|
|
'coordinates': '1,1,1,1,1,1'
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
self.assertRaises(vol.MultipleInvalid, lambda: FrigateConfig(config=config))
|
|
|
|
def test_clips_should_default_to_global_objects(self):
|
|
config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'clips': {
|
|
'retain': {
|
|
'default': 20,
|
|
'objects': {
|
|
'person': 30
|
|
}
|
|
}
|
|
},
|
|
'objects': {
|
|
'track': ['person', 'dog']
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920,
|
|
'clips': {
|
|
'enabled': True
|
|
}
|
|
}
|
|
}
|
|
}
|
|
config = FrigateConfig(config=config)
|
|
assert(config.cameras['back'].clips.objects is None)
|
|
|
|
def test_role_assigned_but_not_enabled(self):
|
|
json_config = {
|
|
'mqtt': {
|
|
'host': 'mqtt'
|
|
},
|
|
'cameras': {
|
|
'back': {
|
|
'ffmpeg': {
|
|
'inputs': [
|
|
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect', 'rtmp'] },
|
|
{ 'path': 'rtsp://10.0.0.1:554/record', 'roles': ['record'] }
|
|
]
|
|
},
|
|
'height': 1080,
|
|
'width': 1920
|
|
}
|
|
}
|
|
}
|
|
|
|
config = FrigateConfig(config=json_config)
|
|
ffmpeg_cmds = config.cameras['back'].ffmpeg_cmds
|
|
assert(len(ffmpeg_cmds) == 1)
|
|
assert(not 'clips' in ffmpeg_cmds[0]['roles'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(verbosity=2)
|