From 323505ca484f793e7c1ab373426ab976a66b0cf0 Mon Sep 17 00:00:00 2001
From: tianfeng <769204422@qq.com>
Date: Mon, 24 Nov 2025 19:38:40 +0800
Subject: [PATCH] test update

---
 .../test/java/badge/BatchSendBadgeTest.java    | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/rc-service/rc-service-other/other-start/src/test/java/badge/BatchSendBadgeTest.java b/rc-service/rc-service-other/other-start/src/test/java/badge/BatchSendBadgeTest.java
index 2431889d..76d682fc 100644
--- a/rc-service/rc-service-other/other-start/src/test/java/badge/BatchSendBadgeTest.java
+++ b/rc-service/rc-service-other/other-start/src/test/java/badge/BatchSendBadgeTest.java
@@ -27,9 +27,16 @@ import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.test.context.junit4.SpringRunner;
 
+import java.text.SimpleDateFormat;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
 import java.util.Collection;
+import java.util.Date;
 import java.util.List;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -57,6 +64,8 @@ public class BatchSendBadgeTest {
     private BadgeBackpackService badgeBackpackService;
     @Autowired
     private BadgeConfigService badgeConfigService;
+    @Autowired
+    private MongoTemplate mongoTemplate;
 
     // 线程池大小
     private static final int THREAD_POOL_SIZE = 10;
@@ -69,10 +78,10 @@ public class BatchSendBadgeTest {
 
         try {
             // 1. 批量发放代理主播勋章 (HOST + AGENCY)
-//            sendHostAndAgencyBadges(executor);
+            sendHostAndAgencyBadges(executor);
 
             // 2. 批量发放BD LEADER勋章
-            sendBdLeaderBadges(executor);
+//            sendBdLeaderBadges(executor);
 
             // 3. 批量发放BD勋章
 //            sendBdBadges(executor);
@@ -94,7 +103,10 @@ public class BatchSendBadgeTest {
      * 发放代理主播勋章.
      */
     private void sendHostAndAgencyBadges(ExecutorService executor) {
-        List<TeamMember> list = teamMemberService.list();
+        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+        LocalDateTime startDateTime = LocalDateTime.parse("2025-11-23 00:00:00", formatter);
+
+        List<TeamMember> list = mongoTemplate.find(Query.query(Criteria.where("createTime").gte(startDateTime)), TeamMember.class);
         log.info("开始发放代理主播勋章，总数: {}", list.size());
 
         BadgeConfig agencyConfig = badgeConfigService.getByBadgeKey(BadgeKeyEnum.AGENCY.name());