tomato-testing/toxcore/test_util_test.cc

65 lines
1.7 KiB
C++
Raw Normal View History

#include "test_util.hh"
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <array>
#include "crypto_core_test_util.hh"
namespace {
using ::testing::Each;
using ::testing::Eq;
TEST(CryptoCoreTestUtil, RandomBytesDoesNotTouchZeroSizeArray)
{
const Test_Random rng;
std::array<uint8_t, 32> bytes{};
for (uint32_t i = 0; i < 100; ++i) {
random_bytes(rng, bytes.data(), 0);
ASSERT_THAT(bytes, Each(Eq(0x00)));
}
}
TEST(CryptoCoreTestUtil, RandomBytesFillsEntireArray)
{
const Test_Random rng;
std::array<uint8_t, 32> bytes{};
for (uint32_t size = 1; size < bytes.size(); ++size) {
bool const success = [&]() {
// Try a few times. There ought to be a non-zero byte in our randomness at
// some point.
for (uint32_t i = 0; i < 100; ++i) {
random_bytes(rng, bytes.data(), bytes.size());
if (bytes[size - 1] != 0x00) {
return true;
}
}
return false;
}();
ASSERT_TRUE(success);
}
}
TEST(CryptoCoreTestUtil, RandomBytesDoesNotBufferOverrun)
{
const Test_Random rng;
std::array<uint8_t, 32> bytes{};
// Try a few times. It should never overrun.
for (uint32_t i = 0; i < 100; ++i) {
for (uint32_t diff = 1; diff < sizeof(uint64_t); ++diff) {
bytes = {};
random_bytes(rng, bytes.data(), bytes.size() - diff);
// All bytes not in the range we want to write should be 0.
ASSERT_THAT(std::vector<uint8_t>(bytes.begin() + (bytes.size() - diff), bytes.end()),
Each(Eq(0x00)));
}
}
}
} // namespace