From 0dc4a74e7e081f02fd46c4e8b540291ecfac0e0b Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Wed, 15 Jun 2022 19:20:21 -0600
Subject: [PATCH] Test retain / un-retain

---
 frigate/test/test_http.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/frigate/test/test_http.py b/frigate/test/test_http.py
index 1cd8b311c..30c8b6307 100644
--- a/frigate/test/test_http.py
+++ b/frigate/test/test_http.py
@@ -189,6 +189,23 @@ class TestHttp(unittest.TestCase):
 
         assert not event
 
+    def test_event_retention(self):
+        app = create_app(FrigateConfig(**self.minimal_config), self.db, None, None, None)
+        id = "123456.random"
+
+        with app.test_client() as client:
+            _insert_mock_event(id)
+            client.post(f"/events/{id}/retain")
+            event = client.get(f"/events/{id}").json
+            assert event
+            assert event["id"] == id
+            assert event["retain_indefinitely"] == True
+            client.delete(f"/events/{id}/retain")
+            event = client.get(f"/events/{id}").json
+            assert event
+            assert event["id"] == id
+            assert event["retain_indefinitely"] == False
+
 def _insert_mock_event(id: str) -> Event:
     """Inserts a basic event model with a given id."""
     return Event.insert(