140 lines
4.0 KiB
C++
140 lines
4.0 KiB
C++
// 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 <cstddef> // std::byte
|
|
#include <cstdint> // uint16_t, uint32_t, uint64_t
|
|
#include <memory>
|
|
#include <string_view>
|
|
|
|
#include <libtransmission/transmission.h>
|
|
|
|
#include <libtransmission/crypto-utils.h>
|
|
#include <libtransmission/tr-buffer.h>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "test-fixtures.h"
|
|
|
|
using BufferTest = ::testing::Test;
|
|
using namespace std::literals;
|
|
using Buffer = libtransmission::StackBuffer<1024, std::byte>;
|
|
|
|
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, Numbers)
|
|
{
|
|
for (auto i = 0; i < 100; ++i)
|
|
{
|
|
auto const expected_u8 = tr_rand_obj<uint8_t>();
|
|
auto const expected_u16 = tr_rand_obj<uint16_t>();
|
|
auto const expected_u32 = tr_rand_obj<uint32_t>();
|
|
auto const expected_u64 = tr_rand_obj<uint64_t>();
|
|
|
|
auto buf = Buffer{};
|
|
|
|
buf.add_uint8(expected_u8);
|
|
buf.add_uint16(expected_u16);
|
|
buf.add_uint32(expected_u32);
|
|
buf.add_uint64(expected_u64);
|
|
|
|
EXPECT_EQ(expected_u8, buf.to_uint8());
|
|
EXPECT_EQ(expected_u16, buf.to_uint16());
|
|
EXPECT_EQ(expected_u32, buf.to_uint32());
|
|
EXPECT_EQ(expected_u64, buf.to_uint64());
|
|
|
|
buf.add_uint64(expected_u64);
|
|
buf.add_uint32(expected_u32);
|
|
buf.add_uint16(expected_u16);
|
|
buf.add_uint8(expected_u8);
|
|
|
|
EXPECT_EQ(expected_u64, buf.to_uint64());
|
|
EXPECT_EQ(expected_u32, buf.to_uint32());
|
|
EXPECT_EQ(expected_u16, buf.to_uint16());
|
|
EXPECT_EQ(expected_u8, buf.to_uint8());
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
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.to_string_view();
|
|
auto const result2 = std::string_view{ reinterpret_cast<char const*>(std::data(out2_vec)), std::size(out2_vec) };
|
|
EXPECT_EQ(result1, result2);
|
|
}
|
|
#endif
|