From b08e6c28cdd8bda2fd36597c11bc7213a72678d8 Mon Sep 17 00:00:00 2001
From: longz <longzhaoz@qq.com>
Date: 星期三, 29 七月 2026 16:24:17 +0800
Subject: [PATCH] 各类修复

---
 ruoyi-service/ruoyi-order/src/test/java/com/ruoyi/order/controller/TChargingOrderControllerTest.java |   32 ++++++++++++++++++++++++++++++++
 1 files changed, 32 insertions(+), 0 deletions(-)

diff --git a/ruoyi-service/ruoyi-order/src/test/java/com/ruoyi/order/controller/TChargingOrderControllerTest.java b/ruoyi-service/ruoyi-order/src/test/java/com/ruoyi/order/controller/TChargingOrderControllerTest.java
index 47a58bf..d49b658 100644
--- a/ruoyi-service/ruoyi-order/src/test/java/com/ruoyi/order/controller/TChargingOrderControllerTest.java
+++ b/ruoyi-service/ruoyi-order/src/test/java/com/ruoyi/order/controller/TChargingOrderControllerTest.java
@@ -1,6 +1,7 @@
 package com.ruoyi.order.controller;
 
 import com.ruoyi.account.api.feignClient.AppUserClient;
+import com.ruoyi.chargingPile.api.model.Site;
 import com.ruoyi.order.api.query.ChargingOrderQuery;
 import com.ruoyi.order.api.vo.TCharingOrderVO;
 import com.ruoyi.order.dto.SixChargingDto;
@@ -80,4 +81,35 @@
         verify(chargingOrderService).chargingOrder(same(query));
         verifyNoInteractions(appUserClient);
     }
+
+    @Test
+    void partnerFilterIncludesPrimaryAndVenueSitesRegardlessOfPartnerType() {
+        Site primarySite = new Site();
+        primarySite.setId(11);
+        primarySite.setPartnerId(9);
+        Site venueSite = new Site();
+        venueSite.setId(12);
+        venueSite.setPartnerId(3);
+        venueSite.setVenuePartnerId(9);
+        Site unrelatedSite = new Site();
+        unrelatedSite.setId(13);
+        unrelatedSite.setPartnerId(4);
+
+        List<Integer> result = TChargingOrderController.filterPlatformRevenueSiteIdsByPartner(
+                Arrays.asList(11, 12, 13), Arrays.asList(primarySite, venueSite, unrelatedSite), 9);
+
+        assertEquals(Arrays.asList(11, 12), result);
+    }
+
+    @Test
+    void partnerFilterReturnsImpossibleSiteWhenPartnerHasNoAuthorizedSites() {
+        Site unrelatedSite = new Site();
+        unrelatedSite.setId(13);
+        unrelatedSite.setPartnerId(4);
+
+        List<Integer> result = TChargingOrderController.filterPlatformRevenueSiteIdsByPartner(
+                Arrays.asList(13), Arrays.asList(unrelatedSite), 9);
+
+        assertEquals(Arrays.asList(-1), result);
+    }
 }

--
Gitblit v1.7.1