From 829513657518c0077622b1904f49d79657b2537a Mon Sep 17 00:00:00 2001
From: Canlin Guo <961750412@qq.com>
Date: Sat, 25 Oct 2025 09:33:16 +0800
Subject: [PATCH] [UT][fix] Add missing get_ascend_config mock to NPUWorker
 initialization tests (#3729)

### What this PR does / why we need it?

Enable the unit tests that #3612 skipped.

### Does this PR introduce _any_ user-facing change?

### How was this patch tested?

Unit tests.

- vLLM main:
https://github.com/vllm-project/vllm/commit/17c540a993af88204ad1b78345c8a865cf58ce44

Signed-off-by: gcanlin <canlinguosdu@gmail.com>
---
 .github/workflows/vllm_ascend_test.yaml |  2 --
 tests/ut/worker/test_worker_v1.py       | 15 +++++++++++++++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/vllm_ascend_test.yaml b/.github/workflows/vllm_ascend_test.yaml
index afd6cd09..eee83d26 100644
--- a/.github/workflows/vllm_ascend_test.yaml
+++ b/.github/workflows/vllm_ascend_test.yaml
@@ -120,8 +120,6 @@ jobs:
         run: |
           export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/x86_64-linux/devlib
           pytest -sv --cov --cov-report=xml:unittests-coverage.xml tests/ut \
-            --ignore tests/ut/torchair/test_torchair_mla.py \
-            --ignore tests/ut/worker/test_worker_v1.py \
             --ignore tests/ut/torchair/models/test_torchair_deepseek_mtp.py \
             --ignore tests/ut/torchair/models/test_torchair_deepseek_v2.py \
             --ignore tests/ut/test_utils.py \
diff --git a/tests/ut/worker/test_worker_v1.py b/tests/ut/worker/test_worker_v1.py
index 2313e716..f77e0563 100644
--- a/tests/ut/worker/test_worker_v1.py
+++ b/tests/ut/worker/test_worker_v1.py
@@ -41,6 +41,7 @@ class TestNPUWorker(TestBase):
     @patch("vllm_ascend.ops")
     @patch("vllm_ascend.worker.worker_v1._register_atb_extensions")
     @patch("vllm_ascend.worker.worker_v1.register_ascend_customop")
+    @patch("vllm_ascend.worker.worker_v1.get_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_soc_version")
     @patch("vllm_ascend.worker.worker_v1.try_register_lib")
@@ -53,6 +54,7 @@ class TestNPUWorker(TestBase):
         mock_try_register_lib,
         mock_init_ascend_soc_version,
         mock_init_ascend_config,
+        mock_get_ascend_config,
         mock_register_ascend_customop,
         mock_register_atb_extensions,
         mock_ops,
@@ -61,6 +63,9 @@ class TestNPUWorker(TestBase):
         """Test NPUWorker normal initialization"""
         # Setup mock behavior
         mock_ops.register_dummy_fusion_op.return_value = None
+        mock_ascend_config = MagicMock()
+        mock_ascend_config.enable_cpu_binding = False
+        mock_get_ascend_config.return_value = mock_ascend_config
 
         # Import and create NPUWorker instance
         from vllm_ascend.worker.worker_v1 import NPUWorker
@@ -98,6 +103,7 @@ class TestNPUWorker(TestBase):
     @patch("vllm_ascend.ops")
     @patch("vllm_ascend.worker.worker_v1._register_atb_extensions")
     @patch("vllm_ascend.worker.worker_v1.register_ascend_customop")
+    @patch("vllm_ascend.worker.worker_v1.get_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_soc_version")
     @patch("vllm_ascend.worker.worker_v1.try_register_lib")
@@ -110,6 +116,7 @@ class TestNPUWorker(TestBase):
         mock_try_register_lib,
         mock_init_ascend_soc_version,
         mock_init_ascend_config,
+        mock_get_ascend_config,
         mock_register_ascend_customop,
         mock_register_atb_extensions,
         mock_ops,
@@ -119,6 +126,9 @@ class TestNPUWorker(TestBase):
         # Set trust_remote_code=True
         self.model_config_mock.trust_remote_code = True
         mock_ops.register_dummy_fusion_op.return_value = None
+        mock_ascend_config = MagicMock()
+        mock_ascend_config.enable_cpu_binding = False
+        mock_get_ascend_config.return_value = mock_ascend_config
 
         # Create NPUWorker instance
         from vllm_ascend.worker.worker_v1 import NPUWorker
@@ -138,6 +148,7 @@ class TestNPUWorker(TestBase):
     @patch("vllm_ascend.ops")
     @patch("vllm_ascend.worker.worker_v1._register_atb_extensions")
     @patch("vllm_ascend.worker.worker_v1.register_ascend_customop")
+    @patch("vllm_ascend.worker.worker_v1.get_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_config")
     @patch("vllm_ascend.worker.worker_v1.init_ascend_soc_version")
     @patch("vllm_ascend.worker.worker_v1.try_register_lib")
@@ -150,6 +161,7 @@ class TestNPUWorker(TestBase):
         mock_try_register_lib,
         mock_init_ascend_soc_version,
         mock_init_ascend_config,
+        mock_get_ascend_config,
         mock_register_ascend_customop,
         mock_register_atb_extensions,
         mock_ops,
@@ -159,6 +171,9 @@ class TestNPUWorker(TestBase):
         # Set custom cache_dtype
         self.cache_config_mock.cache_dtype = "float32"
         mock_ops.register_dummy_fusion_op.return_value = None
+        mock_ascend_config = MagicMock()
+        mock_ascend_config.enable_cpu_binding = False
+        mock_get_ascend_config.return_value = mock_ascend_config
 
         # Create NPUWorker instance
         from vllm_ascend.worker.worker_v1 import NPUWorker