|
1 | 1 | import pytest |
2 | | -from typing import Optional |
| 2 | +from typing import Optional, List |
3 | 3 | from unittest.mock import MagicMock |
4 | 4 |
|
5 | | -from pydantic import BaseModel, Field, PrivateAttr |
| 5 | +from pydantic import BaseModel, Field, PrivateAttr, RootModel |
6 | 6 |
|
7 | 7 | from codesphere.core.handler import _APIOperationExecutor, APIRequestHandler |
8 | 8 | from codesphere.core.operations import APIOperation, AsyncCallable |
@@ -147,3 +147,35 @@ class ModelWithClient(BaseModel): |
147 | 147 | instance = ModelWithClient(id=1) |
148 | 148 | handler._inject_client_into_model(instance) |
149 | 149 | assert instance._http_client is mock_executor._http_client |
| 150 | + |
| 151 | + @pytest.mark.asyncio |
| 152 | + async def test_inject_client_into_root_model_items( |
| 153 | + self, mock_executor, sample_operation |
| 154 | + ): |
| 155 | + """RootModel containers should have _http_client injected into each item in .root""" |
| 156 | + mock_client = MagicMock() |
| 157 | + mock_client.request = MagicMock() |
| 158 | + mock_executor._http_client = mock_client |
| 159 | + |
| 160 | + handler = APIRequestHandler( |
| 161 | + executor=mock_executor, |
| 162 | + operation=sample_operation, |
| 163 | + kwargs={}, |
| 164 | + ) |
| 165 | + |
| 166 | + class ItemWithClient(BaseModel): |
| 167 | + id: int |
| 168 | + _http_client: Optional[MagicMock] = PrivateAttr(default=None) |
| 169 | + |
| 170 | + class ResourceList(RootModel[List[ItemWithClient]]): |
| 171 | + _http_client: Optional[MagicMock] = PrivateAttr(default=None) |
| 172 | + |
| 173 | + item1 = ItemWithClient(id=1) |
| 174 | + item2 = ItemWithClient(id=2) |
| 175 | + resource_list = ResourceList(root=[item1, item2]) |
| 176 | + |
| 177 | + handler._inject_client_into_model(resource_list) |
| 178 | + |
| 179 | + assert resource_list._http_client is mock_executor._http_client |
| 180 | + for item in resource_list.root: |
| 181 | + assert item._http_client is mock_executor._http_client |
0 commit comments