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

问题描述

我正在尝试为我的代码编写JUnit测试，但是在某些方法中调用了其他方法.可以嘲笑这些呼叫吗?

例如

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

解决方案

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字段(添加设置器或使该字段受保护).

I'm trying to write JUnit tests for my code but with in some of my methods other methods are called. Is it possible to mock these calls out?

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

The writeFile method is what I want to mock out and it's part of the class I am testing, but I don't know how to mock it out. I thought mocking out the class I'm testing and then adding the call to my expectations would do it but it still calls the method.

Can anyone give me some advice on what to do here please?

EDIT:

My JMock code looks like this:

@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

解决方案

PowerMock lets you partially mock classes, but it's designed for EasyMock not JMock. In any case, this is not the best approach.

Add a new class FileWriter and move the writeFile method to it, then in your class under test,delegate to one:

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

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

In your test code, overwrite the fileWriter field in a the class under test (add a setter or make the field protected) with a mock FileWriter.

这篇关于您可以在要测试的类中模拟方法调用吗?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！