Fix byte_stream::put_n

This commit is contained in:
Lee Clagett 2020-12-14 14:45:24 -05:00
parent 7a9ba3f154
commit cf4461f40b
2 changed files with 18 additions and 1 deletions

View file

@ -188,7 +188,7 @@ namespace epee
void put_n(const std::uint8_t ch, const std::size_t count) void put_n(const std::uint8_t ch, const std::size_t count)
{ {
check(count); check(count);
std::memset(tellp(), count, ch); std::memset(tellp(), ch, count);
next_write_ += count; next_write_ += count;
} }

View file

@ -972,6 +972,23 @@ TEST(ByteStream, Put)
EXPECT_TRUE(equal(bytes, byte_span{stream.data(), stream.size()})); EXPECT_TRUE(equal(bytes, byte_span{stream.data(), stream.size()}));
} }
TEST(ByteStream, PutN)
{
using boost::range::equal;
using byte_span = epee::span<const std::uint8_t>;
std::vector<std::uint8_t> bytes;
bytes.resize(1000, 'f');
epee::byte_stream stream;
stream.put_n('f', 1000);
EXPECT_EQ(1000u, stream.size());
EXPECT_LE(1000u, stream.capacity());
EXPECT_EQ(stream.available(), stream.capacity() - stream.size());
EXPECT_TRUE(equal(bytes, byte_span{stream.data(), stream.size()}));
}
TEST(ByteStream, Reserve) TEST(ByteStream, Reserve)
{ {
using boost::range::equal; using boost::range::equal;