[英]stub a method in mock bean in spring integration context
I am trying to test a spring integration set up my unit test is as follows, 我正在尝试测试spring集成,我的单元测试如下:
@ContextConfiguration(loader = AnnotationConfigContextLoader.class, classes = { SIContext2Config.class })
public class FileMoverTest extends GenericTest {
@Test
public void testFileMover() throws InterruptedException {
Thread.sleep(1000);
int total = 2;
int n = 0;
for (int i = 1; i <= total; i++) {
@SuppressWarnings("unchecked")
Message<OfferMessage> msg = (Message<OfferMessage>) hdfsReadyChannel.receive(2000);
System.out.println("Message # " + i + " received:" + msg.getPayload().getOfferFile().getAbsolutePath());
n++;
}
Assert.state(n == total);
}
The context class is as follows: 上下文类如下:
@Configuration
public class SIContext2Config {
@Mock
private FsShell fsh;
@InjectMocks
@Mock
private MoveToHdfs fileMover;
@Bean
public Ingester ingester() {
return new Ingester(fileMover);
}
@Mock
private Ingester ingester;
@Bean
public FilePickupHandler filePickupHandler() {
return new FilePickupHandler();
}
}
Now, here is what I am trying to do: The Ingester bean has a method called handle(), inside which the MoveToHdfs object fileMover runs and calls move(). 现在,这是我要尝试的操作:Ingester bean有一个称为handle()的方法,在其中运行MoveToHdfs对象fileMover并调用move()。
public OfferMessage handle(Message<OfferMessage> msg) {
// get hive directory path
String remotePath = msg.getPayload().getOfferComponent().getHiveDirectory();
String localFile = msg.getPayload().getOfferFile().getAbsolutePath();
LOGGER.debug("Moving file {} to remote path:{}", localFile, remotePath);
if (!fileMover.move(localFile, remotePath, true)) {
throw new SomeException();
}
return msg.getPayload();
}
I just want that to return true. 我只想让它返回true。 But I can't figure out where to "stub" that, or how to stub that. 但是我不知道该在哪里“存根”或如何存根。
Why don't rework that fileMover
to the @Bean
? 为什么不将那个fileMover
做到@Bean
?
@Bean
public MoveToHdfs fileMover() {
MoveToHdfs fileMover = Mockito.mock(MoveToHdfs.class);
when(fileMover.move(anyString(), anyString(), anyBoolean())).thenReturn(true);
return fileMover;
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.