From 24a934295f7f62f59bf112dfdba94390062bd601 Mon Sep 17 00:00:00 2001
From: teknium1 <127238744+teknium1@users.noreply.github.com>
Date: Tue, 9 Jun 2026 01:59:13 -0700
Subject: [PATCH] test(yuanbao): add missing patch import to pipeline tests

The salvaged refactor's new tests use unittest.mock.patch (25 call sites)
but the import line only brought in AsyncMock and MagicMock, so 10 of the
new tests failed with NameError. Add patch to the import.
---
 tests/test_yuanbao_pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_yuanbao_pipeline.py b/tests/test_yuanbao_pipeline.py
index 3d5321bdc75..023d949b2cc 100644
--- a/tests/test_yuanbao_pipeline.py
+++ b/tests/test_yuanbao_pipeline.py
@@ -20,7 +20,7 @@ if _REPO_ROOT not in sys.path:
     sys.path.insert(0, _REPO_ROOT)
 
 import pytest
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import AsyncMock, MagicMock, patch
 
 from gateway.platforms.yuanbao import (
     InboundContext,