Class RouteNormalizationTests

tests/test_veo_lite_support.py:309–420

class RouteNormalizationTests(unittest.IsolatedAsyncioTestCase):
    async def asyncSetUp(self):
        self.client = FlowClient(proxy_manager=None)
        self.client._acquire_video_launch_gate = AsyncMock(return_value=(True, None, None))
        self.client._release_video_launch_gate = AsyncMock()
        self.client._get_recaptcha_token = AsyncMock(return_value=("recaptcha-token", "browser-1"))
        self.client._notify_browser_captcha_request_finished = AsyncMock()

    async def test_openai_history_reference_image_can_drive_aspect_ratio(self):
        portrait_image = _make_image_bytes((900, 1600))
        request = ChatCompletionRequest(
            model="gemini-3.0-pro-image",
            messages=[
                ChatMessage(role="user", content="先生成一张图"),
                ChatMessage(
                    role="assistant",
                    content="![cat](https://example.com/cat.png)",
                ),
                ChatMessage(role="user", content="基于上一张图继续编辑"),
            ],
        )

        with patch(
            "src.api.routes.retrieve_image_data",
            new=AsyncMock(return_value=portrait_image),
        ):
            normalized = await _normalize_openai_request(request)

        self.assertEqual(normalized.model, "gemini-3.0-pro-image-portrait")
        self.assertEqual(len(normalized.images), 1)

    async def test_check_video_status_uses_media_payload_and_normalizes_response(self):
        captured = {}

        async def fake_make_request(method, url, json_data, use_at, at_token, **kwargs):
            captured["json_data"] = json_data
            return {
                "media": [
                    {
                        "name": "media-1",
                        "projectId": "project-1",
                        "mediaMetadata": {
                            "mediaStatus": {
                                "mediaGenerationStatus": "MEDIA_GENERATION_STATUS_SUCCESSFUL"
                            }
                        },
                        "video": {
                            "fifeUrl": "https://flow-content.google/video/11111111-1111-1111-1111-111111111111?token=abc",
                            "generatedVideo": {
                                "aspectRatio": "VIDEO_ASPECT_RATIO_LANDSCAPE"
                            },
                        },
                    }
                ]
            }

        self.client._make_request = AsyncMock(side_effect=fake_make_request)