| /* |
| * |
| * 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. |
| * |
| */ |
| #include "unit_test.h" |
| #include "BrokerFixture.h" |
| #include "qpid/client/MessageReplayTracker.h" |
| #include "qpid/sys/Time.h" |
| |
| namespace qpid { |
| namespace tests { |
| |
| QPID_AUTO_TEST_SUITE(MessageReplayTrackerTests) |
| |
| using namespace qpid::client; |
| using namespace qpid::sys; |
| using std::string; |
| |
| class ReplayBufferChecker |
| { |
| public: |
| |
| ReplayBufferChecker(uint from, uint to) : end(to), i(from) {} |
| |
| void operator()(const Message& m) |
| { |
| if (i > end) BOOST_FAIL("Extra message found: " + m.getData()); |
| BOOST_CHECK_EQUAL((boost::format("Message_%1%") % (i++)).str(), m.getData()); |
| } |
| private: |
| const uint end; |
| uint i; |
| |
| }; |
| |
| QPID_AUTO_TEST_CASE(testReplay) |
| { |
| SessionFixture fix; |
| fix.session.queueDeclare(arg::queue="my-queue", arg::exclusive=true, arg::autoDelete=true); |
| |
| MessageReplayTracker tracker(10); |
| tracker.init(fix.session); |
| for (uint i = 0; i < 5; i++) { |
| Message message((boost::format("Message_%1%") % (i+1)).str(), "my-queue"); |
| tracker.send(message); |
| } |
| ReplayBufferChecker checker(1, 10); |
| tracker.foreach(checker); |
| |
| tracker.replay(fix.session); |
| for (uint j = 0; j < 2; j++) {//each message should have been sent twice |
| for (uint i = 0; i < 5; i++) { |
| Message m; |
| BOOST_CHECK(fix.subs.get(m, "my-queue", TIME_SEC)); |
| BOOST_CHECK_EQUAL((boost::format("Message_%1%") % (i+1)).str(), m.getData()); |
| } |
| } |
| Message m; |
| BOOST_CHECK(!fix.subs.get(m, "my-queue")); |
| } |
| |
| QPID_AUTO_TEST_CASE(testCheckCompletion) |
| { |
| SessionFixture fix; |
| fix.session.queueDeclare(arg::queue="my-queue", arg::exclusive=true, arg::autoDelete=true); |
| |
| MessageReplayTracker tracker(10); |
| tracker.init(fix.session); |
| for (uint i = 0; i < 5; i++) { |
| Message message((boost::format("Message_%1%") % (i+1)).str(), "my-queue"); |
| tracker.send(message); |
| } |
| fix.session.sync();//ensures all messages are complete |
| tracker.checkCompletion(); |
| tracker.replay(fix.session); |
| Message received; |
| for (uint i = 0; i < 5; i++) { |
| BOOST_CHECK(fix.subs.get(received, "my-queue")); |
| BOOST_CHECK_EQUAL((boost::format("Message_%1%") % (i+1)).str(), received.getData()); |
| } |
| BOOST_CHECK(!fix.subs.get(received, "my-queue")); |
| } |
| |
| QPID_AUTO_TEST_SUITE_END() |
| |
| }} // namespace qpid::tests |