1 files changed, 198 insertions, 0 deletions
diff --git a/searchlib/src/tests/util/rawbuf_test.cpp b/searchlib/src/tests/util/rawbuf_test.cpp
new file mode 100644
index 00000000000..e9dc139bda5
--- /dev/null
+++ b/searchlib/src/tests/util/rawbuf_test.cpp
@@ -0,0 +1,198 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// Unit tests for rawbuf.
+
+#include <vespa/log/log.h>
+LOG_SETUP("rawbuf_test");
+#include <vespa/fastos/fastos.h>
+
+#include <vespa/searchlib/util/rawbuf.h>
+#include <vespa/vespalib/stllike/string.h>
+#include <vespa/vespalib/testkit/testapp.h>
+
+using vespalib::string;
+using namespace search;
+
+namespace {
+
+string getString(const RawBuf &buf) {
+    return string(buf.GetDrainPos(), buf.GetUsedLen());
+}
+
+TEST("require that rawbuf can append text") {
+    RawBuf buf(10);
+    buf += "foo";
+    buf += "bar";
+    EXPECT_EQUAL("foobar", getString(buf));
+}
+
+TEST("require that rawbuf expands when appended beyond size") {
+    RawBuf buf(4);
+    buf += "foo";
+    EXPECT_EQUAL(1u, buf.GetFreeLen());
+    buf += "bar";
+    EXPECT_EQUAL(2u, buf.GetFreeLen());
+    EXPECT_EQUAL("foobar", getString(buf));
+}
+
+TEST("require that a rawbuf can be appended to another") {
+    RawBuf buf1(10);
+    RawBuf buf2(10);
+    buf1 += "foo";
+    buf2 += "bar";
+    buf1 += buf2;
+    EXPECT_EQUAL("foobar", getString(buf1));
+}
+
+TEST("require that rawbufs can be tested for equality") {
+    RawBuf buf1(10);
+    RawBuf buf2(10);
+    buf1 += "foo";
+    buf2 += "bar";
+    EXPECT_TRUE(buf1 == buf1);
+    EXPECT_FALSE(buf1 == buf2);
+}
+
+template <typename T>
+void checkAddNum(void (RawBuf::*addNum)(T, size_t, char), size_t num,
+                 size_t fieldw, char fill, const string &expected) {
+    RawBuf buf(10);
+    (buf.*addNum)(num, fieldw, fill);
+    EXPECT_EQUAL(expected, getString(buf));
+}
+
+TEST("require that rawbuf can add numbers in decimal") {
+    checkAddNum(&RawBuf::addNum, 0, 4, 'x', "xxx0");
+    checkAddNum(&RawBuf::addNum, 42, 4, '0', "0042");
+    checkAddNum(&RawBuf::addNum, 12345678901234, 4, '0', "12345678901234");
+    checkAddNum(&RawBuf::addNum, -1, 4, '0', "18446744073709551615");
+
+    checkAddNum(&RawBuf::addNum32, 0, 4, 'x', "xxx0");
+    checkAddNum(&RawBuf::addNum32, 42, 4, '0', "0042");
+    checkAddNum(&RawBuf::addNum32, 1234567890, 4, '0', "1234567890");
+    checkAddNum(&RawBuf::addNum32, -1, 0, '0', "-1");
+    checkAddNum(&RawBuf::addNum32, -1, 4, '0', "00-1");
+
+    checkAddNum(&RawBuf::addNum64, 0, 4, 'x', "xxx0");
+    checkAddNum(&RawBuf::addNum64, 42, 4, '0', "0042");
+    checkAddNum(&RawBuf::addNum64, 12345678901234, 4, '0', "12345678901234");
+    checkAddNum(&RawBuf::addNum64, -1, 0, '0', "-1");
+    checkAddNum(&RawBuf::addNum64, -1, 4, '0', "00-1");
+}
+
+TEST("require that rawbuf can add hitrank") {
+    RawBuf buf(10);
+    buf.addHitRank(HitRank(4.2));
+    EXPECT_EQUAL("4.2", getString(buf));
+}
+
+TEST("require that rawbuf can add signedhitrank") {
+    RawBuf buf(10);
+    buf.addHitRank(SignedHitRank(-4.213));
+    EXPECT_EQUAL("-4.213", getString(buf));
+}
+
+TEST("require that rawbuf can append data of known length") {
+    RawBuf buf(10);
+    const string data("foo bar baz qux quux");
+    buf.append(data.data(), data.size());
+    EXPECT_EQUAL(data, getString(buf));
+}
+
+TEST("require that rawbuf can be truncated shorter and longer") {
+    RawBuf buf(10);
+    buf += "foobarbaz";
+    buf.truncate(3);
+    buf += "qux";
+    buf.truncate(9);
+    EXPECT_EQUAL("fooquxbaz", getString(buf));
+}
+
+TEST("require that prealloc makes enough room") {
+    RawBuf buf(10);
+    buf += "foo";
+    EXPECT_EQUAL(7u, buf.GetFreeLen());
+    buf.preAlloc(100);
+    EXPECT_EQUAL("foo", getString(buf));
+    EXPECT_LESS_EQUAL(100u, buf.GetFreeLen());
+}
+
+TEST("require that rawbuf can read from file") {
+    FastOS_File file("mytemporaryfile");
+    file.OpenReadWrite();
+    file.Write2("barbaz", 6);
+    file.SetPosition(0);
+
+    RawBuf buf(10);
+    buf += "foo";
+    buf.readFile(file, 3);
+    EXPECT_EQUAL("foobar", getString(buf));
+    buf.readFile(file, 100);
+    EXPECT_EQUAL("foobarbaz", getString(buf));
+
+    file.Close();
+    file.Delete();
+}
+
+TEST("require that compact discards drained data") {
+    RawBuf buf(10);
+    buf += "foobar";
+    buf.Drain(3);
+    buf.Compact();
+    buf.Fill(3);
+    EXPECT_EQUAL("barbar", getString(buf));
+}
+
+TEST("require that reusing a buffer that has grown 4x will alloc new buffer") {
+    RawBuf buf(10);
+    buf.preAlloc(100);
+    EXPECT_LESS_EQUAL(100u, buf.GetFreeLen());
+    buf.Reuse();
+    EXPECT_EQUAL(10u, buf.GetFreeLen());
+}
+
+TEST("require that various length and position information can be found.") {
+    RawBuf buf(30);
+    buf += "foo bar baz qux quux corge";
+    buf.Drain(7);
+    EXPECT_EQUAL(7u, buf.GetDrainLen());
+    EXPECT_EQUAL(19u, buf.GetUsedLen());
+    EXPECT_EQUAL(26u, buf.GetUsedAndDrainLen());
+    EXPECT_EQUAL(4u, buf.GetFreeLen());
+}
+
+TEST("require that rawbuf can 'putToInet' 16-bit numbers") {
+    RawBuf buf(1);
+    buf.Put16ToInet(0x1234);
+    EXPECT_EQUAL(2, buf.GetFillPos() - buf.GetDrainPos());
+    EXPECT_EQUAL(0x12, (int) buf.GetDrainPos()[0] & 0xff);
+    EXPECT_EQUAL(0x34, (int) buf.GetDrainPos()[1] & 0xff);
+}
+
+TEST("require that rawbuf can 'putToInet' 32-bit numbers") {
+    RawBuf buf(1);
+    buf.PutToInet(0x12345678);
+    EXPECT_EQUAL(4, buf.GetFillPos() - buf.GetDrainPos());
+    EXPECT_EQUAL(0x12, (int) buf.GetDrainPos()[0] & 0xff);
+    EXPECT_EQUAL(0x34, (int) buf.GetDrainPos()[1] & 0xff);
+    EXPECT_EQUAL(0x56, (int) buf.GetDrainPos()[2] & 0xff);
+    EXPECT_EQUAL(0x78, (int) buf.GetDrainPos()[3] & 0xff);
+}
+
+TEST("require that rawbuf can 'putToInet' 64-bit numbers") {
+    RawBuf buf(1);
+    buf.Put64ToInet(0x123456789abcdef0ULL);
+    EXPECT_EQUAL(8, buf.GetFillPos() - buf.GetDrainPos());
+    EXPECT_EQUAL(0x12, (int) buf.GetDrainPos()[0] & 0xff);
+    EXPECT_EQUAL(0x34, (int) buf.GetDrainPos()[1] & 0xff);
+    EXPECT_EQUAL(0x56, (int) buf.GetDrainPos()[2] & 0xff);
+    EXPECT_EQUAL(0x78, (int) buf.GetDrainPos()[3] & 0xff);
+    EXPECT_EQUAL(0x9a, (int) buf.GetDrainPos()[4] & 0xff);
+    EXPECT_EQUAL(0xbc, (int) buf.GetDrainPos()[5] & 0xff);
+    EXPECT_EQUAL(0xde, (int) buf.GetDrainPos()[6] & 0xff);
+    EXPECT_EQUAL(0xf0, (int) buf.GetDrainPos()[7] & 0xff);
+}
+
+
+}  // namespace
+
+TEST_MAIN() { TEST_RUN_ALL(); }