test code

api-module/src/test/java/com/tzld/piaoquan/api/ContentPlatformTest.java

@@ -2,14 +2,14 @@ package com.tzld.piaoquan.api;
 
 import com.aliyun.openservices.shade.com.google.common.collect.Sets;
 import com.tzld.piaoquan.api.dao.mapper.contentplatform.ContentPlatformAccountMapper;
+import com.tzld.piaoquan.api.dao.mapper.contentplatform.ContentPlatformGzhPlanVideoMapper;
 import com.tzld.piaoquan.api.dao.mapper.contentplatform.ContentPlatformIllegalMsgMapper;
 import com.tzld.piaoquan.api.job.contentplatform.ContentPlatformDatastatJob;
-import com.tzld.piaoquan.api.model.po.contentplatform.ContentPlatformAccount;
-import com.tzld.piaoquan.api.model.po.contentplatform.ContentPlatformAccountExample;
-import com.tzld.piaoquan.api.model.po.contentplatform.ContentPlatformIllegalMsg;
+import com.tzld.piaoquan.api.model.po.contentplatform.*;
 import com.tzld.piaoquan.growth.common.model.bo.VideoDetail;
 import com.tzld.piaoquan.growth.common.service.MessageAttachmentService;
 import com.tzld.piaoquan.growth.common.utils.DateUtil;
+import com.tzld.piaoquan.growth.common.utils.MessageUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -34,6 +34,8 @@ public class ContentPlatformTest {
     ContentPlatformAccountMapper accountMapper;
     @Autowired
     private ContentPlatformIllegalMsgMapper contentPlatformIllegalMsgMapper;
+    @Autowired
+    private ContentPlatformGzhPlanVideoMapper gzhPlanVideoMapper;
 
     @Test
     public void testSyncContentPlatformQwDatastatReplyTotalJob() {
@@ -117,4 +119,15 @@ public class ContentPlatformTest {
         }
     }
 
+    @Test
+    public void setRootSourceId() {
+        ContentPlatformGzhPlanVideoExample example = new ContentPlatformGzhPlanVideoExample();
+        example.createCriteria().andPageUrlIsNotNull().andRootSourceIdIsNull();
+        List<ContentPlatformGzhPlanVideo> list = gzhPlanVideoMapper.selectByExample(example);
+        for (ContentPlatformGzhPlanVideo item : list) {
+            item.setRootSourceId(MessageUtil.getRootSourceId(item.getPageUrl()));
+            gzhPlanVideoMapper.updateByPrimaryKeySelective(item);
+        }
+    }
+
 }