ProduceTest.java 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. import com.tzld.supply.Application;
  2. import com.tzld.supply.dao.mapper.supply.spider.ProduceVideoMapper;
  3. import com.tzld.supply.job.ContentMediaSearchJob;
  4. import com.tzld.supply.job.VideoGenerateJob;
  5. import com.tzld.supply.model.po.supply.spider.ProduceVideo;
  6. import com.tzld.supply.model.po.supply.spider.ProduceVideoExample;
  7. import lombok.extern.slf4j.Slf4j;
  8. import org.junit.jupiter.api.Test;
  9. import org.springframework.beans.factory.annotation.Autowired;
  10. import org.springframework.boot.test.context.SpringBootTest;
  11. import java.util.List;
  12. @SpringBootTest(classes = Application.class)
  13. @Slf4j
  14. public class ProduceTest {
  15. @Autowired
  16. private ContentMediaSearchJob contentMediaSearchJob;
  17. @Autowired
  18. private VideoGenerateJob videoGenerateJob;
  19. @Autowired
  20. private ProduceVideoMapper produceVideoMapper;
  21. @Test
  22. public void test() {
  23. String contentId = "25835";
  24. contentMediaSearchJob.contentImageMediaSearchJob(contentId);
  25. contentMediaSearchJob.contentVideoMediaSearchJob(contentId);
  26. videoGenerateJob.videoVoiceGenerateJob(contentId);
  27. ProduceVideoExample example = new ProduceVideoExample();
  28. example.createCriteria().andContentIdEqualTo(Long.valueOf(contentId));
  29. List<ProduceVideo> produceVideoList = produceVideoMapper.selectByExample(example);
  30. if (produceVideoList.isEmpty()) {
  31. return;
  32. }
  33. ProduceVideo produceVideo = produceVideoList.get(0);
  34. videoGenerateJob.videoMaterialGenerateJob(produceVideo.getId());
  35. videoGenerateJob.videoGenerateJob(produceVideo.getId());
  36. }
  37. }