test: format test suite

2026-06-21 13:04:20 +00:00 · 2026-06-16 11:13:05 +08:00
parent 1ae5aacc00
commit ff0c5a6f0a
92 changed files with 1658 additions and 1713 deletions
@@ -167,6 +167,7 @@ class TestLLMModelsServiceGetLLMModels:
        mock_provider_result = _create_mock_result([])

        call_count = 0
+
        async def mock_execute(query):
            return mock_result if call_count == 0 else mock_provider_result

@@ -200,6 +201,7 @@ class TestLLMModelsServiceGetLLMModels:
        mock_provider_result = _create_mock_result([provider])

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -239,6 +241,7 @@ class TestLLMModelsServiceGetLLMModels:
        mock_provider_result = _create_mock_result([provider])

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -279,6 +282,7 @@ class TestLLMModelsServiceGetLLMModel:
        mock_provider_result = _create_mock_result([], first_item=provider)

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -337,9 +341,7 @@ class TestLLMModelsServiceGetLLMModelsByProvider:

        mock_result = _create_mock_result([model1, model2])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)
-        ap.persistence_mgr.serialize_model = Mock(
-            return_value={'uuid': 'model-1', 'name': 'Model 1'}
-        )
+        ap.persistence_mgr.serialize_model = Mock(return_value={'uuid': 'model-1', 'name': 'Model 1'})

        service = LLMModelsService(ap)

@@ -371,12 +373,14 @@ class TestLLMModelsServiceCreateLLMModel:
        service = LLMModelsService(ap)

        # Execute
-        model_uuid = await service.create_llm_model({
-            'name': 'New LLM',
-            'provider_uuid': 'provider-uuid',
-            'abilities': [],
-            'extra_args': {},
-        })
+        model_uuid = await service.create_llm_model(
+            {
+                'name': 'New LLM',
+                'provider_uuid': 'provider-uuid',
+                'abilities': [],
+                'extra_args': {},
+            }
+        )

        # Verify
        assert model_uuid is not None
@@ -400,13 +404,16 @@ class TestLLMModelsServiceCreateLLMModel:
        service = LLMModelsService(ap)

        # Execute
-        model_uuid = await service.create_llm_model({
-            'uuid': 'preserved-uuid',
-            'name': 'Preserved UUID Model',
-            'provider_uuid': 'provider-uuid',
-            'abilities': [],
-            'extra_args': {},
-        }, preserve_uuid=True)
+        model_uuid = await service.create_llm_model(
+            {
+                'uuid': 'preserved-uuid',
+                'name': 'Preserved UUID Model',
+                'provider_uuid': 'provider-uuid',
+                'abilities': [],
+                'extra_args': {},
+            },
+            preserve_uuid=True,
+        )

        # Verify
        assert model_uuid == 'preserved-uuid'
@@ -459,12 +466,14 @@ class TestLLMModelsServiceCreateLLMModel:

        # Execute & Verify
        with pytest.raises(Exception, match='provider not found'):
-            await service.create_llm_model({
-                'name': 'No Provider Model',
-                'provider_uuid': 'nonexistent-provider',
-                'abilities': [],
-                'extra_args': {},
-            })
+            await service.create_llm_model(
+                {
+                    'name': 'No Provider Model',
+                    'provider_uuid': 'nonexistent-provider',
+                    'abilities': [],
+                    'extra_args': {},
+                }
+            )

    async def test_create_llm_model_with_provider_data(self):
        """Creates provider when provider data provided."""
@@ -490,16 +499,18 @@ class TestLLMModelsServiceCreateLLMModel:
        service = LLMModelsService(ap)

        # Execute - with provider data (no UUID)
