administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 01:35:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Test setting and deleting sub label

Browse Source
This commit is contained in:
Nick Mowen 2022-06-15 19:27:32 -06:00
parent 0dc4a74e7e
commit 93b3eab1a1
1 changed files with 40 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
frigate/test/test_http.py
View File

@ -1,8 +1,10 @@
import datetime import datetime
import json
import logging import logging
import os import os
import unittest import unittest
from unittest.mock import patch, MagicMock from unittest.mock import patch, MagicMock
from flask import jsonify
from peewee_migrate import Router from peewee_migrate import Router
from playhouse.sqlite_ext import SqliteExtDatabase from playhouse.sqlite_ext import SqliteExtDatabase
@ -17,7 +19,6 @@ from frigate.test.const import TEST_DB
class TestHttp(unittest.TestCase): class TestHttp(unittest.TestCase):
def setUp(self): def setUp(self):
migrate_db = SqliteExtDatabase("test.db") migrate_db = SqliteExtDatabase("test.db")
del logging.getLogger("peewee_migrate").handlers[:] del logging.getLogger("peewee_migrate").handlers[:]
@ -167,7 +168,9 @@ class TestHttp(unittest.TestCase):
pass pass
def test_get_good_event(self): def test_get_good_event(self):
app = create_app(FrigateConfig(**self.minimal_config), self.db, None, None, None) app = create_app(
FrigateConfig(**self.minimal_config), self.db, None, None, None
)
id = "123456.random" id = "123456.random"
with app.test_client() as client: with app.test_client() as client:
@ -179,7 +182,9 @@ class TestHttp(unittest.TestCase):
assert event == model_to_dict(Event.get(Event.id == id)) assert event == model_to_dict(Event.get(Event.id == id))
def test_get_bad_event(self): def test_get_bad_event(self):
app = create_app(FrigateConfig(**self.minimal_config), self.db, None, None, None) app = create_app(
FrigateConfig(**self.minimal_config), self.db, None, None, None
)
id = "123456.random" id = "123456.random"
bad_id = "654321.other" bad_id = "654321.other"
@ -190,7 +195,9 @@ class TestHttp(unittest.TestCase):
assert not event assert not event
def test_event_retention(self): def test_event_retention(self):
app = create_app(FrigateConfig(**self.minimal_config), self.db, None, None, None) app = create_app(
FrigateConfig(**self.minimal_config), self.db, None, None, None
)
id = "123456.random" id = "123456.random"
with app.test_client() as client: with app.test_client() as client:
@ -206,6 +213,35 @@ class TestHttp(unittest.TestCase):
assert event["id"] == id assert event["id"] == id
assert event["retain_indefinitely"] == False assert event["retain_indefinitely"] == False
def test_set_delete_sub_label(self):
app = create_app(
FrigateConfig(**self.minimal_config), self.db, None, None, None
)
id = "123456.random"
sub_label = "sub"
with app.test_client() as client:
_insert_mock_event(id)
client.post(
f"/events/{id}/sub_label",
data=json.dumps({"subLabel": sub_label}),
content_type="application/json",
)
event = client.get(f"/events/{id}").json
assert event
assert event["id"] == id
assert event["sub_label"] == sub_label
client.post(
f"/events/{id}/sub_label",
data=json.dumps({"subLabel": ""}),
content_type="application/json",
)
event = client.get(f"/events/{id}").json
assert event
assert event["id"] == id
assert event["sub_label"] == ""
def _insert_mock_event(id: str) -> Event: def _insert_mock_event(id: str) -> Event:
"""Inserts a basic event model with a given id.""" """Inserts a basic event model with a given id."""
return Event.insert( return Event.insert(