From 6eaed072c75920dba672347424c3ee7e37300383 Mon Sep 17 00:00:00 2001
From: bigcat88 <bigcat88@icloud.com>
Date: Wed, 10 Sep 2025 09:51:06 +0300
Subject: [PATCH] add some logic tests

---
 tests-assets/test_crud.py | 63 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 tests-assets/test_crud.py

diff --git a/tests-assets/test_crud.py b/tests-assets/test_crud.py
new file mode 100644
index 000000000..5e0d953d5
--- /dev/null
+++ b/tests-assets/test_crud.py
@@ -0,0 +1,63 @@
+import aiohttp
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_get_and_delete_asset(http: aiohttp.ClientSession, api_base: str, seeded_asset: dict):
+    aid = seeded_asset["id"]
+
+    # GET detail
+    async with http.get(f"{api_base}/api/assets/{aid}") as rg:
+        detail = await rg.json()
+        assert rg.status == 200, detail
+        assert detail["id"] == aid
+        assert "user_metadata" in detail
+        assert "filename" in detail["user_metadata"]
+
+    # DELETE
+    async with http.delete(f"{api_base}/api/assets/{aid}") as rd:
+        assert rd.status == 204
+
+    # GET again -> 404
+    async with http.get(f"{api_base}/api/assets/{aid}") as rg2:
+        body = await rg2.json()
+        assert rg2.status == 404
+        assert body["error"]["code"] == "ASSET_NOT_FOUND"
+
+
+@pytest.mark.asyncio
+async def test_update_asset_fields(http: aiohttp.ClientSession, api_base: str, seeded_asset: dict):
+    aid = seeded_asset["id"]
+
+    payload = {
+        "name": "unit_1_renamed.safetensors",
+        "tags": ["models", "checkpoints", "unit-tests", "beta"],
+        "user_metadata": {"purpose": "updated", "epoch": 2},
+    }
+    async with http.put(f"{api_base}/api/assets/{aid}", json=payload) as ru:
+        body = await ru.json()
+        assert ru.status == 200, body
+        assert body["name"] == payload["name"]
+        assert "beta" in body["tags"]
+        assert body["user_metadata"]["purpose"] == "updated"
+        # filename should still be present and normalized by server
+        assert "filename" in body["user_metadata"]
+
+
+@pytest.mark.asyncio
+async def test_head_asset_by_hash_and_invalids(http: aiohttp.ClientSession, api_base: str, seeded_asset: dict):
+    h = seeded_asset["asset_hash"]
+
+    # Existing
+    async with http.head(f"{api_base}/api/assets/hash/{h}") as rh1:
+        assert rh1.status == 200
+
+    # Non-existent
+    async with http.head(f"{api_base}/api/assets/hash/blake3:{'0'*64}") as rh2:
+        assert rh2.status == 404
+
+    # Invalid format
+    async with http.head(f"{api_base}/api/assets/hash/not_a_hash") as rh3:
+        jb = await rh3.json()
+        assert rh3.status == 400
+        assert jb is None  # HEAD request should not include "body" in response