olm/tests/include/unittest.hh
2015-02-25 11:20:35 +00:00

55 lines
1.2 KiB
C++

#include <cstring>
#include <iostream>
#include <iomanip>
#include <cstdlib>
std::ostream & print_hex(
std::ostream & os,
std::uint8_t const * data,
std::size_t length
) {
for (std::size_t i = 0; i < length; i++) {
os << std::setw(2) << std::setfill('0') << std::right
<< std::hex << (int) data[i];
}
return os;
}
char const * TEST_CASE;
template<typename T>
void assert_equals(
T const & expected,
T const & actual
) {
if (expected != actual) {
std::cout << "FAILED: " << TEST_CASE << std::endl;
std::cout << "Expected: " << expected << std::endl;
std::cout << "Actual: " << actual << std::endl;
std::exit(1);
}
}
void assert_equals(
std::uint8_t const * expected,
std::uint8_t const * actual,
std::size_t length
) {
if (std::memcmp(expected, actual, length)) {
std::cout << "FAILED: " << TEST_CASE << std::endl;
print_hex(std::cout << "Expected: ", expected, length) << std::endl;
print_hex(std::cout << "Actual: ", actual, length) << std::endl;
std::exit(1);
}
}
class TestCase {
public:
TestCase(const char *name) { TEST_CASE = name; }
~TestCase() { std::cout << "PASSED: " << TEST_CASE << std::endl; }
};