-        result_uuid = await service.create_llm_model({
-            'name': 'Model with New Provider',
-            'provider': {
-                'requester': 'openai',
-                'base_url': 'https://api.openai.com',
-                'api_keys': ['key'],
-            },
-            'abilities': [],
-            'extra_args': {},
-        })
+        result_uuid = await service.create_llm_model(
+            {
+                'name': 'Model with New Provider',
+                'provider': {
+                    'requester': 'openai',
+                    'base_url': 'https://api.openai.com',
+                    'api_keys': ['key'],
+                },
+                'abilities': [],
+                'extra_args': {},
+            }
+        )

        # Verify - provider_service was called and UUID generated
        ap.provider_service.find_or_create_provider.assert_called_once()
@@ -525,11 +536,14 @@ class TestLLMModelsServiceUpdateLLMModel:
        service = LLMModelsService(ap)

        # Execute
-        await service.update_llm_model('existing-uuid', {
-            'uuid': 'should-be-removed',
-            'name': 'Updated Name',
-            'provider_uuid': 'provider-uuid',
-        })
+        await service.update_llm_model(
+            'existing-uuid',
+            {
+                'uuid': 'should-be-removed',
+                'name': 'Updated Name',
+                'provider_uuid': 'provider-uuid',
+            },
+        )

        # Verify - remove and load called
        ap.model_mgr.remove_llm_model.assert_called_once_with('existing-uuid')
@@ -549,10 +563,13 @@ class TestLLMModelsServiceUpdateLLMModel:

        # Execute & Verify
        with pytest.raises(Exception, match='provider not found'):
-            await service.update_llm_model('model-uuid', {
-                'name': 'Update',
-                'provider_uuid': 'nonexistent-provider',
-            })
+            await service.update_llm_model(
+                'model-uuid',
+                {
+                    'name': 'Update',
+                    'provider_uuid': 'nonexistent-provider',
+                },
+            )

    async def test_update_llm_model_reloads_context_length_as_column(self):
        """Updates runtime model with context_length outside extra_args."""
@@ -618,9 +635,7 @@ class TestEmbeddingModelsServiceGetEmbeddingModels:

        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)
-        ap.persistence_mgr.serialize_model = Mock(
-            return_value={'uuid': 'embedding-uuid', 'name': 'Test'}
-        )
+        ap.persistence_mgr.serialize_model = Mock(return_value={'uuid': 'embedding-uuid', 'name': 'Test'})

        service = EmbeddingModelsService(ap)

@@ -643,6 +658,7 @@ class TestEmbeddingModelsServiceGetEmbeddingModels:
        mock_provider_result = _create_mock_result([provider])

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -683,6 +699,7 @@ class TestEmbeddingModelsServiceGetEmbeddingModel:
        mock_provider_result = _create_mock_result([], first_item=provider)

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -742,11 +759,13 @@ class TestEmbeddingModelsServiceCreateEmbeddingModel:
        service = EmbeddingModelsService(ap)

        # Execute
-        model_uuid = await service.create_embedding_model({
-            'name': 'New Embedding',
-            'provider_uuid': 'provider-uuid',
-            'extra_args': {},
-        })
+        model_uuid = await service.create_embedding_model(
+            {
+                'name': 'New Embedding',
+                'provider_uuid': 'provider-uuid',
+                'extra_args': {},
+            }
+        )

        # Verify
        assert model_uuid is not None
@@ -767,11 +786,13 @@ class TestEmbeddingModelsServiceCreateEmbeddingModel:

        # Execute & Verify
        with pytest.raises(Exception, match='provider not found'):
-            await service.create_embedding_model({
-                'name': 'No Provider Embedding',
-                'provider_uuid': 'nonexistent',
-                'extra_args': {},
-            })
+            await service.create_embedding_model(
+                {
+                    'name': 'No Provider Embedding',
+                    'provider_uuid': 'nonexistent',
+                    'extra_args': {},
+                }
+            )


 class TestEmbeddingModelsServiceDeleteEmbeddingModel:
