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