你可以在你正在测试的类中模拟方法调用吗?

时间:2012-01-11 11:37:53

标签: java testing jmock

我正在尝试为我的代码编写JUnit测试,但在我的一些方法中调用了其他方法。是否可以模拟这些调用?

E.g。

s3FileWrite(File file, Status status)
{
  S3 s3         = new S3(file.getName, s3Service)
  String key    = s3.getKey();
  String bucket = s3.getBucket();
  File tmp = new File("tmp/" + s3.getName());

  writeFile(key, bucket, tmp, status); //local method call I want to mock out
}//awsFileWrite

writeFile方法是我想要模拟的,它是我正在测试的类的一部分,但我不知道如何模拟它。我想嘲笑我正在测试的课程,然后将调用添加到我的期望中,但它仍会调用该方法。

有人可以给我一些建议吗?

编辑:

我的JMock代码如下所示:

@Test
public void testS3FileWrite()
{       
    fileName     = context.mock(File.class);
    s3Service    = context.mock(FileDataAccessor.class);
    s3           = context.mock(S3.class);          
    reportWriter = context.mock(ReportWriter.class); 

    try
    {
        context.checking(new Expectations(){{           
            oneOf(fileMetaData).getKey();
            will(returnValue("s3Key")); 

            oneOf(fileMetaData).getBucketName();
            will(returnValue("BucketName"));

            oneOf(fileMetaData).getName();
            will(returnValue("TempFile"));  

            ((MethodClause) oneOf (any(File.class))).method("File").with(same("tmp/TempFile")); 

            oneOf(reportWriter).writeFile(with(same("s3Key")),
                                   with(same("BucketName")), 
                                   with(any(File.class), 
                                   with(same(Status.OK)));
        }});//Expectations
    }
    catch (Exception e)
    {
        ErrorStatus.debug("Exception in ReportTest.testS3FileWrite: " + e);
    }//try-catch  

    ReportWriter test = new ReportWriter(status);
    test.awsFileWrite(fileName, Status.OK);
}//testAWSFileWrite

1 个答案:

答案 0 :(得分:4)

PowerMock允许您部分模拟类,但它是为EasyMock而非JMock设计的。无论如何,这不是最好的方法。

添加新课程FileWriter并将writeFile方法移至该课程,然后在您的课程中测试, 委托给一个人:

// default implementation, can be replaced in tests
FileWriter fileWriter = new FileWriter();

writeFile(key, bucket, tmp, status) {
    fileWriter.write(key, bucket, tmp, status);
}; 

在测试代码中,使用模拟fileWriter覆盖被测试类中的FileWriter字段(添加setter或使字段受保护)。