@@ -829,6 +850,7 @@ class TestRerankModelsServiceGetRerankModels:
        mock_provider_result = _create_mock_result([provider])

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -869,6 +891,7 @@ class TestRerankModelsServiceGetRerankModel:
        mock_provider_result = _create_mock_result([], first_item=provider)

        call_count = 0
+
        async def mock_execute(query):
            nonlocal call_count
            call_count += 1
@@ -928,11 +951,13 @@ class TestRerankModelsServiceCreateRerankModel:
        service = RerankModelsService(ap)

        # Execute
-        model_uuid = await service.create_rerank_model({
-            'name': 'New Rerank',
-            'provider_uuid': 'provider-uuid',
-            'extra_args': {},
-        })
+        model_uuid = await service.create_rerank_model(
+            {
+                'name': 'New Rerank',
+                'provider_uuid': 'provider-uuid',
+                'extra_args': {},
+            }
+        )

        # Verify
        assert model_uuid is not None
@@ -952,11 +977,13 @@ class TestRerankModelsServiceCreateRerankModel:

        # Execute & Verify
        with pytest.raises(Exception, match='provider not found'):
-            await service.create_rerank_model({
-                'name': 'No Provider Rerank',
-                'provider_uuid': 'nonexistent',
-                'extra_args': {},
-            })
+            await service.create_rerank_model(
+                {
+                    'name': 'No Provider Rerank',
+                    'provider_uuid': 'nonexistent',
+                    'extra_args': {},
+                }
+            )


 class TestRerankModelsServiceDeleteRerankModel:
@@ -995,9 +1022,7 @@ class TestEmbeddingModelsServiceGetEmbeddingModelsByProvider:

        mock_result = _create_mock_result([model1, model2])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)
-        ap.persistence_mgr.serialize_model = Mock(
-            return_value={'uuid': 'emb-1', 'name': 'Embedding 1'}
-        )
+        ap.persistence_mgr.serialize_model = Mock(return_value={'uuid': 'emb-1', 'name': 'Embedding 1'})

        service = EmbeddingModelsService(ap)

@@ -1022,9 +1047,7 @@ class TestRerankModelsServiceGetRerankModelsByProvider:

        mock_result = _create_mock_result([model1, model2])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)
-        ap.persistence_mgr.serialize_model = Mock(
-            return_value={'uuid': 'rerank-1', 'name': 'Rerank 1'}
-        )
+        ap.persistence_mgr.serialize_model = Mock(return_value={'uuid': 'rerank-1', 'name': 'Rerank 1'})

        service = RerankModelsService(ap)

@@ -1042,14 +1065,10 @@ class TestValidateProviderSupports:
    def _make_ap(requester_name: str, support_type):
        """Build a fake ap whose model_mgr resolves a manifest with support_type."""
        manifest = SimpleNamespace(spec={'support_type': support_type})
-        runtime_provider = SimpleNamespace(
-            provider_entity=SimpleNamespace(requester=requester_name)
-        )
+        runtime_provider = SimpleNamespace(provider_entity=SimpleNamespace(requester=requester_name))
        model_mgr = SimpleNamespace(
            provider_dict={'p1': runtime_provider},
-            get_available_requester_manifest_by_name=lambda name: manifest
-            if name == requester_name
-            else None,
+            get_available_requester_manifest_by_name=lambda name: manifest if name == requester_name else None,
        )
        return SimpleNamespace(model_mgr=model_mgr)

@@ -1066,9 +1085,7 @@ class TestValidateProviderSupports:
    async def test_allows_when_support_type_missing(self):
        # Manifest without support_type must not block (backward compatible)
        manifest = SimpleNamespace(spec={})
-        runtime_provider = SimpleNamespace(
-            provider_entity=SimpleNamespace(requester='legacy')
-        )
+        runtime_provider = SimpleNamespace(provider_entity=SimpleNamespace(requester='legacy'))
        model_mgr = SimpleNamespace(
            provider_dict={'p1': runtime_provider},
            get_available_requester_manifest_by_name=lambda name: manifest,