diff options
author | Tor Brede Vekterli <vekterli@yahooinc.com> | 2023-04-19 14:50:26 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-19 14:50:26 +0200 |
commit | 6c98021a888d31632eeb2140c771b4a07a60ed73 (patch) | |
tree | 0b2b2ff0d0b02b314e9649431c26e550c2d32218 /documentapi/src/tests/messages/messages60test.cpp | |
parent | 7406547ddead0bf97c30a912b7873d2c5fbd9e3a (diff) | |
parent | 021caaf5074b7f2d561c15068e3f2d52ca0e36f2 (diff) |
Merge pull request #26782 from vespa-engine/havardpe/create-if-non-existent-for-document-put
add create-if-non-existent flag for document put
Diffstat (limited to 'documentapi/src/tests/messages/messages60test.cpp')
-rw-r--r-- | documentapi/src/tests/messages/messages60test.cpp | 42 |
1 files changed, 39 insertions, 3 deletions
diff --git a/documentapi/src/tests/messages/messages60test.cpp b/documentapi/src/tests/messages/messages60test.cpp index 258e8821e0c..93c7b5ef7cb 100644 --- a/documentapi/src/tests/messages/messages60test.cpp +++ b/documentapi/src/tests/messages/messages60test.cpp @@ -15,6 +15,17 @@ using document::DataType; using document::DocumentTypeRepo; +template <typename T> +struct Unwrap { + mbus::Routable::UP value; + const T *ptr = nullptr; + explicit Unwrap(mbus::Routable::UP value_in) : value(std::move(value_in)) { + ptr = dynamic_cast<T*>(value.get()); + ASSERT_TRUE(ptr != nullptr); + } + const T *operator->() const noexcept { return ptr; } +}; + /////////////////////////////////////////////////////////////////////////////// // // Setup @@ -399,10 +410,12 @@ Messages60Test::testPutDocumentMessage() EXPECT_EQUAL(sizeof(vespalib::string), sizeof(TestAndSetCondition)); EXPECT_EQUAL(112u, sizeof(DocumentMessage)); EXPECT_EQUAL(sizeof(TestAndSetCondition) + sizeof(DocumentMessage), sizeof(TestAndSetMessage)); - EXPECT_EQUAL(sizeof(TestAndSetMessage) + 24, sizeof(PutDocumentMessage)); + EXPECT_EQUAL(sizeof(TestAndSetMessage) + 32, sizeof(PutDocumentMessage)); + int size_of_create_if_non_existent_flag = 1; EXPECT_EQUAL(MESSAGE_BASE_LENGTH + 45u + - serializedLength(msg.getCondition().getSelection()), + serializedLength(msg.getCondition().getSelection()) + + size_of_create_if_non_existent_flag, serialize("PutDocumentMessage", msg)); for (uint32_t lang = 0; lang < NUM_LANGUAGES; ++lang) { @@ -413,11 +426,34 @@ Messages60Test::testPutDocumentMessage() EXPECT_EQUAL(msg.getDocument().getType().getName(), deserializedMsg.getDocument().getType().getName()); EXPECT_EQUAL(msg.getDocument().getId().toString(), deserializedMsg.getDocument().getId().toString()); EXPECT_EQUAL(msg.getTimestamp(), deserializedMsg.getTimestamp()); - EXPECT_EQUAL(71u, deserializedMsg.getApproxSize()); + EXPECT_EQUAL(72u, deserializedMsg.getApproxSize()); EXPECT_EQUAL(msg.getCondition().getSelection(), deserializedMsg.getCondition().getSelection()); + EXPECT_EQUAL(false, deserializedMsg.get_create_if_non_existent()); } } + //------------------------------------------------------------------------- + + PutDocumentMessage msg2(createDoc(getTypeRepo(), "testdoc", "id:ns:testdoc::")); + msg2.set_create_if_non_existent(true); + uint32_t expected_message_size = MESSAGE_BASE_LENGTH + 45u + + serializedLength(msg2.getCondition().getSelection()) + + size_of_create_if_non_existent_flag; + auto trunc1 = [](mbus::Blob x) noexcept { return truncate(std::move(x), 1); }; + auto pad1 = [](mbus::Blob x) noexcept { return pad(std::move(x), 1); }; + EXPECT_EQUAL(expected_message_size, serialize("PutDocumentMessage-create", msg2)); + EXPECT_EQUAL(expected_message_size - 1, serialize("PutDocumentMessage-create-truncate", msg2, trunc1)); + EXPECT_EQUAL(expected_message_size + 1, serialize("PutDocumentMessage-create-pad", msg2, pad1)); + + for (uint32_t lang = 0; lang < NUM_LANGUAGES; ++lang) { + auto decoded = Unwrap<PutDocumentMessage>(deserialize("PutDocumentMessage-create", DocumentProtocol::MESSAGE_PUTDOCUMENT, lang)); + auto decoded_trunc = Unwrap<PutDocumentMessage>(deserialize("PutDocumentMessage-create-truncate", DocumentProtocol::MESSAGE_PUTDOCUMENT, lang)); + auto decoded_pad = Unwrap<PutDocumentMessage>(deserialize("PutDocumentMessage-create-pad", DocumentProtocol::MESSAGE_PUTDOCUMENT, lang)); + EXPECT_EQUAL(true, decoded->get_create_if_non_existent()); + EXPECT_EQUAL(false, decoded_trunc->get_create_if_non_existent()); + EXPECT_EQUAL(true, decoded_pad->get_create_if_non_existent()); + } + return true; } |