diff --git a/frigate/test/test_camera_pw_clean.py b/frigate/test/test_camera_pw_clean.py new file mode 100644 index 000000000..84df96e58 --- /dev/null +++ b/frigate/test/test_camera_pw_clean.py @@ -0,0 +1,24 @@ +"""Test camera user and password cleanup.""" + +import unittest + +from frigate.util import clean_camera_user_pass + + +class TestUserPassCleanup(unittest.TestCase): + + def setUp(self) -> None: + self.rtsp_with_pass = "rtsp://user:password@192.168.0.2:554/live" + self.rtsp_no_pass = "rtsp://192.168.0.3/live" + + + def test_cleanup(self): + """Test that user / pass are cleaned up.""" + clean = clean_camera_user_pass(self.rtsp_with_pass) + assert clean != self.rtsp_with_pass + assert "user:password" not in self.rtsp_with_pass + + def test_no_cleanup(self): + """Test that nothing changes when no user / pass are defined.""" + clean = clean_camera_user_pass(self.rtsp_no_pass) + assert clean == self.rtsp_no_pass