spring單元測試下模擬rabbitmq
gradle添加引用
compile 'org.springframework.boot:spring-boot-starter-amqp'
testCompile 'com.github.fridujo:rabbitmq-mock:1.0.10'
添加bean對象
/**
* 模擬rabbitmq.
*/
@ActiveProfiles("test")
@Component
public class RabbitMqMock {
@Bean
public ConnectionFactory connectionFactory() {
return new CachingConnectionFactory(MockConnectionFactoryFactory.build());
}
}
添加測試的隊列
public static final String LIND_EXCHANGE = "test.basic.exchange";
public static final String LIND_QUEUE_ROUTEKEY = "test.basic.*";
public static final String LIND_QUEUE_ROUTEKEY1 = "test.basic.a1";
public static final String LIND_QUEUE_ROUTEKEY2 = "test.basic.a2";
/**
* 創建普通交換機.
*/
@Bean
public TopicExchange lindExchange() {
return (TopicExchange) ExchangeBuilder.topicExchange(LIND_EXCHANGE).durable(true)
.build();
}
@Bean
public Queue key1() {
return new Queue(LIND_QUEUE_ROUTEKEY1);
}
@Bean
public Queue key2() {
return new Queue(LIND_QUEUE_ROUTEKEY2);
}
/**
* 綁定了routekey,一個routekey可以被多個隊列綁定,類似于廣播.
*
* @return
*/
@Bean
public Binding bindBuildersRouteKey1() {
return BindingBuilder.bind(key1())
.to(lindExchange())
.with(LIND_QUEUE_ROUTEKEY);
}
/**
* bind.
*
* @return
*/
@Bean
public Binding bindBuildersRouteKey2() {
return BindingBuilder.bind(key2())
.to(lindExchange())
.with(LIND_QUEUE_ROUTEKEY);
}
@Autowired
private RabbitTemplate rabbitTemplate;
/**
* 發送撥打電話消息.
*/
public void publish(String message) {
try {
rabbitTemplate
.convertAndSend(MqConfig.LIND_EXCHANGE, MqConfig.LIND_QUEUE_ROUTEKEY,
message);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* subscriber.
*
* @param data .
*/
@RabbitListener(queues = MqConfig.LIND_DEAD_QUEUE)
public void customerSign(String data) {
try {
logger.info("從隊列拿到數據 :{}", data);
} catch (Exception ex) {
logger.error("簽約同步異常", ex);
}
}
總結:通過上面的幾行代碼,我們可以對rabbitmq隊列在測試環境中去模擬,方便了我們的測試,而這種方法比org.apache.qpid:qpid-broker:6.1.2這(zhe)個(ge)包要方便的(de)多,當(dang)然(ran)這(zhe)個(ge)包也(ye)支(zhi)持其(qi)它的(de)qpid協議的(de)隊列。