xiaoyong931011
2022-08-09 489b38cff851af79c1908c7493e9025eec87be7c
src/test/java/com/xcong/farmer/cms/KssframeworkApplicationTests.java
@@ -2,7 +2,10 @@
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.file.FileWriter;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -12,13 +15,11 @@
import com.xcong.farmer.cms.configurations.properties.CmsProperties;
import com.xcong.farmer.cms.modules.system.dto.AdminArticleDto;
import com.xcong.farmer.cms.modules.system.dto.WebArticleInPageDto;
import com.xcong.farmer.cms.modules.system.entity.ArticleEntity;
import com.xcong.farmer.cms.modules.system.entity.CmsTemplateEntity;
import com.xcong.farmer.cms.modules.system.entity.ColumnEntity;
import com.xcong.farmer.cms.modules.system.entity.UserEntity;
import com.xcong.farmer.cms.modules.system.entity.*;
import com.xcong.farmer.cms.modules.system.mapper.ArticleMapper;
import com.xcong.farmer.cms.modules.system.mapper.CmsTemplateMapper;
import com.xcong.farmer.cms.modules.system.mapper.ColumnMapper;
import com.xcong.farmer.cms.modules.system.mapper.CompanyMapper;
import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
import com.xcong.farmer.cms.modules.system.vo.AdminArticleVo;
import com.xcong.farmer.cms.modules.system.vo.WebArticleVo;
@@ -34,6 +35,8 @@
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
@@ -50,6 +53,32 @@
    @Resource
    private CmsTemplateMapper cmsTemplateMapper;
    @Test
    public void testFileWrite(){
        Long companyId = 24L;
        Long id = 38L;
        CmsTemplateEntity cmsTemplateEntity = cmsTemplateMapper.selectByIdAndCompanyId(id,companyId);
        String name = cmsTemplateEntity.getName();
        Integer type = cmsTemplateEntity.getType();
        String templatePath = "D:\\cmshtml";
        String path = cmsTemplateEntity.getPath();
        cmsTemplateMapper.delete(id,companyId);
        String pathNew = FileUtils.path(templatePath, path);
        File file = new File(pathNew);
        FileUtil.touch(file);
        //文件写入,直接覆盖
        FileWriter writer = new FileWriter(file);
        writer.write("测试121634964949", false);
        CmsTemplateEntity cmsTemplate = new CmsTemplateEntity();
        cmsTemplate.setCompanyId(companyId);
        cmsTemplate.setType(type);
        cmsTemplate.setName(name);
        cmsTemplate.setPath(path);
        cmsTemplateMapper.insert(cmsTemplate);
    }
    @Test
    public void testUserInsert() {
        for(int i = 0; i < 20; i++) {
@@ -79,7 +108,7 @@
    public void viewTemplateInfo() {
        Long id = 4L;
        CmsTemplateEntity cmsTemplateEntity = cmsTemplateMapper.selectById(id);
        String htmlUrl = cmsProperties.getBaseUrl() + cmsProperties.getTemplatePath();
        String htmlUrl = "" + cmsProperties.getTemplatePath();
        File uploadDir = new File(htmlUrl);
        if (!uploadDir.isDirectory()) {
            uploadDir.mkdir();
@@ -125,6 +154,37 @@
    @Autowired
    private ArticleMapper articleMapper;
    @Autowired
    private CompanyMapper companyMapper;
    @Test
    public void testCompanyId(){
        String website = "127.0.0.1";
        Long companyIdFromWebsite = getCompanyIdFromWebsite(website);
        System.out.print(companyIdFromWebsite);
    }
    private Long getCompanyIdFromWebsite(String website){
        List<CompanyEntity> companyEntities = companyMapper.selectList(new QueryWrapper<>());
        Long companyId = 0L;
        if(CollUtil.isNotEmpty(companyEntities)){
            for(CompanyEntity companyEntity : companyEntities){
                boolean contains = StrUtil.contains(companyEntity.getWebAddress(), website);
                if(contains){
                    companyId = companyEntity.getId();
                }
            }
        }
        return companyId;
    }
    public static void main(String[] args) {
        for(int i=0;i<10;i++){
            BigDecimal partnerIncomeNum = RandomUtil.randomBigDecimal(new BigDecimal("5000.001"), new BigDecimal("10000.0001")).setScale(6, RoundingMode.HALF_DOWN);
            System.out.println(partnerIncomeNum);
        }
    }
    @Test
    public void articleTest(){
        WebArticleInPageDto webArticleInPageDto = new WebArticleInPageDto();