tests/unit/test_sprintf.c: Test SNPRINTF()

Signed-off-by: Alejandro Colomar <alx@kernel.org>
This commit is contained in:
Alejandro Colomar 2023-11-22 19:39:02 +01:00 committed by Iker Pedrosa
parent cf9cc6963c
commit ddbd3a36c1
2 changed files with 80 additions and 0 deletions

View File

@ -5,6 +5,7 @@ TESTS = $(check_PROGRAMS)
check_PROGRAMS = \
test_chkname \
test_sprintf \
test_strncpy \
test_strtcpy \
test_xasprintf
@ -47,6 +48,19 @@ test_logind_LDADD = \
$(LIBSYSTEMD) \
$(NULL)
test_sprintf_SOURCES = \
../../lib/string/sprintf.c \
test_sprintf.c \
$(NULL)
test_sprintf_CFLAGS = \
$(AM_FLAGS) \
$(NULL)
test_sprintf_LDFLAGS = \
$(NULL)
test_sprintf_LDADD = \
$(CMOCKA_LIBS) \
$(NULL)
test_strncpy_SOURCES = \
test_strncpy.c \
$(NULL)

66
tests/unit/test_sprintf.c Normal file
View File

@ -0,0 +1,66 @@
// SPDX-FileCopyrightText: 2023, Alejandro Colomar <alx@kernel.org>
// SPDX-License-Identifier: BSD-3-Clause
#include <config.h>
#include <stdio.h>
#include <string.h>
#include <stdarg.h> // Required by <cmocka.h>
#include <stddef.h> // Required by <cmocka.h>
#include <setjmp.h> // Required by <cmocka.h>
#include <stdint.h> // Required by <cmocka.h>
#include <cmocka.h>
#include "sizeof.h"
#include "string/sprintf.h"
static void test_SNPRINTF_trunc(void **state);
static void test_SNPRINTF_ok(void **state);
int
main(void)
{
const struct CMUnitTest tests[] = {
cmocka_unit_test(test_SNPRINTF_trunc),
cmocka_unit_test(test_SNPRINTF_ok),
};
return cmocka_run_group_tests(tests, NULL, NULL);
}
static void
test_SNPRINTF_trunc(void **state)
{
char buf[NITEMS("foo")];
// Test that we're not returning SIZE_MAX
assert_true(SNPRINTF(buf, "f%su", "oo") < 0);
assert_true(strcmp(buf, "foo") == 0);
assert_true(SNPRINTF(buf, "barbaz") == -1);
assert_true(strcmp(buf, "bar") == 0);
}
static void
test_SNPRINTF_ok(void **state)
{
char buf[NITEMS("foo")];
assert_true(SNPRINTF(buf, "%s", "foo") == strlen("foo"));
assert_true(strcmp(buf, "foo") == 0);
assert_true(SNPRINTF(buf, "%do", 1) == strlen("1o"));
assert_true(strcmp(buf, "1o") == 0);
assert_true(SNPRINTF(buf, "f") == strlen("f"));
assert_true(strcmp(buf, "f") == 0);
assert_true(SNPRINTF(buf, "") == strlen(""));
assert_true(strcmp(buf, "") == 0);
}