From 072a9b96eb2036718db1212c9f35b9cfe72e3f78 Mon Sep 17 00:00:00 2001 From: Anna Karas Date: Mon, 8 Jun 2020 15:16:19 +0200 Subject: tests: Add lib/b64_decode-test test case Implement unit tests for lib/b64_decode module. Signed-off-by: Anna Karas Change-Id: Id5fe9272e30eaff3d086a95241b3819101089c2b Reviewed-on: https://review.coreboot.org/c/coreboot/+/42313 Reviewed-by: Paul Fagerburg Tested-by: build bot (Jenkins) --- tests/lib/Makefile.inc | 4 +++ tests/lib/b64_decode-test.c | 67 +++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 71 insertions(+) create mode 100644 tests/lib/b64_decode-test.c (limited to 'tests') diff --git a/tests/lib/Makefile.inc b/tests/lib/Makefile.inc index c490317644..5d47fea887 100644 --- a/tests/lib/Makefile.inc +++ b/tests/lib/Makefile.inc @@ -1,6 +1,10 @@ # SPDX-License-Identifier: GPL-2.0-only tests-y += string-test +tests-y += b64_decode-test string-test-srcs += tests/lib/string-test.c string-test-srcs += src/lib/string.c + +b64_decode-test-srcs += tests/lib/b64_decode-test.c +b64_decode-test-srcs += src/lib/b64_decode.c diff --git a/tests/lib/b64_decode-test.c b/tests/lib/b64_decode-test.c new file mode 100644 index 0000000000..452c67f3ae --- /dev/null +++ b/tests/lib/b64_decode-test.c @@ -0,0 +1,67 @@ +/* SPDX-License-Identifier: GPL-2.0-only */ + +#include +#include +#include +#include +#include +#include + +struct messages_t { + const char *enc; + const char *dec; +} messages[] = { + {"QQ==", "A"}, + {"Q\r\nUI=", "AB"}, + {"QUJD", "ABC"}, + {"\nQUJDRA==", "ABCD"}, + {"SGVsbG8\r=", "Hello"}, + {"SGVsbG8h", "Hello!"} +}; + +const char *invalid[] = { + "QQ=-=", + "SGVsbG-8=" +}; + +/* Provide necessary definition in order to satisfy dependencies. */ +int do_printk(int msg_level, const char *fmt, ...) +{ + return 0; +} + +static void test_b64_decode(void **state) +{ + uint8_t *decoded; + size_t res; + + for (int i = 0; i < ARRAY_SIZE(messages); i++) { + decoded = malloc(strlen(messages[i].enc) * sizeof(char)); + + res = b64_decode((uint8_t *)messages[i].enc, strlen(messages[i].enc), decoded); + + assert_int_equal(res, (strlen(messages[i].dec))); + assert_string_equal(decoded, messages[i].dec); + + free(decoded); + } + + for (int i = 0; i < ARRAY_SIZE(invalid); i++) { + decoded = malloc(strlen(invalid[i]) * sizeof(char)); + + res = b64_decode((uint8_t *)invalid[i], strlen(invalid[i]), decoded); + + assert_int_equal(res, 0); + + free(decoded); + } +} + +int main(void) +{ + const struct CMUnitTest tests[] = { + cmocka_unit_test(test_b64_decode), + }; + + return cmocka_run_group_tests(tests, NULL, NULL); +} -- cgit v1.2.3