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

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