Add file selector and file upload integration (#76672)

tests/components/file_upload/test_init.py

@@ -0,0 +1,66 @@
+"""Test the File Upload integration."""
+from pathlib import Path
+from random import getrandbits
+from unittest.mock import patch
+
+import pytest
+
+from homeassistant.components import file_upload
+from homeassistant.core import HomeAssistant
+from homeassistant.setup import async_setup_component
+
+from tests.components.image import TEST_IMAGE
+
+
+@pytest.fixture
+async def uploaded_file_dir(hass: HomeAssistant, hass_client) -> Path:
+    """Test uploading and using a file."""
+    assert await async_setup_component(hass, "file_upload", {})
+    client = await hass_client()
+
+    with patch(
+        # Patch temp dir name to avoid tests fail running in parallel
+        "homeassistant.components.file_upload.TEMP_DIR_NAME",
+        file_upload.TEMP_DIR_NAME + f"-{getrandbits(10):03x}",
+    ), TEST_IMAGE.open("rb") as fp:
+        res = await client.post("/api/file_upload", data={"file": fp})
+
+    assert res.status == 200
+    response = await res.json()
+
+    file_dir = hass.data[file_upload.DOMAIN].file_dir(response["file_id"])
+    assert file_dir.is_dir()
+    return file_dir
+
+
+async def test_using_file(hass: HomeAssistant, uploaded_file_dir):
+    """Test uploading and using a file."""
+    # Test we can use it
+    with file_upload.process_uploaded_file(hass, uploaded_file_dir.name) as file_path:
+        assert file_path.is_file()
+        assert file_path.parent == uploaded_file_dir
+        assert file_path.read_bytes() == TEST_IMAGE.read_bytes()
+
+    # Test it's removed
+    assert not uploaded_file_dir.exists()
+
+
+async def test_removing_file(hass: HomeAssistant, hass_client, uploaded_file_dir):
+    """Test uploading and using a file."""
+    client = await hass_client()
+
+    response = await client.delete(
+        "/api/file_upload", json={"file_id": uploaded_file_dir.name}
+    )
+    assert response.status == 200
+
+    # Test it's removed
+    assert not uploaded_file_dir.exists()
+
+
+async def test_removed_on_stop(hass: HomeAssistant, hass_client, uploaded_file_dir):
+    """Test uploading and using a file."""
+    await hass.async_stop()
+
+    # Test it's removed
+    assert not uploaded_file_dir.exists()