/*
* (C) 2007-2012 Alibaba Group Holding Limited.
*
* Licensed 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.
* Authors:
* wuhua <wq163@163.com> , boyan <killme2008@gmail.com>
*/
package com.taobao.metamorphosis.client.extension.producer;
import java.util.concurrent.TimeUnit;
import org.easymock.classextension.EasyMock;
import org.easymock.classextension.IMocksControl;
import org.junit.Before;
import org.junit.Test;
import com.taobao.metamorphosis.Message;
import com.taobao.metamorphosis.cluster.Partition;
/**
*
* @author ��
* @since 2011-8-8 ����11:21:40
*/
public class OrderedMessageSenderUnitTest {
private OrderedMessageProducer producer;
private IMocksControl mocksControl;
private OrderedMessageSender sender;
private final Partition partition = new Partition("0-0");
@Before
public void setUp() {
this.mocksControl = EasyMock.createControl();
this.producer = this.mocksControl.createMock(OrderedMessageProducer.class);
this.sender = new OrderedMessageSender(this.producer);
}
@Test
public void testSendMessage_PartitionNumWrong() throws Exception {
// ������������,������Ϣ�洢������
final Message message = this.createDefaultMessage();
// EasyMock.expect(this.producer.getLocalMessageCount(message.getTopic(),
// this.partition)).andReturn(2);
EasyMock.expect(this.producer.selectPartition(message)).andThrow(
new AvailablePartitionNumException("xx[0-0]xx"));
EasyMock.expect(this.producer.saveMessageToLocal(message, this.partition, 10000, TimeUnit.MILLISECONDS))
.andReturn(null);
this.mocksControl.replay();
this.sender.sendMessage(message, 10000, TimeUnit.MILLISECONDS);
this.mocksControl.verify();
}
@Test
public void testSendMessage_swichToNomal() throws Exception {
final Message message = this.createDefaultMessage();
// �����Ѿ�����,���ػ������Ϣ����Ϊ0���л�Ϊ��������ģʽ�����ѱ�����Ϣд�������
EasyMock.expect(this.producer.getLocalMessageCount(message.getTopic(), this.partition)).andReturn(0);
EasyMock.expect(this.producer.selectPartition(message)).andReturn(new Partition("0-0"));
EasyMock.expect(this.producer.sendMessageToServer(message, 10000, TimeUnit.MILLISECONDS, true)).andReturn(null);
this.mocksControl.replay();
this.sender.sendMessage(message, 10000, TimeUnit.MILLISECONDS);
this.mocksControl.verify();
}
@Test
public void testSendMessage_PartitionNumRight_butHaveFewLocalMessage() throws Exception {
// �����Ѿ�����,���ػ�������������Ϣ��ͣ��һ���ټ��,����n�κ�������·���Ļ�,������Ϣд����
final Message message = this.createDefaultMessage();
EasyMock.expect(this.producer.getLocalMessageCount(message.getTopic(), this.partition)).andReturn(10).times(3);
EasyMock.expect(this.producer.selectPartition(message)).andReturn(new Partition("0-0")).times(3);
EasyMock.expect(this.producer.saveMessageToLocal(message, this.partition, 10000, TimeUnit.MILLISECONDS))
.andReturn(null);
this.producer.tryRecoverMessage(message.getTopic(), this.partition);
EasyMock.expectLastCall().times(3);
this.mocksControl.replay();
this.sender.sendMessage(message, 10000, TimeUnit.MILLISECONDS);
this.mocksControl.verify();
}
@Test
public void testSendMessage_PartitionNumRight_butHaveFewLocalMessage2() throws Exception {
// �����Ѿ�����,���ػ�������������Ϣ��ͣ��һ���ټ��,�ڶ��μ�������Ϣ�ָ����,������Ϣд������,�л�����������ģʽ
final Message message = this.createDefaultMessage();
EasyMock.expect(this.producer.getLocalMessageCount(message.getTopic(), this.partition)).andReturn(10);
EasyMock.expect(this.producer.getLocalMessageCount(message.getTopic(), this.partition)).andReturn(0);
EasyMock.expect(this.producer.selectPartition(message)).andReturn(new Partition("0-0")).times(2);
EasyMock.expect(this.producer.sendMessageToServer(message, 10000, TimeUnit.MILLISECONDS, true)).andReturn(null);
this.producer.tryRecoverMessage(message.getTopic(), this.partition);
EasyMock.expectLastCall().times(1);
this.mocksControl.replay();
this.sender.sendMessage(message, 10000, TimeUnit.MILLISECONDS);
this.mocksControl.verify();
}
@Test
public void testSendMessage_PartitionNumRight_butHaveManyLocalMessage() throws Exception {
}
private Message createDefaultMessage() {
final String topic = "topic1";
final byte[] data = "hello".getBytes();
final Message message = new Message(topic, data);
return message;
}
}