From 1b02792fcab2c2a5b0a2abddcc1d767effd81c84 Mon Sep 17 00:00:00 2001
From: j45856021-dev <259477747+j45856021-dev@users.noreply.github.com>
Date: Sat, 10 Oct 2026 20:15:22 -0700
Subject: [PATCH] test: cover blob upload chunk boundaries

Verify binary content, digest paths, and returned digests for both clients
at and beyond the 32 KiB read boundary, with HTTP 200 and 201 responses.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
---
 tests/test_client.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/tests/test_client.py b/tests/test_client.py
index c212a11d..fd6ed7af 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -4,10 +4,12 @@
 import os
 import re
 import tempfile
+from hashlib import sha256
 from pathlib import Path
 from typing import Any
 
 import pytest
+from httpx import MockTransport
 from httpx import Response as httpxResponse
 from pydantic import BaseModel
 from pytest_httpserver import HTTPServer, URIPattern
@@ -866,6 +868,41 @@ def test_client_create_blob_exists(httpserver: HTTPServer):
     assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'
 
 
+@pytest.fixture(params=[0, 1, 32 * 1024, 32 * 1024 + 1, 64 * 1024 + 1])
+def blob_file(request, tmp_path):
+  size = request.param
+  content = (bytes(range(256)) * (size // 256 + 1))[:size]
+  path = tmp_path / 'blob.bin'
+  path.write_bytes(content)
+  return path, content
+
+
+@pytest.mark.parametrize('client_type', [Client, AsyncClient])
+@pytest.mark.parametrize('status_code', [200, 201])
+async def test_create_blob_content(blob_file, client_type, status_code):
+  path, content = blob_file
+  digest = f'sha256:{sha256(content).hexdigest()}'
+  requests = []
+
+  def handle(request):
+    requests.append(request)
+    assert request.method == 'POST'
+    assert request.url.path == f'/api/blobs/{digest}'
+    assert request.content == content
+    return httpxResponse(status_code)
+
+  client = client_type(host='http://sdk.test:11434', transport=MockTransport(handle))
+  if isinstance(client, AsyncClient):
+    async with client:
+      response = await client.create_blob(path)
+  else:
+    with client:
+      response = client.create_blob(path)
+
+  assert response == digest
+  assert len(requests) == 1
+
+
 def test_client_delete(httpserver: HTTPServer):
   httpserver.expect_ordered_request(PrefixPattern('/api/delete'), method='DELETE').respond_with_response(Response(status=200))
   client = Client(httpserver.url_for('/api/delete'))