olm/tests/test_base64.cpp

89 lines
2.7 KiB
C++
Raw Normal View History

#include "olm/base64.hh"
2016-05-18 18:13:39 +02:00
#include "olm/base64.h"
2015-06-15 18:48:09 +02:00
#include "unittest.hh"
#include <cstring>
#include <vector>
2015-06-15 18:48:09 +02:00
int main() {
{ /* Base64 encode test */
2016-05-18 18:13:39 +02:00
TestCase test_case("Base64 C++ binding encode test");
2015-06-15 18:48:09 +02:00
std::uint8_t input[] = "Hello World";
std::uint8_t expected_output[] = "SGVsbG8gV29ybGQ";
std::size_t input_length = sizeof(input) - 1;
std::size_t output_length = olm::encode_base64_length(input_length);
2015-06-15 18:48:09 +02:00
assert_equals(std::size_t(15), output_length);
std::uint8_t output[15];
olm::encode_base64(input, input_length, output);
2015-06-15 18:48:09 +02:00
assert_equals(expected_output, output, output_length);
}
2016-05-18 18:13:39 +02:00
{
TestCase test_case("Base64 C binding encode test");
std::uint8_t input[] = "Hello World";
std::uint8_t expected_output[] = "SGVsbG8gV29ybGQ";
std::size_t input_length = sizeof(input) - 1;
std::size_t output_length = ::_olm_encode_base64_length(input_length);
2016-05-18 18:13:39 +02:00
assert_equals(std::size_t(15), output_length);
std::uint8_t output[15];
output_length = ::_olm_encode_base64(input, input_length, output);
2016-05-18 18:13:39 +02:00
assert_equals(std::size_t(15), output_length);
assert_equals(expected_output, output, output_length);
}
2015-06-15 18:48:09 +02:00
{ /* Base64 decode test */
2016-05-18 18:13:39 +02:00
TestCase test_case("Base64 C++ binding decode test");
2015-06-15 18:48:09 +02:00
std::uint8_t input[] = "SGVsbG8gV29ybGQ";
std::uint8_t expected_output[] = "Hello World";
std::size_t input_length = sizeof(input) - 1;
std::size_t output_length = olm::decode_base64_length(input_length);
2015-06-15 18:48:09 +02:00
assert_equals(std::size_t(11), output_length);
std::uint8_t output[11];
olm::decode_base64(input, input_length, output);
2015-06-15 18:48:09 +02:00
assert_equals(expected_output, output, output_length);
}
2016-05-18 18:13:39 +02:00
{
TestCase test_case("Base64 C binding decode test");
std::uint8_t input[] = "SGVsbG8gV29ybGQ";
std::uint8_t expected_output[] = "Hello World";
std::size_t input_length = sizeof(input) - 1;
std::size_t output_length = ::_olm_decode_base64_length(input_length);
2016-05-18 18:13:39 +02:00
assert_equals(std::size_t(11), output_length);
std::uint8_t output[11];
output_length = ::_olm_decode_base64(input, input_length, output);
2016-05-18 18:13:39 +02:00
assert_equals(std::size_t(11), output_length);
assert_equals(expected_output, output, output_length);
}
{
TestCase test_case("Decoding base64 of invalid length fails with -1");
std::uint8_t input[] = "SGVsbG8gV29ybGQab";
std::size_t input_length = sizeof(input) - 1;
/* We use a longer but valid input length here so that we don't get back -1.
* Nothing will be written to the output buffer anyway because the input is
* invalid. */
std::size_t buf_length = olm::decode_base64_length(input_length + 1);
std::vector<std::uint8_t> output(buf_length, 0);
std::vector<std::uint8_t> expected_output(buf_length, 0);
std::size_t output_length = ::_olm_decode_base64(input, input_length, output.data());
assert_equals(std::size_t(-1), output_length);
assert_equals(0, memcmp(output.data(), expected_output.data(), buf_length));
}
2016-05-18 18:13:39 +02:00
2015-06-15 18:48:09 +02:00
}