Tag M4 RC9M4

git-svn-id: https://svn.apache.org/repos/asf/qpid/tags/M4@734202 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 224 insertions, 0 deletions

diff --git a/RC9/qpid/cpp/src/tests/MessageBuilderTest.cpp b/RC9/qpid/cpp/src/tests/MessageBuilderTest.cpp
new file mode 100644
index 0000000000..313a91c053
--- /dev/null
+++ b/RC9/qpid/cpp/src/tests/MessageBuilderTest.cpp
@@ -0,0 +1,224 @@
+/*
+ *
+ * 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 "qpid/broker/Message.h"
+#include "qpid/broker/MessageBuilder.h"
+#include "qpid/broker/NullMessageStore.h"
+#include "qpid/framing/frame_functors.h"
+#include "qpid/framing/MessageTransferBody.h"
+#include "qpid/framing/TypeFilter.h"
+#include "unit_test.h"
+#include <list>
+
+using namespace boost;
+using namespace qpid::broker;
+using namespace qpid::framing;
+using namespace qpid::sys;
+
+class MockMessageStore : public NullMessageStore
+{
+    enum Op {STAGE=1, APPEND=2};
+
+    uint64_t id;
+    intrusive_ptr<PersistableMessage> expectedMsg;        
+    string expectedData;
+    std::list<Op> ops;
+        
+    void checkExpectation(Op actual)
+    {
+        BOOST_CHECK_EQUAL(ops.front(), actual);
+        ops.pop_front();
+    }
+
+  public:
+    MockMessageStore() : id(0), expectedMsg(0) {}
+
+    void expectStage(PersistableMessage& msg) 
+    { 
+        expectedMsg = &msg;
+        ops.push_back(STAGE); 
+    }
+
+    void expectAppendContent(PersistableMessage& msg, const string& data) 
+    { 
+        expectedMsg = &msg;
+        expectedData = data;
+        ops.push_back(APPEND); 
+    }
+
+    void stage(const intrusive_ptr<PersistableMessage>& msg)
+    {
+        checkExpectation(STAGE);
+        BOOST_CHECK_EQUAL(expectedMsg, msg);
+        msg->setPersistenceId(++id);
+    }
+
+    void appendContent(const intrusive_ptr<const PersistableMessage>& msg,
+                       const string& data)
+    {
+        checkExpectation(APPEND);
+        BOOST_CHECK_EQUAL(static_pointer_cast<const PersistableMessage>(expectedMsg), msg);
+        BOOST_CHECK_EQUAL(expectedData, data);            
+    }
+
+    bool expectationsMet()
+    {
+        return ops.empty();
+    }
+
+    //don't treat this store as a null impl
+    bool isNull() const
+    {
+        return false;
+    }
+
+};
+    
+QPID_AUTO_TEST_SUITE(MessageBuilderTestSuite)
+
+QPID_AUTO_TEST_CASE(testHeaderOnly)
+{
+    MessageBuilder builder(0, 0);
+    builder.start(SequenceNumber());
+
+    std::string exchange("builder-exchange");
+    std::string key("builder-exchange");
+
+    AMQFrame method(in_place<MessageTransferBody>(
+                        ProtocolVersion(), exchange, 0, 0));
+    AMQFrame header(in_place<AMQHeaderBody>());
+
+    header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(0);        
+    header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key);
+
+    builder.handle(method);
+    builder.handle(header);
+
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK_EQUAL(exchange, builder.getMessage()->getExchangeName());
+    BOOST_CHECK_EQUAL(key, builder.getMessage()->getRoutingKey());
+    BOOST_CHECK(builder.getMessage()->getFrames().isComplete());
+}
+
+QPID_AUTO_TEST_CASE(test1ContentFrame)
+{
+    MessageBuilder builder(0, 0);
+    builder.start(SequenceNumber());
+
+    std::string data("abcdefg");
+    std::string exchange("builder-exchange");
+    std::string key("builder-exchange");
+
+    AMQFrame method(in_place<MessageTransferBody>(ProtocolVersion(), exchange, 0, 0));
+    AMQFrame header(in_place<AMQHeaderBody>());
+    AMQFrame content(in_place<AMQContentBody>(data));
+    method.setEof(false);
+    header.setBof(false);
+    header.setEof(false);
+    content.setBof(false);
+
+    header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data.size());        
+    header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key);
+
+    builder.handle(method);
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK(!builder.getMessage()->getFrames().isComplete());
+
+    builder.handle(header);
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK(!builder.getMessage()->getFrames().isComplete());
+
+    builder.handle(content);        
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK(builder.getMessage()->getFrames().isComplete());
+}
+
+QPID_AUTO_TEST_CASE(test2ContentFrames)
+{
+    MessageBuilder builder(0, 0);
+    builder.start(SequenceNumber());
+
+    std::string data1("abcdefg");
+    std::string data2("hijklmn");
+    std::string exchange("builder-exchange");
+    std::string key("builder-exchange");
+
+    AMQFrame method(in_place<MessageTransferBody>(
+                        ProtocolVersion(), exchange, 0, 0));
+    AMQFrame header(in_place<AMQHeaderBody>());
+    AMQFrame content1(in_place<AMQContentBody>(data1));
+    AMQFrame content2(in_place<AMQContentBody>(data2));
+    method.setEof(false);
+    header.setBof(false);
+    header.setEof(false);
+    content1.setBof(false);
+    content1.setEof(false);
+    content2.setBof(false);
+
+    header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size());        
+    header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key);
+
+    builder.handle(method);
+    builder.handle(header);
+    builder.handle(content1);
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK(!builder.getMessage()->getFrames().isComplete());
+
+    builder.handle(content2);
+    BOOST_CHECK(builder.getMessage());
+    BOOST_CHECK(builder.getMessage()->getFrames().isComplete());
+}
+
+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");
+    std::string key("builder-exchange");
+
+    AMQFrame method(in_place<MessageTransferBody>(
+                        ProtocolVersion(), exchange, 0, 0));
+    AMQFrame header(in_place<AMQHeaderBody>());
+    AMQFrame content1(in_place<AMQContentBody>(data1));
+    AMQFrame content2(in_place<AMQContentBody>(data2));
+
+    header.castBody<AMQHeaderBody>()->get<MessageProperties>(true)->setContentLength(data1.size() + data2.size());        
+    header.castBody<AMQHeaderBody>()->get<DeliveryProperties>(true)->setRoutingKey(key);
+
+    builder.handle(method);
+    builder.handle(header);
+
+    store.expectStage(*builder.getMessage());
+    builder.handle(content1);
+    BOOST_CHECK(store.expectationsMet());
+    BOOST_CHECK_EQUAL((uint64_t) 1, builder.getMessage()->getPersistenceId());
+
+    store.expectAppendContent(*builder.getMessage(), data2);
+    builder.handle(content2);
+    BOOST_CHECK(store.expectationsMet());
+    //were the content frames dropped?
+    BOOST_CHECK(!builder.getMessage()->isContentLoaded());
+}
+
+QPID_AUTO_TEST_SUITE_END()