Skip to content

Commit 4063b77

Browse files
timmalichgaryrussell
authored andcommitted
Enhance mocking in TestRabbitTemplateTests
Fix exception in log fix travis Polishing
1 parent cf221da commit 4063b77

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

spring-rabbit-test/src/test/java/org/springframework/amqp/rabbit/test/examples/TestRabbitTemplateTests.java

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,14 @@
1818

1919
import static org.assertj.core.api.Assertions.assertThat;
2020
import static org.mockito.ArgumentMatchers.anyBoolean;
21+
import static org.mockito.ArgumentMatchers.anyMap;
22+
import static org.mockito.ArgumentMatchers.anyString;
2123
import static org.mockito.BDDMockito.given;
2224
import static org.mockito.BDDMockito.willReturn;
2325
import static org.mockito.Mockito.mock;
2426

27+
import java.io.IOException;
28+
2529
import org.junit.jupiter.api.Test;
2630

2731
import org.springframework.amqp.rabbit.annotation.EnableRabbit;
@@ -37,6 +41,7 @@
3741
import org.springframework.context.annotation.Configuration;
3842
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
3943

44+
import com.rabbitmq.client.AMQP;
4045
import com.rabbitmq.client.Channel;
4146

4247

@@ -91,13 +96,16 @@ public TestRabbitTemplate template() {
9196
}
9297

9398
@Bean
94-
public ConnectionFactory connectionFactory() {
99+
public ConnectionFactory connectionFactory() throws IOException {
95100
ConnectionFactory factory = mock(ConnectionFactory.class);
96101
Connection connection = mock(Connection.class);
97102
Channel channel = mock(Channel.class);
103+
AMQP.Queue.DeclareOk declareOk = mock(AMQP.Queue.DeclareOk.class);
98104
willReturn(connection).given(factory).createConnection();
99105
willReturn(channel).given(connection).createChannel(anyBoolean());
100106
given(channel.isOpen()).willReturn(true);
107+
given(channel.queueDeclare(anyString(), anyBoolean(), anyBoolean(), anyBoolean(), anyMap()))
108+
.willReturn(declareOk);
101109
return factory;
102110
}
103111

0 commit comments

Comments
 (0)