1

我尝试搜索论坛,但找不到任何适合我的问题的解决方案。

我的代码是:

public class ClassTest {

    /** The context. */
    private Mockery context = new Mockery();

    /** The mapper service. */
    private MapperService mapperService;

    @Before
    public void init() {
        mapperService = context.mock(MapperService.class);
    }

    @Test
    public void testUpdateRequestStatusToSuccessful()
    {
        final RequestQueue inRequestQueue = new RequestQueue();
        final String finalOutputZipFileName = "abc.txt";
        final Mapper mapperEntityObj = new Mapper();
        mapperEntityObj.setAutoMarkSuccessful(true);
        mapperEntityObj.setMarkRecordsAsExported(true);

        MapperTypeReference mapperType=new MapperTypeReference();
        mapperType.setCode("C");
        mapperEntityObj.setMapperType(mapperType);

        DataSetTypeReference datasetType=new DataSetTypeReference();
        datasetType.setCode("TA");
        mapperEntityObj.setDataSet(datasetType);

        inRequestQueue.setRowCount(15);
        inRequestQueue.setId(25128002L);

        context.checking(new Expectations() {
        {
            allowing(referenceDataService).retrieveReferenceCodeByCode(RequestStatus.class, "L");
            will(returnValue(new RequestStatus()));

            allowing(referenceDataService).retrieveReferenceCodeByCode(RequestStatus.class, "N");
            will(returnValue(new RequestStatus()));

            allowing(requestQueueDao).persist(with(any(RequestQueue.class)));

            one(mapperService).persist(mapperEntityObj);

            allowing(referenceDataService).retrieveReferenceCodeByCode(RequestStatus.class, "A");
            will(returnValue(new RequestStatus()));

        }
    });

    // Row count > 0; MarkRecordsAsExported(true)
    deuDataAccessServiceImpl.updateRequestStatusToSuccessful(inRequestQueue, finalOutputZipFileName, mapperEntityObj);
}

我的服务类代码是

protected void updateRequestStatusToSuccessful(RequestQueue inRequestQueue,
        String finalOutputZipFileName, Mapper mapperEntityObj) {

    boolean mapperChanged = false;

    logger.info("updateRequestStatusToSuccessful(" + 
            "inRequestQueue, finalOutputZipFileName, mapperEntityObj) - begin");
    int rowCount=inRequestQueue.getRowCount();

    RequestStatus requestStatusToSet = null;

    logger.info("updateRequestStatusToSuccessful(" + 
            "inRequestQueue, finalOutputZipFileName, mapperEntityObj) - requestQueue: " + inRequestQueue);

    // Added to change the status of the request
    Boolean autoMarkSuccessful = mapperEntityObj.getAutoMarkSuccessful();
    Boolean markRecordsAsExported = mapperEntityObj.getMarkRecordsAsExported();
    if(rowCount==0){
        logger.info("No data found from dataSetKeys: Will set RequestStatus to No data found");
        requestStatusToSet = referenceDataService
                                                .retrieveReferenceCodeByCode(RequestStatus.class,
                                                RequestStatusEnum.NODATAFOUND.getCode());
        if(markRecordsAsExported){
            // Unlocking a mapper and setting the lastSuccessfulEndDate if successful
            mapperEntityObj.setIsLocked(false);
            mapperChanged = true;
        }
    }else if(markRecordsAsExported){
        logger.info("Mark transactions as exported enabled: Will set RequestStatus based on autoMarkSuccesful");
        if(autoMarkSuccessful){
            logger.info("autoMarkSuccesful enabled: Will set RequestStatus to Successful");
            requestStatusToSet = referenceDataService
                                                    .retrieveReferenceCodeByCode(RequestStatus.class,
                                                    RequestStatusEnum.SUCCESSFUL.getCode());
        } else {
            logger.info("autoMarkSuccesful disabled: Will set RequestStatus to Awaiting");
            requestStatusToSet = referenceDataService
                                                    .retrieveReferenceCodeByCode(RequestStatus.class,
                                                    RequestStatusEnum.AWAITING.getCode());
        }
    } else {
        logger.info("Mark transactions as exported disabled: Will set RequestStatus to Successful");
        requestStatusToSet = referenceDataService
                                            .retrieveReferenceCodeByCode(RequestStatus.class,
                                            RequestStatusEnum.SUCCESSFUL.getCode());
    }


    logger.info("Setting requestQueueStatus to : " + requestStatusToSet.getCode());
    inRequestQueue.setRequestStatus(requestStatusToSet);

    if (finalOutputZipFileName != null) {
        inRequestQueue.setOutputFilePathText(finalOutputZipFileName);
    }


    Date currentTimeStamp = Calendar.getInstance().getTime();
    // Values to set RequestQueueStartedTimestamp
    inRequestQueue.setCompletedTimestamp(currentTimeStamp);

    // Request queue object to persist
    logger.info("updateRequestStatusToSuccessful(" + 
            "inRequestQueue, finalOutputZipFileName, mapperEntityObj) - requestQueue: " + inRequestQueue);
    requestQueueDao.persist(inRequestQueue);
    // End of added to change the status
    logger.info("updateRequestStatusToSuccessful(" + 
            "inRequestQueue, finalOutputZipFileName, mapperEntityObj) - end");

    // PRD296 - FAST DEU
    logger.info("Before check to set mapper last successful end date.");
    if (((MapperTypesEnum.CUSTOMMAPPER.getCode().equals(mapperEntityObj.getMapperType().getCode()) 
            && DeuUtil.isTransactionBasedMapper(mapperEntityObj.getDataSet().getCode())) 
            || MapperTypesEnum.FLEXMAPPER.getCode().equals(mapperEntityObj.getMapperType().getCode()))
            && (autoMarkSuccessful || !markRecordsAsExported)
        ){
        logger.info("Set Last Successful end date: Copied Over Mapper End Date [" + mapperEntityObj.getMapperEndDate() + "]");  
        mapperEntityObj.setLastSuccessfulEndDate(mapperEntityObj.getMapperEndDate());
        mapperChanged = true;
    }
    else {
        logger.info("Mapper last successful end date not set: Mapper end date = " + mapperEntityObj.getMapperEndDate() 
                + ", Mapper type = " + mapperEntityObj.getMapperType().getCode() 
                + ", Mapper dataset code = "+ mapperEntityObj.getDataSet().getCode()
                + ", Is transaction based mapper = "
                + DeuUtil.isTransactionBasedMapper(mapperEntityObj.getDataSet().getCode()));
    }

    // Optimization
    if(mapperChanged){
        mapperService.persist(mapperEntityObj);
    }
}

但我仍然意外调用 mapperService.persists。

谁能帮我吗?

4

0 回答 0