项目笔记:测试类的编写

  由于需要客户终端上报数据,所以做一个测试类,模拟上报的数据信息。

  首先看下测试类的目录,主要是在 http 下面

  客户端上报的数据是这样的JSON:

{"pcInfo":{"ip":"192.168.118.45","mac":"11-22-12-23","onlyId":"1"},"user":{"name":"test"},"log":{"version":"1.0","type":"8001","content":{"policyName":"VRV","templateType":"0","contentlist":[{"SoftID":"1","MatchResult":"1","GenRule":"1001,1002","PirRule":"2001,2002","Remark1":"12365"},{"SoftID":"2","MatchResult":"0","GenRule":"1001","PirRule":"2002","Remark1":"125"}]},"areaCode":"4310"}}

  我们就来模拟生成这样的JSON。

  注意:测试方式的@Test是必须得加的。

/**
 * @author 正版化上报测试类
 * @version 2017/9/14
 * */
public class TestHttp {

    public HttpURLConnection connection = null;
    public OutputStreamWriter out = null;
    public JSONObject jsonObject = null;
    public JSONObject jsonObject1 = null;
    public JSONObject jsonObject2 = null;
    public JSONObject jsonObject3 = null;
    public JSONObject jsonObject4 = null;
    public JSONObject jsonObject5 = null;
    public JSONObject jsonObject6 = null;//分别定义6个JSONObject来存放不同的层级内容
    List<String> list = new ArrayList<String>();//需要定义一个list来存放contentlist的内容
    public URL postUrl = null;
    public TreeMap<String, Object> data = new TreeMap<String, Object>();
    
