diff options
Diffstat (limited to 'qpid/cpp/src/tests/MessageBuilderTest.cpp')
-rw-r--r-- | qpid/cpp/src/tests/MessageBuilderTest.cpp | 46 |
1 files changed, 26 insertions, 20 deletions
diff --git a/qpid/cpp/src/tests/MessageBuilderTest.cpp b/qpid/cpp/src/tests/MessageBuilderTest.cpp index 1f3f830633..c2fb8ad32e 100644 --- a/qpid/cpp/src/tests/MessageBuilderTest.cpp +++ b/qpid/cpp/src/tests/MessageBuilderTest.cpp @@ -7,9 +7,9 @@ * 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 @@ -31,15 +31,18 @@ using namespace qpid::broker; using namespace qpid::framing; using namespace qpid::sys; +namespace qpid { +namespace tests { + class MockMessageStore : public NullMessageStore { enum Op {STAGE=1, APPEND=2}; uint64_t id; - boost::intrusive_ptr<PersistableMessage> expectedMsg; + boost::intrusive_ptr<PersistableMessage> expectedMsg; string expectedData; std::list<Op> ops; - + void checkExpectation(Op actual) { BOOST_CHECK_EQUAL(ops.front(), actual); @@ -49,17 +52,17 @@ class MockMessageStore : public NullMessageStore public: MockMessageStore() : id(0), expectedMsg(0) {} - void expectStage(PersistableMessage& msg) - { + void expectStage(PersistableMessage& msg) + { expectedMsg = &msg; - ops.push_back(STAGE); + ops.push_back(STAGE); } - void expectAppendContent(PersistableMessage& msg, const string& data) - { + void expectAppendContent(PersistableMessage& msg, const string& data) + { expectedMsg = &msg; expectedData = data; - ops.push_back(APPEND); + ops.push_back(APPEND); } void stage(const boost::intrusive_ptr<PersistableMessage>& msg) @@ -74,7 +77,7 @@ class MockMessageStore : public NullMessageStore { checkExpectation(APPEND); BOOST_CHECK_EQUAL(boost::static_pointer_cast<const PersistableMessage>(expectedMsg), msg); - BOOST_CHECK_EQUAL(expectedData, data); + BOOST_CHECK_EQUAL(expectedData, data); } bool expectationsMet() @@ -89,7 +92,7 @@ class MockMessageStore : public NullMessageStore } }; - + QPID_AUTO_TEST_SUITE(MessageBuilderTestSuite) QPID_AUTO_TEST_CASE(testHeaderOnly) @@ -103,7 +106,7 @@ QPID_AUTO_TEST_CASE(testHeaderOnly) AMQFrame method((MessageTransferBody(ProtocolVersion(), exchange, 0, 0))); AMQFrame header((AMQHeaderBody())); - header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(0); + header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(0); header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key); builder.handle(method); @@ -132,7 +135,7 @@ QPID_AUTO_TEST_CASE(test1ContentFrame) header.setEof(false); content.setBof(false); - header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data.size()); + header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data.size()); header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key); builder.handle(method); @@ -143,7 +146,7 @@ QPID_AUTO_TEST_CASE(test1ContentFrame) BOOST_CHECK(builder.getMessage()); BOOST_CHECK(!builder.getMessage()->getFrames().isComplete()); - builder.handle(content); + builder.handle(content); BOOST_CHECK(builder.getMessage()); BOOST_CHECK(builder.getMessage()->getFrames().isComplete()); } @@ -169,7 +172,7 @@ QPID_AUTO_TEST_CASE(test2ContentFrames) content1.setEof(false); content2.setBof(false); - header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size()); + header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size()); header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key); builder.handle(method); @@ -188,7 +191,7 @@ QPID_AUTO_TEST_CASE(testStaging) MockMessageStore store; MessageBuilder builder(&store, 5); builder.start(SequenceNumber()); - + std::string data1("abcdefg"); std::string data2("hijklmn"); std::string exchange("builder-exchange"); @@ -199,7 +202,7 @@ QPID_AUTO_TEST_CASE(testStaging) AMQFrame content1((AMQContentBody(data1))); AMQFrame content2((AMQContentBody(data2))); - header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size()); + header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size()); header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key); builder.handle(method); @@ -223,7 +226,7 @@ QPID_AUTO_TEST_CASE(testNoManagementStaging) MockMessageStore store; MessageBuilder builder(&store, 5); builder.start(SequenceNumber()); - + std::string data1("abcdefg"); std::string exchange("qpid.management"); std::string key("builder-exchange"); @@ -232,7 +235,7 @@ QPID_AUTO_TEST_CASE(testNoManagementStaging) AMQFrame header((AMQHeaderBody())); AMQFrame content1((AMQContentBody(data1))); - header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size()); + header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size()); header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key); builder.handle(method); @@ -242,4 +245,7 @@ QPID_AUTO_TEST_CASE(testNoManagementStaging) BOOST_CHECK(store.expectationsMet()); BOOST_CHECK_EQUAL((uint64_t) 0, builder.getMessage()->getPersistenceId()); } + QPID_AUTO_TEST_SUITE_END() + +}} // namespace qpid::tests |