Move unit test code into a separate header

This commit is contained in:
Mark Haines 2015-02-24 11:28:07 +00:00
parent 7c1da4896f
commit 0e13cd3562
2 changed files with 61 additions and 55 deletions

55
tests/include/unittest.hh Normal file
View file

@ -0,0 +1,55 @@
#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(
const T & expected,
const T & 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 *expected,
std::uint8_t *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; }
};

View file

@ -1,52 +1,13 @@
#include "axolotl/crypto.hh"
#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;
void assert_equals(
std::uint8_t *expected,
std::uint8_t *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);
}
}
void pass() {
std::cout << "PASSED: " << TEST_CASE << std::endl;
}
#include "unittest.hh"
int main() {
{ /* Curve25529 Test Case 1 */
TEST_CASE = "Curve25529 Test Case 1";
TestCase test_case("Curve25529 Test Case 1");
std::uint8_t alice_private[32] = {
0x77, 0x07, 0x6D, 0x0A, 0x73, 0x18, 0xA5, 0x7D,
@ -103,14 +64,12 @@ axolotl::curve25519_shared_secret(bob_pair, alice_pair, actual_agreement);
assert_equals(expected_agreement, actual_agreement, 32);
pass();
} /* Curve25529 Test Case 1 */
{ /* AES Test Case 1 */
TEST_CASE = "AES Test Case 1";
TestCase test_case("AES Test Case 1");
axolotl::Aes256Key key = {};
axolotl::Aes256Iv iv = {};
@ -133,14 +92,12 @@ axolotl::aes_decrypt_cbc(key, iv, expected, sizeof(expected), actual);
assert_equals(input, actual, 32);
pass();
} /* AES Test Case 1 */
{ /* SHA 256 Test Case 1 */
TEST_CASE = "SHA 256 Test Case 1";
TestCase test_case("SHA 256 Test Case 1");
std::uint8_t input[0] = {};
@ -157,13 +114,11 @@ axolotl::sha256(input, sizeof(input), actual);
assert_equals(expected, actual, 32);
pass();
} /* SHA 256 Test Case 1 */
{ /* HMAC Test Case 1 */
TEST_CASE = "HMAC Test Case 1";
TestCase test_case("HMAC Test Case 1");
std::uint8_t input[0] = {};
@ -180,13 +135,11 @@ axolotl::hmac_sha256(input, sizeof(input), input, sizeof(input), actual);
assert_equals(expected, actual, 32);
pass();
} /* HMAC Test Case 1 */
{ /* HDKF Test Case 1 */
TEST_CASE = "HDKF Test Case 1";
TestCase test_case("HDKF Test Case 1");
std::uint8_t input[22] = {
0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b,
@ -241,8 +194,6 @@ axolotl::hkdf_sha256(
assert_equals(hkdf_expected_output, hkdf_actual_output, 42);
pass();
} /* HDKF Test Case 1 */
}