Repository: incubator-rocketmq Updated Branches: refs/heads/ROCKETMQ-51 d87223f8c -> 8d2cf163a
ROCKETMQ-51] Add unit tests for ProducerManager Project: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/commit/4a79d9c7 Tree: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/tree/4a79d9c7 Diff: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/diff/4a79d9c7 Branch: refs/heads/ROCKETMQ-51 Commit: 4a79d9c7d6d56780cd6ac23d446934c7570b9d12 Parents: d87223f Author: yukon <yu...@apache.org> Authored: Sat Jan 21 22:01:18 2017 +0800 Committer: yukon <yu...@apache.org> Committed: Sat Jan 21 22:01:18 2017 +0800 ---------------------------------------------------------------------- .../broker/client/ProducerManagerTest.java | 90 ++++++++++++++++++++ 1 file changed, 90 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/4a79d9c7/broker/src/test/java/org/apache/rocketmq/broker/client/ProducerManagerTest.java ---------------------------------------------------------------------- diff --git a/broker/src/test/java/org/apache/rocketmq/broker/client/ProducerManagerTest.java b/broker/src/test/java/org/apache/rocketmq/broker/client/ProducerManagerTest.java new file mode 100644 index 0000000..08dbb9c --- /dev/null +++ b/broker/src/test/java/org/apache/rocketmq/broker/client/ProducerManagerTest.java @@ -0,0 +1,90 @@ +/* + * 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.broker.client; + +import io.netty.channel.Channel; +import io.netty.channel.ChannelFuture; +import java.lang.reflect.Field; +import java.util.HashMap; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import static org.assertj.core.api.Assertions.assertThat; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + +@RunWith(MockitoJUnitRunner.class) +public class ProducerManagerTest { + private ProducerManager producerManager; + private String group = "FooBar"; + private ClientChannelInfo clientInfo; + + @Mock + private Channel channel; + + @Before + public void init() { + producerManager = new ProducerManager(); + clientInfo = new ClientChannelInfo(channel); + } + + @Test + public void scanNotActiveChannel() throws Exception { + producerManager.registerProducer(group, clientInfo); + assertThat(producerManager.getGroupChannelTable().get(group).get(channel)).isNotNull(); + + Field field = ProducerManager.class.getDeclaredField("CHANNEL_EXPIRED_TIMEOUT"); + field.setAccessible(true); + long CHANNEL_EXPIRED_TIMEOUT = field.getLong(producerManager); + clientInfo.setLastUpdateTimestamp(System.currentTimeMillis() - CHANNEL_EXPIRED_TIMEOUT - 10); + when(channel.close()).thenReturn(mock(ChannelFuture.class)); + producerManager.scanNotActiveChannel(); + assertThat(producerManager.getGroupChannelTable().get(group).get(channel)).isNull(); + } + + @Test + public void doChannelCloseEvent() throws Exception { + producerManager.registerProducer(group, clientInfo); + assertThat(producerManager.getGroupChannelTable().get(group).get(channel)).isNotNull(); + producerManager.doChannelCloseEvent("127.0.0.1", channel); + assertThat(producerManager.getGroupChannelTable().get(group).get(channel)).isNull(); + } + + @Test + public void testRegisterProducer() throws Exception { + producerManager.registerProducer(group, clientInfo); + HashMap<Channel, ClientChannelInfo> channelMap = producerManager.getGroupChannelTable().get(group); + assertThat(channelMap).isNotNull(); + assertThat(channelMap.get(channel)).isEqualTo(clientInfo); + } + + @Test + public void unregisterProducer() throws Exception { + producerManager.registerProducer(group, clientInfo); + HashMap<Channel, ClientChannelInfo> channelMap = producerManager.getGroupChannelTable().get(group); + assertThat(channelMap).isNotNull(); + assertThat(channelMap.get(channel)).isEqualTo(clientInfo); + + producerManager.unregisterProducer(group, clientInfo); + channelMap = producerManager.getGroupChannelTable().get(group); + assertThat(channelMap).isNull(); + } + +} \ No newline at end of file