From 7b95e2a75fa66c02a1d13ca880c6300e24298860 Mon Sep 17 00:00:00 2001 From: Asuka Minato Date: Wed, 22 Jul 2026 13:52:06 +0900 Subject: [PATCH] test: use sqlite3 session in test_generator_api_missing (#38719) --- .../console/app/test_generator_api_missing.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/api/tests/unit_tests/controllers/console/app/test_generator_api_missing.py b/api/tests/unit_tests/controllers/console/app/test_generator_api_missing.py index bdc3976e14e..d6f6bd703f4 100644 --- a/api/tests/unit_tests/controllers/console/app/test_generator_api_missing.py +++ b/api/tests/unit_tests/controllers/console/app/test_generator_api_missing.py @@ -1,5 +1,6 @@ import pytest from flask import Flask +from sqlalchemy.orm import Session from controllers.console.app import generator as generator_module from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError @@ -102,12 +103,14 @@ def test_structured_output_generate_exceptions(app: Flask, monkeypatch: pytest.M method(api, "t1") -def test_instruction_generate_exceptions(app: Flask, monkeypatch: pytest.MonkeyPatch) -> None: +@pytest.mark.parametrize("sqlite_session", [()], indirect=True) +def test_instruction_generate_exceptions( + app: Flask, + monkeypatch: pytest.MonkeyPatch, + sqlite_session: Session, +) -> None: api = generator_module.InstructionGenerateApi() method = unwrap(api.post) - from types import SimpleNamespace - - session = SimpleNamespace() exceptions_to_test = [ (ProviderTokenNotInitError("token error"), generator_module.ProviderNotInitializeError), @@ -135,4 +138,4 @@ def test_instruction_generate_exceptions(app: Flask, monkeypatch: pytest.MonkeyP }, ): with pytest.raises(expected_exception): - method(api, session, "t1") + method(api, sqlite_session, "t1")