Add image integration (#38969)

tests/components/image/test_init.py

@@ -0,0 +1,76 @@
+"""Test that we can upload images."""
+import pathlib
+import tempfile
+
+from aiohttp import ClientSession, ClientWebSocketResponse
+
+from homeassistant.components.websocket_api import const as ws_const
+from homeassistant.setup import async_setup_component
+from homeassistant.util import dt as util_dt
+
+from tests.async_mock import patch
+
+
+async def test_upload_image(hass, hass_client, hass_ws_client):
+    """Test we can upload an image."""
+    now = util_dt.utcnow()
+    test_image = pathlib.Path(__file__).parent / "logo.png"
+
+    with tempfile.TemporaryDirectory() as tempdir, patch.object(
+        hass.config, "path", return_value=tempdir
+    ), patch("homeassistant.util.dt.utcnow", return_value=now):
+        assert await async_setup_component(hass, "image", {})
+        ws_client: ClientWebSocketResponse = await hass_ws_client()
+        client: ClientSession = await hass_client()
+
+        with test_image.open("rb") as fp:
+            res = await client.post("/api/image/upload", data={"file": fp})
+
+        assert res.status == 200
+
+        item = await res.json()
+
+        assert item["content_type"] == "image/png"
+        assert item["filesize"] == 38847
+        assert item["name"] == "logo.png"
+        assert item["uploaded_at"] == now.isoformat()
+
+        tempdir = pathlib.Path(tempdir)
+        item_folder: pathlib.Path = tempdir / item["id"]
+        assert (item_folder / "original").read_bytes() == test_image.read_bytes()
+
+        # fetch non-existing image
+        res = await client.get("/api/image/serve/non-existing/256x256")
+        assert res.status == 404
+
+        # fetch invalid sizes
+        for inv_size in ("256", "256x25A", "100x100", "25Ax256"):
+            res = await client.get(f"/api/image/serve/{item['id']}/{inv_size}")
+            assert res.status == 400
+
+        # fetch resized version
+        res = await client.get(f"/api/image/serve/{item['id']}/256x256")
+        assert res.status == 200
+        assert (item_folder / "256x256").is_file()
+
+        # List item
+        await ws_client.send_json({"id": 6, "type": "image/list"})
+        msg = await ws_client.receive_json()
+
+        assert msg["id"] == 6
+        assert msg["type"] == ws_const.TYPE_RESULT
+        assert msg["success"]
+        assert msg["result"] == [item]
+
+        # Delete item
+        await ws_client.send_json(
+            {"id": 7, "type": "image/delete", "image_id": item["id"]}
+        )
+        msg = await ws_client.receive_json()
+
+        assert msg["id"] == 7
+        assert msg["type"] == ws_const.TYPE_RESULT
+        assert msg["success"]
+
+        # Ensure removed from disk
+        assert not item_folder.is_dir()