blob: 9a0036aef38d8f76bd4c045fbd36d6f900cb0401 [file] [log] [blame]
/*
* 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 "proton/message.hpp"
#include "proton/scalar.hpp"
#include "test_bits.hpp"
#include <string>
#include <fstream>
#include <streambuf>
#include <iosfwd>
namespace {
using namespace std;
using namespace proton;
#define CHECK_STR(ATTR) \
m.ATTR(#ATTR); \
ASSERT_EQUAL(std::string(#ATTR), m.ATTR())
#define CHECK_MESSAGE_ID(ATTR) \
m.ATTR(#ATTR); \
ASSERT_EQUAL(scalar(#ATTR), m.ATTR())
void test_message_defaults() {
message m;
ASSERT(m.body().empty());
ASSERT(m.id().empty());
ASSERT(m.user().empty());
ASSERT(m.to().empty());
ASSERT(m.subject().empty());
ASSERT(m.reply_to().empty());
ASSERT(m.correlation_id().empty());
ASSERT(m.content_type().empty());
ASSERT(m.content_encoding().empty());
ASSERT(m.group_id().empty());
ASSERT(m.reply_to_group_id().empty());
ASSERT_EQUAL(0, m.expiry_time().milliseconds());
ASSERT_EQUAL(0, m.creation_time().milliseconds());
ASSERT_EQUAL(false, m.inferred());
ASSERT_EQUAL(false, m.durable());
ASSERT_EQUAL(0, m.ttl().milliseconds());
ASSERT_EQUAL(message::default_priority, m.priority());
ASSERT_EQUAL(false, m.first_acquirer());
ASSERT_EQUAL(0u, m.delivery_count());
}
void test_message_properties() {
message m("hello");
std::string s = get<std::string>(m.body());
ASSERT_EQUAL("hello", s);
CHECK_MESSAGE_ID(id);
CHECK_STR(user);
CHECK_STR(to);
CHECK_STR(subject);
CHECK_STR(reply_to);
CHECK_MESSAGE_ID(correlation_id);
CHECK_STR(content_type);
CHECK_STR(content_encoding);
CHECK_STR(group_id);
CHECK_STR(reply_to_group_id);
m.expiry_time(timestamp(42));
ASSERT_EQUAL(m.expiry_time().milliseconds(), 42);
m.creation_time(timestamp(4242));
ASSERT_EQUAL(m.creation_time().milliseconds(), 4242);
m.ttl(duration(30));
ASSERT_EQUAL(m.ttl().milliseconds(), 30);
m.priority(3);
ASSERT_EQUAL(m.priority(), 3);
message m2(m);
ASSERT_EQUAL("hello", get<std::string>(m2.body()));
ASSERT_EQUAL(message_id("id"), m2.id());
ASSERT_EQUAL("user", m2.user());
ASSERT_EQUAL("to", m2.to());
ASSERT_EQUAL("subject", m2.subject());
ASSERT_EQUAL("reply_to", m2.reply_to());
ASSERT_EQUAL(message_id("correlation_id"), m2.correlation_id());
ASSERT_EQUAL("content_type", m2.content_type());
ASSERT_EQUAL("content_encoding", m2.content_encoding());
ASSERT_EQUAL("group_id", m2.group_id());
ASSERT_EQUAL("reply_to_group_id", m2.reply_to_group_id());
ASSERT_EQUAL(42, m2.expiry_time().milliseconds());
ASSERT_EQUAL(4242, m.creation_time().milliseconds());
m2 = m;
ASSERT_EQUAL("hello", get<std::string>(m2.body()));
ASSERT_EQUAL(message_id("id"), m2.id());
ASSERT_EQUAL("user", m2.user());
ASSERT_EQUAL("to", m2.to());
ASSERT_EQUAL("subject", m2.subject());
ASSERT_EQUAL("reply_to", m2.reply_to());
ASSERT_EQUAL(message_id("correlation_id"), m2.correlation_id());
ASSERT_EQUAL("content_type", m2.content_type());
ASSERT_EQUAL("content_encoding", m2.content_encoding());
ASSERT_EQUAL("group_id", m2.group_id());
ASSERT_EQUAL("reply_to_group_id", m2.reply_to_group_id());
ASSERT_EQUAL(42, m2.expiry_time().milliseconds());
ASSERT_EQUAL(4242, m.creation_time().milliseconds());
}
void test_message_body() {
std::string s("hello");
message m1(s.c_str());
ASSERT_EQUAL(s, get<std::string>(m1.body()));
message m2(s);
ASSERT_EQUAL(s, coerce<std::string>(m2.body()));
message m3;
m3.body(s);
ASSERT_EQUAL(s, coerce<std::string>(m3.body()));
ASSERT_EQUAL(5, coerce<int64_t>(message(5).body()));
ASSERT_EQUAL(3.1, coerce<double>(message(3.1).body()));
}
void test_message_maps() {
message m;
ASSERT(m.properties().empty());
ASSERT(m.message_annotations().empty());
ASSERT(m.delivery_annotations().empty());
m.properties().put("foo", 12);
m.delivery_annotations().put("bar", "xyz");
m.message_annotations().put(23, int8_t(42));
ASSERT_EQUAL(m.properties().get("foo"), scalar(12));
ASSERT_EQUAL(m.delivery_annotations().get("bar"), scalar("xyz"));
ASSERT_EQUAL(m.message_annotations().get(23), scalar(int8_t(42)));
ASSERT_EQUAL(proton::get<int8_t>(m.message_annotations().get(23)), 42);
ASSERT_EQUAL(m.message_annotations().get(23).get<int8_t>(), 42);
message m2(m);
ASSERT_EQUAL(m.properties().get("foo"), scalar(12)); // Decoding shouldn't change it
ASSERT_EQUAL(m2.properties().get("foo"), scalar(12));
ASSERT_EQUAL(m2.delivery_annotations().get("bar"), scalar("xyz"));
ASSERT_EQUAL(m2.message_annotations().get(23), scalar(int8_t(42)));
m.properties().put("foo","newfoo");
m.delivery_annotations().put(24, 1000);
m.message_annotations().erase(23);
message m3 = m;
size_t size = m3.properties().size();
ASSERT_EQUAL(1u, size);
ASSERT_EQUAL(m3.properties().get("foo"), scalar("newfoo"));
ASSERT_EQUAL(2u, m3.delivery_annotations().size());
ASSERT_EQUAL(m3.delivery_annotations().get("bar"), scalar("xyz"));
ASSERT_EQUAL(m3.delivery_annotations().get(24), scalar(1000));
ASSERT(m3.message_annotations().empty());
// PROTON-1498
message msg;
msg.message_annotations().put("x-opt-jms-msg-type", int8_t(1));
proton::message::annotation_map& am_ref = msg.message_annotations();
uint8_t t = am_ref.get(proton::symbol("x-opt-jms-msg-type")).get<int8_t>();
ASSERT_EQUAL(1, t);
proton::message::annotation_map am_val = msg.message_annotations();
t = am_val.get(proton::symbol("x-opt-jms-msg-type")).get<int8_t>();
ASSERT_EQUAL(1, t);
}
void test_message_reuse() {
message m1("one");
m1.properties().put("x", "y");
message m2("two");
m2.properties().put("a", "b");
m1.decode(m2.encode()); // Use m1 for a newly decoded message
ASSERT_EQUAL(value("two"), m1.body());
ASSERT_EQUAL(value("b"), m1.properties().get("a"));
}
void test_message_print() {
message m("hello");
m.to("to");
m.user("user");
m.subject("subject");
m.content_type("weird");
m.correlation_id(23);
m.properties().put("foo", "bar");
m.properties().put("num", 9);
m.delivery_annotations().put("deliverme", "please");
ASSERT_EQUAL("Message{address=\"to\", user_id=\"user\", subject=\"subject\", correlation_id=23, content_type=\"weird\", body=\"hello\"}", to_string(m));
}
}
int main(int, char**) {
int failed = 0;
RUN_TEST(failed, test_message_properties());
RUN_TEST(failed, test_message_defaults());
RUN_TEST(failed, test_message_body());
RUN_TEST(failed, test_message_maps());
RUN_TEST(failed, test_message_reuse());
RUN_TEST(failed, test_message_print());
return failed;
}