^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * KUnit test for struct string_stream.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) * Copyright (C) 2019, Google LLC.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) * Author: Brendan Higgins <brendanhiggins@google.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include <kunit/test.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <linux/slab.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include "string-stream.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) static void string_stream_test_empty_on_creation(struct kunit *test)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) struct string_stream *stream = alloc_string_stream(test, GFP_KERNEL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) KUNIT_EXPECT_TRUE(test, string_stream_is_empty(stream));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) static void string_stream_test_not_empty_after_add(struct kunit *test)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) struct string_stream *stream = alloc_string_stream(test, GFP_KERNEL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) string_stream_add(stream, "Foo");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) KUNIT_EXPECT_FALSE(test, string_stream_is_empty(stream));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) static void string_stream_test_get_string(struct kunit *test)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) struct string_stream *stream = alloc_string_stream(test, GFP_KERNEL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) char *output;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) string_stream_add(stream, "Foo");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) string_stream_add(stream, " %s", "bar");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) output = string_stream_get_string(stream);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) KUNIT_ASSERT_STREQ(test, output, "Foo bar");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) static struct kunit_case string_stream_test_cases[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) KUNIT_CASE(string_stream_test_empty_on_creation),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) KUNIT_CASE(string_stream_test_not_empty_after_add),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) KUNIT_CASE(string_stream_test_get_string),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) {}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) static struct kunit_suite string_stream_test_suite = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) .name = "string-stream-test",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) .test_cases = string_stream_test_cases
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) kunit_test_suites(&string_stream_test_suite);