コード例 #1
0
  @Test
  public void testRenewLeaseFailed() throws Exception {
    long id = CorrelationIdGenerator.generateCorrelationId() + 1;

    ConsumerNotifier notifier = lookup(ConsumerNotifier.class);

    metaProxyActions4LeaseOperation(LeaseAnswer.SUCCESS, LeaseAnswer.SUCCESS);
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));

    TestMessageListener listener = new TestMessageListener().receiveCount(1);
    ConsumerHolder holder = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, listener);
    waitUntilConsumerStarted(holder);
    listener.waitUntilReceivedAllMessage();
    Assert.assertEquals(1, listener.getReceivedMessages().size());
    holder.close();
    listener.countDownAll();

    metaProxyActions4LeaseOperation(LeaseAnswer.SUCCESS, LeaseAnswer.FAILED);
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));

    listener = new TestMessageListener().receiveCount(1);
    ConsumerHolder holder_failed = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, listener);
    waitUntilConsumerStarted(holder_failed);
    listener.waitUntilReceivedAllMessage();
    Assert.assertNull(notifier.find(id));
    holder_failed.close();
    listener.countDownAll();
  }
コード例 #2
0
 private void doTestConsumeFailed() throws Exception {
   brokerActions4PollMessageCmd(
       PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));
   TestMessageListener listener = new TestMessageListener().receiveCount(1);
   ConsumerHolder holder = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, listener);
   waitUntilConsumerStarted(holder);
   listener.waitUntilReceivedAllMessage(300);
   holder.close();
   Assert.assertEquals(0, listener.getReceivedMessages().size());
   listener.countDownAll();
 }
コード例 #3
0
  @Test
  public void testGetEndpointFailed() throws Exception {
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));
    TestMessageListener listener = new TestMessageListener().receiveCount(1);
    ConsumerHolder holder = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, listener);
    long preCost = listener.waitUntilReceivedAllMessage();
    Assert.assertEquals(1, listener.getReceivedMessages().size());
    holder.close();
    listener.countDownAll();

    Meta meta = loadLocalMeta();
    meta.getEndpoints().clear();
    when(m_metaHolder.getMeta()).thenReturn(meta);

    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));
    listener = new TestMessageListener().receiveCount(1);
    holder = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, listener);
    listener.waitUntilReceivedAllMessage(preCost * 2);
    Assert.assertEquals(0, listener.getReceivedMessages().size());
    holder.close();
    listener.countDownAll();
  }
コード例 #4
0
  @Test
  public void testConsumeFailed() {
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));

    TestMessageListener msgListener = new TestMessageListener().receiveCount(1).withError(true);
    ConsumerHolder consumer = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, msgListener);

    msgListener.waitUntilReceivedAllMessage();

    consumer.close();

    Assert.assertEquals(1, msgListener.getReceivedMessages().size());
    Assert.assertEquals(TEST_MSG, msgListener.getReceivedMessages().get(0));
    msgListener.countDownAll();
  }
コード例 #5
0
  @Test
  public void testBasicMultipleMessageConsume() throws Exception {
    final int batchCount = 3;
    final int msgCountPerBatch = 5;
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC
            .channel(m_channel)
            .creator(
                new RawMessageCreator<TestObjectMessage>() {
                  @Override
                  public List<List<TestObjectMessage>> createRawMessages() {
                    List<List<TestObjectMessage>> batchs = new ArrayList<>();
                    for (int i = 0; i < batchCount; i++) {
                      List<TestObjectMessage> msgs = new ArrayList<>();
                      for (int j = 0; j < msgCountPerBatch; j++) {
                        msgs.add(
                            new TestObjectMessage(
                                "hermes.msg." + (i * msgCountPerBatch + j),
                                i * msgCountPerBatch + j,
                                new byte[] {17}));
                      }
                      batchs.add(msgs);
                    }
                    return batchs;
                  }
                }));

    TestMessageListener msgListener =
        new TestMessageListener().receiveCount(batchCount * msgCountPerBatch);
    ConsumerHolder consumer = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, msgListener);

    msgListener.waitUntilReceivedAllMessage();

    consumer.close();
    System.out.println(msgListener.getReceivedMessages());

    Assert.assertFalse(!((DefaultConsumerHolder) consumer).isConsuming());
    Assert.assertEquals(batchCount * msgCountPerBatch, msgListener.getReceivedMessages().size());
    for (int i = 0; i < batchCount; i++) {
      for (int j = 0; j < msgCountPerBatch; j++) {
        Assert.assertEquals(
            17,
            msgListener.getReceivedMessages().get(i * msgCountPerBatch + j).getByteArrayValue()[0]);
      }
    }
    msgListener.countDownAll();
  }
コード例 #6
0
  @Test
  public void testBasicConsume() throws Exception {
    brokerActions4PollMessageCmd(
        PullMessageAnswer.BASIC.channel(m_channel).creator(SIMPLE_CREATOR));

    TestMessageListener msgListener = new TestMessageListener().receiveCount(1);
    ConsumerHolder consumer = Consumer.getInstance().start(TEST_TOPIC, TEST_GROUP, msgListener);

    msgListener.waitUntilReceivedAllMessage();

    consumer.close();

    Assert.assertFalse(!((DefaultConsumerHolder) consumer).isConsuming());
    Assert.assertEquals(1, msgListener.getReceivedMessages().size());
    Assert.assertEquals(TEST_MSG, msgListener.getReceivedMessages().get(0));

    msgListener.countDownAll();
  }