    public void initData(){
         try {
            postUrl = new URL("http://192.168.118.221:8080/JJFX/reportDataAction.do");//这里的ip要是我本机的ip
            connection = (HttpURLConnection) postUrl.openConnection();
            connection.setDoInput(true);
            connection.setDoOutput(true);
            connection.setRequestMethod("POST");
            connection.setUseCaches(false);
            connection.setInstanceFollowRedirects(true);
            connection.setRequestProperty("Content-Type",
                    "application/x-www-form-urlencoded");
            connection.setChunkedStreamingMode(5);
            connection.connect();
            out = new OutputStreamWriter(connection.getOutputStream(), "utf-8");
            jsonObject = new JSONObject();
            jsonObject1 = new JSONObject();
            jsonObject2 = new JSONObject();
            jsonObject3 = new JSONObject();
            jsonObject4 = new JSONObject();
            jsonObject5 = new JSONObject();
            jsonObject6 = new JSONObject();
            List<String> list = new ArrayList<String>();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 正版化信息上报接口
     * @author gwf
     * @version 2017-9-14
     * */
    @Test
    public void test1001(){
        
        initData();    
        //pcInfo
        jsonObject1.put("ip", "192.158.129.35");
        jsonObject1.put("mac", "11-32-22-43");
        jsonObject1.put("onlyId", "4");
        jsonObject.put("pcInfo", jsonObject1);//jsonObject1用于存放pcInfo的内容
        
        //user
        jsonObject2.put("name", "test4");
        jsonObject.put("user", jsonObject2);//jsonObject2用于存放user的内容
        
        //log
    //    jsonObject3.put("time", "");
        jsonObject3.put("version", "1.5");
        jsonObject3.put("type", "8001");//jsonObject3用于存放log的内容
                        
        //content
        jsonObject4.put("policyName", "VRV");
        jsonObject4.put("templateType", "0");//jsonObject4用于存放content的内容
        //contentList里的数组
        jsonObject5.put("softId", "1");
        jsonObject5.put("matchResult", "1");
        jsonObject5.put("genRule", "1001,1002");
        jsonObject5.put("pirRule", "2001,2002");
        jsonObject5.put("remark1", "123665");
        jsonObject5.put("softVer", "ff199");//jsonObject5用于存放contentlist的内容1
        
        jsonObject6.put("softId", "2");
        jsonObject6.put("matchResult", "0");
        jsonObject6.put("genRule", "1001");
        jsonObject6.put("pirRule", "2002");
        jsonObject6.put("remark1", "1245");
        jsonObject6.put("softVer", "ff59");//jsonObject5用于存放contentlist的内容2
        
        list.add(jsonObject5.toString());
        list.add(jsonObject6.toString());//把contentlist里面的内容1/2分别放入一个list中保存
        
        /*jsonObject4.put("antivirusDbVer", "1.0");
        jsonObject4.put("operationType", "1");
        jsonObject4.put("optTime", new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));*/
        jsonObject4.put("contentlist", list);//把保存的list放入contentlist
        
        jsonObject3.put("content", jsonObject4);//把contentlist放入content
        jsonObject3.put("areaCode", "4310");//jsonObject3继续存放log的内容areaCode
        jsonObject.put("log", jsonObject3);//jsonObject存放log的键值对
        String content ="jsonData="+ jsonObject.toString();    
        try {
            out.write(content);
        
            out.flush();
            // out.close();
            BufferedReader reader = new BufferedReader(new InputStreamReader(
                    connection.getInputStream()));
            out.flush();
            // out.close();
            String line;
            while ((line = reader.readLine()) != null) {
                System.out.println(line);
            }
            out.close();
            reader.close();
            connection.disconnect();
        } catch (IOException e) {
        
            e.printStackTrace();
        }
    }

    public static void main(String args[]){
          JSONObject jsonObject = new JSONObject();
          JSONObject jsonObject1 = new JSONObject();
          JSONObject jsonObject2 = new JSONObject();
          JSONObject jsonObject3 = new JSONObject();
          JSONObject jsonObject4 = new JSONObject();
          jsonObject1.put("ip", "192.168.118.212");
        jsonObject1.put("mac", "88-12-33-67-80");
        jsonObject1.put("onlyId", "123456");
        jsonObject.put("pcInfo", jsonObject1);
        
        //user
        jsonObject2.put("name", "服务器");
        jsonObject.put("user", jsonObject2);
        
        //log
    //    jsonObject3.put("time", "");
        jsonObject3.put("version", "1.0");
        jsonObject3.put("type", "7001");
                        
        //content
        jsonObject4.put("antivirusName", "VRV");
        jsonObject4.put("antivirusVer", "test2");
        jsonObject4.put("antivirusDbVer", "1.0");
        jsonObject4.put("operationType", "1");
        jsonObject4.put("optTime", "2017-7-12 15:27:01");
    
        jsonObject3.put("content", jsonObject4);
        jsonObject3.put("areaCode", "4310");
        jsonObject.put("log", jsonObject3);
        String jsonData = jsonObject.toString();
        System.out.println("===客户端上报数据================:"+jsonData);    
         Gson gson = new Gson();

        //System.out.println("===客户端上报数据================:"+ jsonObj111.getString("type"));
         
        ReportDataBean2 reportDataBean = gson.fromJson(jsonData, ReportDataBean2.class);
    
        PCInfo pcInfo = reportDataBean.getPcInfo();
        CloudAccount user = reportDataBean.getUser();
        ReportLog2 log = reportDataBean.getLog();
    //    ContentInfo content=log.getContent();
    }
    
}

  然后选择test1001,右键Run As,选择JUnit Test,运行

/**<B>说 明</B>: 数据上报Action
 * @author 作 者 名:gwf<br/>
 * @version 版 本 号:V1.0<br/>
 */
@Controller
@Scope("prototype")
public class ReportDataAction extends BaseAction {
    private static final long serialVersionUID = 7143456155737588152L;
    private Gson gson = new Gson();
    private String jsonData;
    private String crc;
    public String getJsonData() {
        return jsonData;
    }
    public void setJsonData(String jsonData) {
        this.jsonData = jsonData;
    }
    public String getCrc() {
        return crc;
    }
    public void setCrc(String crc) {
        this.crc = crc;
    }
    @Resource
    private PCInfoService pcInfoService;
    @Resource
    private CloudAccountService cloudAccountService;
    @Resource
    private ReportLogService reportLogService;
    @Resource
    private GenuineManagementStaticDao genuineManagementStaticDao;
    
    @SuppressWarnings("unchecked")
    @Override
    public String execute() {
        try {
            // TODO 开发测试使用数据,发布时删除
        //    String data = IOUtils.toString(HttpServletRequest.getInputStream(), "UTF-8")
            log.info("1:===信息上报数据================:"+jsonData);
            // 1.判断上报信息
            if (StringUtils.isBlank(jsonData)) {//这个jsonData就是我模拟过来的JSON字符串
                this.print("{\"result\":1,\"msg\":\"上报信息为空!\",\"description\":\"\"}");
                return null;
            }
            // 2.CRC验证判断jsonData数据是否完整
//            if (StringUtils.isBlank(crc) || !crc.equals(String.valueOf(CRCUtil.getCRC32Value(jsonData)))) {
//                this.print("{\"result\":1,\"msg\":\"上报数据不完整或存在安全风险\"},\"description\":\"CRC为空或不一致\"");
//                return null;
//            }
            // 3.解析JSON data
             JSONObject jsonObj111 =JSONObject.fromObject(jsonData);//解析json对象
             JSONArray ja=JSONArray.fromObject(jsonObj111.get("log"));//从解析的json对象里面获取key值log的内容
             String type="";
             for (int i = 0; i < ja.size(); i++) {//上面数据ja.size()为1
                 JSONObject o=ja.getJSONObject(i);
                 if(o.get("type")!=null){
                    type=o.get("type").toString();//获取接口type
                 }
             }
            ReportLog reportLog = new ReportLog();
            PCInfo pcInfo = new PCInfo();
            CloudAccount user = new CloudAccount();
            
            ReportDataBean2 reportDataBean = gson.fromJson(jsonData, ReportDataBean2.class);
            pcInfo = reportDataBean.getPcInfo();
            user = reportDataBean.getUser();
            JSONObject json = null;
            //JSONObject.fromObject(((ReportLog2<StrategyContentInfo>)reportDataBean.getLog()).getContent());
            if("3002".equals(type)||"3002".equals(type)){     //匹配接口的解析方法
                this.log.info("=======安全基线策略执行上报类型========");
//                ReportDataBean2 reportDataBean = gson.fromJson(jsonData, ReportDataBean2.class);
//                pcInfo = reportDataBean.getPcInfo();
//                user = reportDataBean.getUser();
                json = JSONObject.fromObject(((ReportLog2<StrategyContentInfo>)reportDataBean.getLog()).getContent());
                StrategyContentInfo contentInfo = gson.fromJson(json.toString(), StrategyContentInfo.class);
                ReportLog2<StrategyContentInfo> log2 = (ReportLog2<StrategyContentInfo>)reportDataBean.getLog();
                if(log2 != null){
                    reportLog.setType(log2.getType());
                    reportLog.setContent(JSONObject.fromObject(contentInfo).toString());
                    reportLog.setTime(log2.getTime());
                    reportLog.setVersion(log2.getVersion());
                    reportLog.setAreaCode(log2.getAreaCode());
                }
            }else if("5001".equals(type)){
                this.log.info("=======合规检查===》总体检测结果统计上报类型========");
                json = JSONObject.fromObject(((ReportLog2<ComplianceCheckTotal>)reportDataBean.getLog()).getContent());
                ReportLog2<ComplianceCheckTotal> log2 = (ReportLog2<ComplianceCheckTotal>)reportDataBean.getLog();
                ComplianceCheckTotal contentInfo = gson.fromJson(json.toString(), ComplianceCheckTotal.class);
                if(log2 != null){
                    reportLog.setType(log2.getType());
                    reportLog.setContent(JSONObject.fromObject(contentInfo).toString());
                    reportLog.setTime(log2.getTime());
                    reportLog.setVersion(log2.getVersion());
                    reportLog.setAreaCode(log2.getAreaCode());
                }
            }else if("5002".equals(type)){
                this.log.info("=======合规检查===》单个检测项结果上报类型========");
                json = JSONObject.fromObject(((ReportLog2<ComplianceCheckSingle>)reportDataBean.getLog()).getContent());
                ReportLog2<ComplianceCheckSingle> log2 = (ReportLog2<ComplianceCheckSingle>)reportDataBean.getLog();
                ComplianceCheckSingle contentInfo = gson.fromJson(json.toString(), ComplianceCheckSingle.class);
                if(log2 != null){
                    reportLog.setType(log2.getType());
                    reportLog.setContent(JSONObject.fromObject(contentInfo).toString());
                    reportLog.setTime(log2.getTime());
                    reportLog.setVersion(log2.getVersion());
                    reportLog.setAreaCode(log2.getAreaCode());
                }
            }else if("6001".equals(type)){
                this.log.info("=======补丁信息数据上报类型========");
                json = JSONObject.fromObject(((ReportLog2<PatchInfo>)reportDataBean.getLog()).getContent());
                ReportLog2<PatchInfo> log2 = (ReportLog2<PatchInfo>)reportDataBean.getLog();
                PatchInfo contentInfo = gson.fromJson(json.toString(), PatchInfo.class);
                if(log2 != null){
                    reportLog.setType(log2.getType());
                    reportLog.setContent(JSONObject.fromObject(contentInfo).toString());
                    reportLog.setTime(log2.getTime());
                    reportLog.setVersion(log2.getVersion());
                    reportLog.setAreaCode(log2.getAreaCode());
                }
            }else if("1001".equals(type)){//客户端注册,需要检查授权文件
                ReportDataBean reportDataBean1 = gson.fromJson(jsonData, ReportDataBean.class);
                pcInfo = reportDataBean1.getPcInfo();
                user = reportDataBean1.getUser();
                reportLog = reportDataBean1.getLog();
                //String onlyId = pcInfo.getOnlyId();
                PCInfo temp = pcInfoService.queryByOnlyID(pcInfo);
                if(temp == null){
                    
                }
            }else if("8001".equals(type)){   //我的接口type是8001
                this.log.info("=======正版化执行上报类型========");
                json = JSONObject.fromObject(((ReportLog2<GenuineManagementContentInfo>)reportDataBean.getLog()).getContent());
                //获取content里面内容
                //{"policyName":"VRV","templateType":"0","contentlist":[{"softId":"1","matchResult":"1","genRule":"1001,1002","pirRule":"2001,2002","remark1":"123665","softVer":"ff11"}]}
                GenuineManagementContentInfo contentInfo = gson.fromJson(json.toString(), GenuineManagementContentInfo.class);
                ReportLog2<GenuineManagementContentInfo> log2 = (ReportLog2<GenuineManagementContentInfo>)reportDataBean.getLog();
                if(log2 != null){
                    reportLog.setType(log2.getType());
                    reportLog.setContent(JSONObject.fromObject(contentInfo).toString());
                    reportLog.setTime(log2.getTime());
                    reportLog.setVersion(log2.getVersion());
                    reportLog.setAreaCode(log2.getAreaCode());//把一个个数据设置到reportLog里面去
                }
            }else{//没有content字段或者content字段内容为字符串非json对象
                ReportDataBean reportDataBean1 = gson.fromJson(jsonData, ReportDataBean.class);
                pcInfo = reportDataBean1.getPcInfo();
                user = reportDataBean1.getUser();
                reportLog = reportDataBean1.getLog();        
            }
            // 4.入库操作
            reportLogService.saveReportDate(pcInfo, user, reportLog);
            this.print("{\"result\":0,\"msg\":\"上报数据成功\",\"description\":\"\"}");
        } catch (Exception e) {
            this.print("{\"result\":1,\"msg\":\"上报数据失败\",\"description\":\"\"}");
            e.printStackTrace();
        }
        return null;
    }
}

上面获取数据的这三行要特别注意:GenuineManagementContentInfo 里面的数据必须跟json里面的一一对应才可以获取到数据,否则是获取不到的

json = JSONObject.fromObject(((ReportLog2<GenuineManagementContentInfo>)reportDataBean.getLog()).getContent());
GenuineManagementContentInfo contentInfo = gson.fromJson(json.toString(), GenuineManagementContentInfo.class);
ReportLog2<GenuineManagementContentInfo> log2 = (ReportLog2<GenuineManagementContentInfo>)reportDataBean.getLog();

  先看下我的  GenuineManagementContentInfo 里的内容,与content里一致,这个是在bean目录下

/**
 * 正版化数据上报content数据类
 * @author gwf
 * @version 2017.09.06 16:06
 * */
public class GenuineManagementContentInfo {
    
    private String policyName;
    private String templateType;
    private List<GenuineManagementStaticList> contentlist;
    
    public String getPolicyName() {
        return policyName;
    }
    public String getTemplateType() {
        return templateType;
    }
    public void setPolicyName(String policyName) {
        this.policyName = policyName;
    }
    public void setTemplateType(String templateType) {
        this.templateType = templateType;
    }
    public List<GenuineManagementStaticList> getContentlist() {
        return contentlist;
    }
    public void setContentlist(List<GenuineManagementStaticList> contentlist) {
        this.contentlist = contentlist;
    }
    
}

   注意:这里面还有个List,这个list的实体类型GenuineManagementStaticList,也要和contentlist里的数据一一对应才行。看下下面数据,与contentlist里的数据一一对应。

public class GenuineManagementStaticList  implements Serializable{

    private static final long serialVersionUID = -7081067901120934024L;
    /** 软件ID  **/
    private Integer softId;    
    /** 匹配结果 **/
    private Integer matchResult;
    /** 正版规则的检查依据 **/
    private String genRule;
    /** 非正版规则的检查依据 **/
    private String pirRule;
    /** 备注 **/
    private String remark1;
    /** 版本 **/
    private String softVer;

    public String getSoftVer() {
        return softVer;
    }
    public void setSoftVer(String softVer) {
        this.softVer = softVer;
    }
    public Integer getSoftId() {
        return softId;
    }
    public void setSoftId(Integer softId) {
        this.softId = softId;
    }
    public Integer getMatchResult() {
        return matchResult;
    }
    public void setMatchResult(Integer matchResult) {
        this.matchResult = matchResult;
    }
    public String getGenRule() {
        return genRule;
    }
    public void setGenRule(String genRule) {
        this.genRule = genRule;
    }
    public String getPirRule() {
        return pirRule;
    }
    public void setPirRule(String pirRule) {
        this.pirRule = pirRule;
    }
    public String getRemark1() {
        return remark1;
    }
    public void setRemark1(String remark1) {
        this.remark1 = remark1;
    }

    public GenuineManagementStaticList() {
        super();
    }
    public GenuineManagementStaticList(Integer softId, Integer matchResult, String genRule, String pirRule,
            String remark1, String softVer) {
        this.softId = softId;
        this.matchResult = matchResult;
        this.genRule = genRule;
        this.pirRule = pirRule;
        this.remark1 = remark1;
        this.softVer = softVer;
    }
    
}

  最后看下保存各数据的saveReportDate方法。

    @Override
    public void saveReportDate(PCInfo pcInfo, CloudAccount user, ReportLog log) {

        String type=log.getType();//获取接口type
        pcInfo = pcInfoDao.saveNoExist(pcInfo);//保存pcInfo表
        if(null != pcInfo){
            log.setPcInfo(pcInfo);//然后呢,把pcInfo设置到log中,主要是为了拿到关联的pcInfoId
        }
        else{
             System.out.println("设备不存在!");
        }
        if(null != user){
            user = cloudAccountDao.saveByName(user);//保存user表
        }
        else{
             System.out.println("云账号为空,不存在!");
        }
        log.setCloudAccount(user);//然后呢,把user设置到log中,主要是为了拿到关联的cloudAccountId

        if ("1001".equals(type) || "1003".equals(type)) {//这些都是匹配接口类型
            log = specialTypeOperate("1001", log);
        } else if ("1002".equals(type)) {
            log = specialTypeOperate("1002", log);
        }
        if (type.equals("5001") ) { // 合规检查总体检测结果
            if (StringUtils.isNotBlank(log.getContent())) {
                Gson gson = new Gson();
                ComplianAlarmInfo complianAlarmInfo =gson.fromJson(log.getContent(), ComplianAlarmInfo.class);
                complianAlarmInfo.setIpAddress(pcInfo.getIp());
                complianAlarmInfo.setMacAddress(pcInfo.getMac());
                complianAlarmInfo.setUserName(user.getName());
                complianAlarmInfo.setOrganization(log.getAreaCode());
                complianAlarmInfoDao.save(complianAlarmInfo);   
            }

        } else if (type.equals("5002")) {// 合规检查单个检测结果
            if (StringUtils.isNotBlank(log.getContent())) {
                Gson gson = new Gson();
                ComplianAlarmInfoDetail complianAlarmInfoDetail =gson.fromJson(log.getContent(), ComplianAlarmInfoDetail.class);            
                complianAlarmInfoDetailDao.save(complianAlarmInfoDetail);
            }
        }  else if (log.getType().equals("5101")) {// 杀毒软件安装上报
            if (StringUtils.isNotBlank(log.getContent())) {
                Gson gson = new Gson();
                AntivirusSoftStatic antivirusSoftStatic =gson.fromJson(log.getContent(), AntivirusSoftStatic.class);    
                String pcInfoId=pcInfoDao.queryByOnlyID(log.getPcInfo()).getId();
                AntivirusSoftStatic  rl=antivirusSoftStaticDao.queryByPcInfoId(pcInfoId,antivirusSoftStatic.getAntivirusName());
                if (rl!=null) {            
                    try {
                        rl.setAntivirusName(antivirusSoftStatic.getAntivirusName());
                        rl.setInstallTime(antivirusSoftStatic.getInstallTime());
                        rl.setInstallType(antivirusSoftStatic.getInstallType());
                        rl.setPcInfo(log.getPcInfo());
                        rl.setCloudAccount(log.getCloudAccount());
                        rl.setVersion(antivirusSoftStatic.getVersion());
                    }  catch (Exception e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    }                    
                }else{
                    antivirusSoftStatic.setPcInfo(log.getPcInfo());
                    antivirusSoftStatic.setCloudAccount(log.getCloudAccount());
                    antivirusSoftStaticDao.save(antivirusSoftStatic);
                }
            }
        }else if (log.getType().equals("8001")) {// 正版化上报
            if (StringUtils.isNotBlank(log.getContent())) {
                Gson gson = new Gson();
                GenuineManagementContentInfo genuineManagementStatic =gson.fromJson(log.getContent(), GenuineManagementContentInfo.class);
                //和前面一样获取到content里面内容,主要是为了拿到genuineManagementStatic.getContentlist()
                for(GenuineManagementStaticList gm : genuineManagementStatic.getContentlist()){//遍历contentlist
                    GenuineManagementStatic gms = new GenuineManagementStatic();//声明一个新的实体类实例用于保存到数据库表
                    gms.setSoftId(gm.getSoftId());
                    gms.setGenRule(gm.getGenRule());
                    gms.setMatchResult(gm.getMatchResult());
                    gms.setPirRule(gm.getPirRule());
                    gms.setRemark1(gm.getRemark1());
                    gms.setVersion(gm.getSoftVer());
                    gms.setPcInfo(log.getPcInfo());
                    gms.setCloudAccount(log.getCloudAccount());
                    genuineManagementStaticDao.save(gms);    //保存到数据库表genuineManagementStatic里
                }
            }
        }else {
            if(log.getType().equals("1001")||log.getType().equals("1002")||log.getType().equals("1003")){
                String pcInfoId=pcInfoDao.queryByOnlyID(log.getPcInfo()).getId();
                ReportLog rl=reportLogDao.queryByPcInfoId(pcInfoId,log.getType());
                if (rl!=null) {
                    //更新reportLog
                    log.setId(rl.getId());
                    rl.setAreaCode(log.getAreaCode());
                    rl.setCloudAccount(log.getCloudAccount());
                    //rl.setCloudAccount(cloudAccount);
                    rl.setContent(log.getContent());
                    //rl.setPcInfo(pcInfo2);
                    rl.setPcInfo(log.getPcInfo());
                    rl.setTime(log.getTime());
                    rl.setType(log.getType());
                    rl.setVersion(log.getVersion());
//                    log.setId(rl.getId());
//                    reportLogDao.update(log);    
                } else {
                    reportLogDao.save(log);
                }
            }else{
                reportLogDao.save(log);
            }        
        }
    }

  至此,测试类编写完毕,测试数据,发现数据都存入到了表里。

posted @ 2017-09-15 20:19  古兰精  阅读(450)  评论(0编辑  收藏  举报