spring单元测试下模拟rabbitmq

原创

仓储大叔 2022-08-25 10:37:45 ©著作权

©著作权归作者所有：来自51CTO博客作者仓储大叔的原创作品，请联系作者获取转载授权，否则将追究法律责任

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);
    }
  }