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/testbase.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/testbase.cpp')
-rw-r--r-- | documentapi/src/tests/messages/testbase.cpp | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/documentapi/src/tests/messages/testbase.cpp b/documentapi/src/tests/messages/testbase.cpp index db2c08704e1..ddce9f82a5d 100644 --- a/documentapi/src/tests/messages/testbase.cpp +++ b/documentapi/src/tests/messages/testbase.cpp @@ -76,6 +76,24 @@ TestBase::putTest(uint32_t type, TEST_METHOD_PT test) return *this; } +mbus::Blob +TestBase::truncate(mbus::Blob data, size_t bytes) +{ + ASSERT_GREATER(data.size(), bytes); + mbus::Blob res(data.size() - bytes); + memcpy(res.data(), data.data(), res.size()); + return res; +} + +mbus::Blob +TestBase::pad(mbus::Blob data, size_t bytes) +{ + mbus::Blob res(data.size() + bytes); + memset(res.data(), 0, res.size()); + memcpy(res.data(), data.data(), data.size()); + return res; +} + bool TestBase::testCoverage(const std::vector<uint32_t> &expected, const std::vector<uint32_t> &actual, bool report) const { @@ -119,13 +137,13 @@ bool TestBase::file_content_is_unchanged(const string& filename, const mbus::Blo } uint32_t -TestBase::serialize(const string &filename, const mbus::Routable &routable) +TestBase::serialize(const string &filename, const mbus::Routable &routable, Tamper tamper) { const vespalib::Version version = getVersion(); string path = getPath(version.toString() + "-cpp-" + filename + ".dat"); LOG(info, "Serializing to '%s'..", path.c_str()); - mbus::Blob blob = _protocol.encode(version, routable); + mbus::Blob blob = tamper(_protocol.encode(version, routable)); if (file_content_is_unchanged(path, blob)) { LOG(info, "Serialization for '%s' is unchanged; not overwriting it", path.c_str()); } else if (!EXPECT_TRUE(writeFile(path, blob))) { @@ -213,5 +231,3 @@ TestBase::readFile(const string &filename) const return blob; } - - |