frigate/frigate/test/test_camera_pw_clean.py
2022-11-02 05:51:08 -06:00

25 lines
766 B
Python

"""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