diff options
Diffstat (limited to 'cpp/test/unit/qpid/broker/MessageBuilderTest.cpp')
| -rw-r--r-- | cpp/test/unit/qpid/broker/MessageBuilderTest.cpp | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/cpp/test/unit/qpid/broker/MessageBuilderTest.cpp b/cpp/test/unit/qpid/broker/MessageBuilderTest.cpp index a5f7911fc8..fa80f8f939 100644 --- a/cpp/test/unit/qpid/broker/MessageBuilderTest.cpp +++ b/cpp/test/unit/qpid/broker/MessageBuilderTest.cpp @@ -56,13 +56,19 @@ class MessageBuilderTest : public CppUnit::TestCase header = new Buffer(msg->encodedHeaderSize()); msg->encodeHeader(*header); content = new Buffer(contentBufferSize); - msg->encodeContent(*content); - } else if (!header || !content) { - throw qpid::Exception("Buffers not initialised!"); + msg->setPersistenceId(1); } else { - msg->encodeContent(*content); + throw qpid::Exception("Message already staged!"); + } + } + + void appendContent(u_int64_t msgId, const string& data) + { + if (msgId == 1) { + content->putRawData(data); + } else { + throw qpid::Exception("Invalid message id!"); } - msg->setPersistenceId(1); } Message::shared_ptr getRestoredMessage() @@ -159,7 +165,7 @@ class MessageBuilderTest : public CppUnit::TestCase void testStaging(){ DummyHandler handler; - TestMessageStore store(50);//more than enough for two frames of 14 bytes + TestMessageStore store(14); MessageBuilder builder(&handler, &store, 5); string data1("abcdefg"); |