import com.tzld.supply.Application;
import com.tzld.supply.dao.mapper.supply.spider.ProduceVideoMapper;
import com.tzld.supply.job.ContentMediaSearchJob;
import com.tzld.supply.job.VideoGenerateJob;
import com.tzld.supply.model.po.supply.spider.ProduceVideo;
import com.tzld.supply.model.po.supply.spider.ProduceVideoExample;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.List;

@SpringBootTest(classes = Application.class)
@Slf4j
public class ProduceTest {

    @Autowired
    private ContentMediaSearchJob contentMediaSearchJob;
    @Autowired
    private VideoGenerateJob videoGenerateJob;
    @Autowired
    private ProduceVideoMapper produceVideoMapper;

    @Test
    public void test() {
        String contentId = "25835";
        contentMediaSearchJob.contentImageMediaSearchJob(contentId);
        contentMediaSearchJob.contentVideoMediaSearchJob(contentId);
        videoGenerateJob.videoVoiceGenerateJob(contentId);
        ProduceVideoExample example = new ProduceVideoExample();
        example.createCriteria().andContentIdEqualTo(Long.valueOf(contentId));
        List<ProduceVideo> produceVideoList = produceVideoMapper.selectByExample(example);
        if (produceVideoList.isEmpty()) {
            return;
        }
        ProduceVideo produceVideo = produceVideoList.get(0);
        videoGenerateJob.videoMaterialGenerateJob(produceVideo.getId());
        videoGenerateJob.videoGenerateJob(produceVideo.getId());
    }
}