Pass width and height when requesting camera snapshot (#53835)

tests/components/camera/test_img_util.py

@@ -0,0 +1,72 @@
+"""Test img_util module."""
+from unittest.mock import patch
+
+from turbojpeg import TurboJPEG
+
+from homeassistant.components.camera import Image
+from homeassistant.components.camera.img_util import (
+    TurboJPEGSingleton,
+    scale_jpeg_camera_image,
+)
+
+from .common import EMPTY_8_6_JPEG, mock_turbo_jpeg
+
+EMPTY_16_12_JPEG = b"empty_16_12"
+
+
+def _clear_turbojpeg_singleton():
+    TurboJPEGSingleton.__instance = None
+
+
+def _reset_turbojpeg_singleton():
+    TurboJPEGSingleton.__instance = TurboJPEG()
+
+
+def test_turbojpeg_singleton():
+    """Verify the instance always gives back the same."""
+    _clear_turbojpeg_singleton()
+    assert TurboJPEGSingleton.instance() == TurboJPEGSingleton.instance()
+
+
+def test_scale_jpeg_camera_image():
+    """Test we can scale a jpeg image."""
+    _clear_turbojpeg_singleton()
+
+    camera_image = Image("image/jpeg", EMPTY_16_12_JPEG)
+
+    turbo_jpeg = mock_turbo_jpeg(first_width=16, first_height=12)
+    with patch("turbojpeg.TurboJPEG", return_value=False):
+        TurboJPEGSingleton()
+        assert scale_jpeg_camera_image(camera_image, 16, 12) == camera_image.content
+
+    turbo_jpeg = mock_turbo_jpeg(first_width=16, first_height=12)
+    turbo_jpeg.decode_header.side_effect = OSError
+    with patch("turbojpeg.TurboJPEG", return_value=turbo_jpeg):
+        TurboJPEGSingleton()
+        assert scale_jpeg_camera_image(camera_image, 16, 12) == camera_image.content
+
+    turbo_jpeg = mock_turbo_jpeg(first_width=16, first_height=12)
+    with patch("turbojpeg.TurboJPEG", return_value=turbo_jpeg):
+        TurboJPEGSingleton()
+        assert scale_jpeg_camera_image(camera_image, 16, 12) == EMPTY_16_12_JPEG
+
+    turbo_jpeg = mock_turbo_jpeg(
+        first_width=16, first_height=12, second_width=8, second_height=6
+    )
+    with patch("turbojpeg.TurboJPEG", return_value=turbo_jpeg):
+        TurboJPEGSingleton()
+        jpeg_bytes = scale_jpeg_camera_image(camera_image, 8, 6)
+
+    assert jpeg_bytes == EMPTY_8_6_JPEG
+
+
+def test_turbojpeg_load_failure():
+    """Handle libjpegturbo not being installed."""
+    _clear_turbojpeg_singleton()
+    with patch("turbojpeg.TurboJPEG", side_effect=Exception):
+        TurboJPEGSingleton()
+        assert TurboJPEGSingleton.instance() is False
+
+    _clear_turbojpeg_singleton()
+    TurboJPEGSingleton()
+    assert TurboJPEGSingleton.instance() is not None