From baf83437ea1fba3691ee4b8b1b4f4b5b21e1ff4f Mon Sep 17 00:00:00 2001
From: YS <ys@gm.com>
Date: Wed, 9 Feb 2022 14:53:29 +0300
Subject: [PATCH] fix unit tests

---
 frigate/test/test_config.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/frigate/test/test_config.py b/frigate/test/test_config.py
index 32c22f380..d4eb88642 100644
--- a/frigate/test/test_config.py
+++ b/frigate/test/test_config.py
@@ -1,4 +1,5 @@
 import unittest
+from unittest import mock
 import numpy as np
 from pydantic import ValidationError
 from frigate.config import (
@@ -1244,11 +1245,11 @@ class TestConfig(unittest.TestCase):
         runtime_config = frigate_config.runtime_config
         assert runtime_config.cameras["back"].snapshots.retain.default == 1.5
 
-    @unittest.mock.patch(
+    @mock.patch(
         "frigate.config.gst_discover",
         return_value={"video": "video/x-h265"},
     )
-    @unittest.mock.patch(
+    @mock.patch(
         "frigate.gstreamer.gst_inspect_find_codec",
         return_value=["nvv4l2decoder", "nvvidconv"],
     )
@@ -1299,11 +1300,11 @@ class TestConfig(unittest.TestCase):
         # custom rtspsrc arguments
         assert "protocols=tcp" in runtime_config.cameras["back"].decoder_cmds[0]["cmd"]
 
-    @unittest.mock.patch(
+    @mock.patch(
         "frigate.config.gst_discover",
         side_effect=Exception("should not call gst_discover"),
     )
-    @unittest.mock.patch(
+    @mock.patch(
         "frigate.gstreamer.gst_inspect_find_codec",
         return_value=["nvv4l2decoder", "nvvidconv"],
     )