Add image test cases to generic (#69040)

home-assistant · Apr 2, 2022 · d4a31b0 · d4a31b0
1 parent 5db1c67
commit d4a31b0
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 0 deletions.
diff --git a/tests/components/generic/sample1_animate.png b/tests/components/generic/sample1_animate.png
diff --git a/tests/components/generic/sample2_jpeg_odd_header.jpg b/tests/components/generic/sample2_jpeg_odd_header.jpg
diff --git a/tests/components/generic/sample3_jpeg_odd_header.jpg b/tests/components/generic/sample3_jpeg_odd_header.jpg
diff --git a/tests/components/generic/sample4_K5-60mileAnim-320x240.gif b/tests/components/generic/sample4_K5-60mileAnim-320x240.gif
diff --git a/tests/components/generic/test_config_flow.py b/tests/components/generic/test_config_flow.py
@@ -1,6 +1,7 @@
 """Test The generic (IP Camera) config flow."""
 
 import errno
+import os.path
 from unittest.mock import patch
 
 import av
@@ -137,6 +138,30 @@ async def test_form_only_svg_whitespace(hass, fakeimgbytes_svg, user_flow):
  assert result2["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
 
 
+@respx.mock
+@pytest.mark.parametrize(
+ "image_file",
+ [
+ ("sample1_animate.png"),
+ ("sample2_jpeg_odd_header.jpg"),
+ ("sample3_jpeg_odd_header.jpg"),
+ ("sample4_K5-60mileAnim-320x240.gif"),
+ ],
+)
+async def test_form_only_still_sample(hass, user_flow, image_file):
+ """Test various sample images #69037."""
+ image_path = os.path.join(os.path.dirname(__file__), image_file)
+ with open(image_path, "rb") as image:
+ respx.get("http://127.0.0.1/testurl/1").respond(stream=image.read())
+ data = TESTDATA.copy()
+ data.pop(CONF_STREAM_SOURCE)
+ result2 = await hass.config_entries.flow.async_configure(
+ user_flow["flow_id"],
+ data,
+ )
+ assert result2["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
+
+
 @respx.mock
 async def test_form_rtsp_mode(hass, fakeimg_png, mock_av_open, user_flow):
  """Test we complete ok if the user enters a stream url."""