| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.rocketmq.spring.core; |
| |
| import javax.annotation.Resource; |
| import org.apache.rocketmq.client.AccessChannel; |
| import org.apache.rocketmq.client.producer.SendCallback; |
| import org.apache.rocketmq.client.producer.SendResult; |
| import org.apache.rocketmq.client.producer.TransactionMQProducer; |
| import org.apache.rocketmq.spring.annotation.RocketMQTransactionListener; |
| import org.apache.rocketmq.spring.autoconfigure.RocketMQAutoConfiguration; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.springframework.beans.factory.annotation.Value; |
| import org.springframework.boot.test.context.SpringBootTest; |
| import org.springframework.messaging.Message; |
| import org.springframework.messaging.MessagingException; |
| import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
| |
| import static org.assertj.core.api.Assertions.assertThat; |
| |
| @RunWith(SpringJUnit4ClassRunner.class) |
| @SpringBootTest(properties = { |
| "rocketmq.nameServer=127.0.0.1:9876", "rocketmq.producer.group=rocketMQTemplate-test-producer_group", |
| "test.rocketmq.topic=test", "rocketmq.producer.access-key=test-ak", |
| "rocketmq.producer.secret-key=test-sk", "rocketmq.accessChannel=LOCAL", |
| "rocketmq.producer.sendMessageTimeout= 3500", "rocketmq.producer.retryTimesWhenSendFailed=3", |
| "rocketmq.producer.retryTimesWhenSendAsyncFailed=3"}, classes = {RocketMQAutoConfiguration.class, TransactionListenerImpl.class}) |
| |
| public class RocketMQTemplateTest { |
| @Resource |
| RocketMQTemplate rocketMQTemplate; |
| |
| @Value("${test.rocketmq.topic}") |
| String topic; |
| |
| @Test |
| public void testSendMessage() { |
| try { |
| rocketMQTemplate.syncSend(topic, "payload"); |
| } catch (MessagingException e) { |
| assertThat(e).hasMessageContaining("org.apache.rocketmq.remoting.exception.RemotingConnectException: connect to [127.0.0.1:9876] failed"); |
| } |
| |
| try { |
| rocketMQTemplate.asyncSend(topic, "payload", new SendCallback() { |
| @Override public void onSuccess(SendResult sendResult) { |
| |
| } |
| |
| @Override public void onException(Throwable e) { |
| |
| } |
| }); |
| } catch (MessagingException e) { |
| assertThat(e).hasMessageContaining("org.apache.rocketmq.remoting.exception.RemotingConnectException: connect to [127.0.0.1:9876] failed"); |
| } |
| |
| try { |
| rocketMQTemplate.syncSendOrderly(topic, "payload", "hashkey"); |
| } catch (MessagingException e) { |
| assertThat(e).hasMessageContaining("org.apache.rocketmq.remoting.exception.RemotingConnectException: connect to [127.0.0.1:9876] failed"); |
| } |
| } |
| |
| @Test |
| public void testProperties() { |
| assertThat(rocketMQTemplate.getProducer().getNamesrvAddr()).isEqualTo("127.0.0.1:9876"); |
| assertThat(rocketMQTemplate.getProducer().getProducerGroup()).isEqualTo("rocketMQTemplate-test-producer_group"); |
| assertThat(rocketMQTemplate.getProducer().getAccessChannel()).isEqualTo(AccessChannel.LOCAL); |
| assertThat(rocketMQTemplate.getProducer().getSendMsgTimeout()).isEqualTo(3500); |
| assertThat(rocketMQTemplate.getProducer().getMaxMessageSize()).isEqualTo(4 * 1024 * 1024); |
| assertThat(rocketMQTemplate.getProducer().getRetryTimesWhenSendAsyncFailed()).isEqualTo(3); |
| assertThat(rocketMQTemplate.getProducer().getRetryTimesWhenSendFailed()).isEqualTo(3); |
| assertThat(rocketMQTemplate.getProducer().getCompressMsgBodyOverHowmuch()).isEqualTo(1024 * 4); |
| } |
| |
| @Test |
| public void testTransactionListener() { |
| assertThat(((TransactionMQProducer) rocketMQTemplate.getProducer()).getTransactionListener()).isNotNull(); |
| } |
| } |
| |
| @RocketMQTransactionListener |
| class TransactionListenerImpl implements RocketMQLocalTransactionListener { |
| @Override |
| public RocketMQLocalTransactionState executeLocalTransaction(Message msg, Object arg) { |
| return RocketMQLocalTransactionState.UNKNOWN; |
| } |
| |
| @Override |
| public RocketMQLocalTransactionState checkLocalTransaction(Message msg) { |
| return RocketMQLocalTransactionState.COMMIT; |
| } |
| } |