transmission/tests/libtransmission/buffer-test.cc

// This file Copyright (C) 2013-2022 Mnemosyne LLC.
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
// or any future license endorsed by Mnemosyne LLC.
// License text can be found in the licenses/ folder.

#include <libtransmission/transmission.h>

#include <libtransmission/tr-buffer.h>

#include "test-fixtures.h"

using BufferTest = ::testing::Test;
using namespace std::literals;
using Buffer = libtransmission::Buffer;

TEST_F(BufferTest, startsWithInSingleSegment)
{
    auto constexpr Hello = "Hello, "sv;
    auto constexpr World = "World"sv;
    auto constexpr Bang = "!"sv;

    auto buf = Buffer{};
    buf.add(Hello);
    EXPECT_TRUE(buf.starts_with(Hello));

    buf.add(World);
    EXPECT_TRUE(buf.starts_with(Hello));
    EXPECT_TRUE(buf.starts_with("Hello, Worl"sv));
    EXPECT_TRUE(buf.starts_with("Hello, World"sv));
    EXPECT_FALSE(buf.starts_with("Hello, World!"sv));
    EXPECT_FALSE(buf.starts_with("Hello!"sv));

    buf.add(Bang);
    EXPECT_FALSE(buf.starts_with("Hello!"));
    EXPECT_TRUE(buf.starts_with(Hello));
    EXPECT_TRUE(buf.starts_with("Hello, Worl"sv));
    EXPECT_TRUE(buf.starts_with("Hello, World"sv));
    EXPECT_TRUE(buf.starts_with("Hello, World!"sv));
}
TEST_F(BufferTest, startsWithInMultiSegment)
{
    auto constexpr Hello = "Hello, "sv;
    auto constexpr World = "World"sv;
    auto constexpr Bang = "!"sv;

    auto buf = std::make_unique<Buffer>();
    buf->add(Buffer{ Hello });
    EXPECT_TRUE(buf->starts_with(Hello));

    buf->add(Buffer{ World });
    EXPECT_TRUE(buf->starts_with(Hello));
    EXPECT_TRUE(buf->starts_with("Hello, Worl"sv));
    EXPECT_TRUE(buf->starts_with("Hello, World"sv));
    EXPECT_FALSE(buf->starts_with("Hello, World!"sv));
    EXPECT_FALSE(buf->starts_with("Hello!"sv));

    buf->add(Buffer{ Bang });
    EXPECT_FALSE(buf->starts_with("Hello!"));
    EXPECT_TRUE(buf->starts_with(Hello));
    EXPECT_TRUE(buf->starts_with("Hello, Worl"sv));
    EXPECT_TRUE(buf->starts_with("Hello, World"sv));
    EXPECT_TRUE(buf->starts_with("Hello, World!"sv));
}

TEST_F(BufferTest, Move)
{
    auto constexpr TwoChars = "12"sv;
    auto constexpr SixChars = "123456"sv;
    auto constexpr TenChars = "1234567890"sv;

    auto a = Buffer{ TwoChars };
    auto b = Buffer{ SixChars };
    auto c = Buffer{ TenChars };

    auto lens = std::array<size_t, 3>{ std::size(TwoChars), std::size(SixChars), std::size(TenChars) };

    EXPECT_EQ(lens[0], std::size(a));
    EXPECT_EQ(lens[1], std::size(b));
    EXPECT_EQ(lens[2], std::size(c));

    std::swap(a, b);
    EXPECT_EQ(lens[0], std::size(b));
    EXPECT_EQ(lens[1], std::size(a));
    EXPECT_EQ(lens[2], std::size(c));

    std::swap(a, c);
    EXPECT_EQ(lens[0], std::size(b));
    EXPECT_EQ(lens[1], std::size(c));
    EXPECT_EQ(lens[2], std::size(a));

    std::swap(b, c);
    EXPECT_EQ(lens[0], std::size(c));
    EXPECT_EQ(lens[1], std::size(b));
    EXPECT_EQ(lens[2], std::size(a));

    a.add(std::data(TwoChars), std::size(TwoChars));

    {
        auto constexpr OneChar = "1"sv;
        auto d = Buffer{ OneChar };

        std::swap(a, d);
        EXPECT_EQ(1U, std::size(a));
    }

    EXPECT_EQ(1U, std::size(a));
    a.add(std::data(TwoChars), std::size(TwoChars));
    EXPECT_EQ(3U, std::size(a));
}

TEST_F(BufferTest, NonBufferWriter)
{
    auto constexpr Hello = "Hello, "sv;
    auto constexpr World = "World"sv;
    auto constexpr Bang = "!"sv;

    auto out1 = Buffer{};

    auto out2_vec = std::vector<std::byte>{};
    auto out2 = libtransmission::BufferWriter<std::vector<std::byte>, std::byte>{ &out2_vec };

    out1.add_uint8(1);
    out2.add_uint8(1);

    out1.add_uint16(1);
    out2.add_uint16(1);

    out1.add_uint32(1);
    out2.add_uint32(1);

    out1.add(Hello);
    out2.add(Hello);

    out1.add(World);
    out2.add(World);

    out1.add(Bang);
    out2.add(Bang);

    auto const result1 = out1.pullup_sv();
    auto const result2 = std::string_view{ reinterpret_cast<char const*>(std::data(out2_vec)), std::size(out2_vec) };
    EXPECT_EQ(result1, result2);
}