Question

我正在尝试为我的代码编写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

Answer 1

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或使字段受保护）。

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

1 个答案: