16 #include "string-stream.h"
31 static char *get_concatenated_string(struct kunit *test, struct string_stream *stream)  in get_concatenated_string()  argument
33 	char *str = string_stream_get_string(stream);  in get_concatenated_string()
44 	struct string_stream *stream;  in string_stream_managed_init_test()  local
47 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_managed_init_test()
48 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_managed_init_test()
50 	KUNIT_EXPECT_EQ(test, stream->length, 0);  in string_stream_managed_init_test()
51 	KUNIT_EXPECT_TRUE(test, list_empty(&stream->fragments));  in string_stream_managed_init_test()
52 	KUNIT_EXPECT_TRUE(test, (stream->gfp == GFP_KERNEL));  in string_stream_managed_init_test()
53 	KUNIT_EXPECT_FALSE(test, stream->append_newlines);  in string_stream_managed_init_test()
54 	KUNIT_EXPECT_TRUE(test, string_stream_is_empty(stream));  in string_stream_managed_init_test()
60 	struct string_stream *stream;  in string_stream_unmanaged_init_test()  local
62 	stream = alloc_string_stream(GFP_KERNEL);  in string_stream_unmanaged_init_test()
63 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_unmanaged_init_test()
64 	kunit_add_action(test, cleanup_raw_stream, stream);  in string_stream_unmanaged_init_test()
66 	KUNIT_EXPECT_EQ(test, stream->length, 0);  in string_stream_unmanaged_init_test()
67 	KUNIT_EXPECT_TRUE(test, list_empty(&stream->fragments));  in string_stream_unmanaged_init_test()
68 	KUNIT_EXPECT_TRUE(test, (stream->gfp == GFP_KERNEL));  in string_stream_unmanaged_init_test()
69 	KUNIT_EXPECT_FALSE(test, stream->append_newlines);  in string_stream_unmanaged_init_test()
71 	KUNIT_EXPECT_TRUE(test, string_stream_is_empty(stream));  in string_stream_unmanaged_init_test()
74 static void string_stream_destroy_stub(struct string_stream *stream)  in string_stream_destroy_stub()  argument
80 	if (stream == priv->expected_free_stream) {  in string_stream_destroy_stub()
93 	string_stream_clear(stream);  in string_stream_destroy_stub()
94 	kfree(stream);  in string_stream_destroy_stub()
166 	struct string_stream *stream;  in string_stream_line_add_test()  local
172 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_line_add_test()
173 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_line_add_test()
184 		string_stream_add(stream, line);  in string_stream_line_add_test()
189 	concat_string = get_concatenated_string(test, stream);  in string_stream_line_add_test()
222 	struct string_stream *stream;  in string_stream_variable_length_line_test()  local
228 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_variable_length_line_test()
229 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_variable_length_line_test()
240 		string_stream_add(stream, "%s\n", &line[offset]);  in string_stream_variable_length_line_test()
245 	concat_string = get_concatenated_string(test, stream);  in string_stream_variable_length_line_test()
272 /* Appending the content of one string stream to another. */
294 	/* Append content of empty stream to empty stream */  in string_stream_append_test()
304 	/* Append content of empty stream to non-empty stream */  in string_stream_append_test()
313 	/* Append content of non-empty stream to non-empty stream */  in string_stream_append_test()
330 	/* Append content of non-empty stream to empty stream */  in string_stream_append_test()
340 /* Appending the content of one string stream to one with auto-newlining. */
345 	/* Stream 1 has newline appending enabled */  in string_stream_append_auto_newline_test()
351 	/* Stream 2 does not append newlines */  in string_stream_append_auto_newline_test()
355 	/* Appending a stream with a newline should not add another newline */  in string_stream_append_auto_newline_test()
368 	 * Appending a stream without newline should add a final newline.  in string_stream_append_auto_newline_test()
382 	struct string_stream *stream;  in string_stream_append_empty_string_test()  local
385 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_append_empty_string_test()
386 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_append_empty_string_test()
389 	string_stream_add(stream, "%s", "");  in string_stream_append_empty_string_test()
390 	KUNIT_EXPECT_TRUE(test, string_stream_is_empty(stream));  in string_stream_append_empty_string_test()
391 	KUNIT_EXPECT_TRUE(test, list_empty(&stream->fragments));  in string_stream_append_empty_string_test()
393 	/* Adding an empty string to a non-empty stream */  in string_stream_append_empty_string_test()
394 	string_stream_add(stream, "Add this line");  in string_stream_append_empty_string_test()
395 	original_frag_count = list_count_nodes(&stream->fragments);  in string_stream_append_empty_string_test()
397 	string_stream_add(stream, "%s", "");  in string_stream_append_empty_string_test()
398 	KUNIT_EXPECT_EQ(test, list_count_nodes(&stream->fragments), original_frag_count);  in string_stream_append_empty_string_test()
399 	KUNIT_EXPECT_STREQ(test, get_concatenated_string(test, stream), "Add this line");  in string_stream_append_empty_string_test()
405 	struct string_stream *stream;  in string_stream_no_auto_newline_test()  local
407 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_no_auto_newline_test()
408 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_no_auto_newline_test()
414 	string_stream_add(stream, "One");  in string_stream_no_auto_newline_test()
415 	string_stream_add(stream, "Two\n");  in string_stream_no_auto_newline_test()
416 	string_stream_add(stream, "%s\n", "Three");  in string_stream_no_auto_newline_test()
417 	string_stream_add(stream, "%s", "Four\n");  in string_stream_no_auto_newline_test()
418 	string_stream_add(stream, "Five\n%s", "Six");  in string_stream_no_auto_newline_test()
419 	string_stream_add(stream, "Seven\n\n");  in string_stream_no_auto_newline_test()
420 	string_stream_add(stream, "Eight");  in string_stream_no_auto_newline_test()
421 	KUNIT_EXPECT_STREQ(test, get_concatenated_string(test, stream),  in string_stream_no_auto_newline_test()
428 	struct string_stream *stream;  in string_stream_auto_newline_test()  local
430 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_auto_newline_test()
431 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_auto_newline_test()
433 	string_stream_set_append_newlines(stream, true);  in string_stream_auto_newline_test()
434 	KUNIT_EXPECT_TRUE(test, stream->append_newlines);  in string_stream_auto_newline_test()
441 	string_stream_add(stream, "One");  in string_stream_auto_newline_test()
442 	string_stream_add(stream, "Two\n");  in string_stream_auto_newline_test()
443 	string_stream_add(stream, "%s\n", "Three");  in string_stream_auto_newline_test()
444 	string_stream_add(stream, "%s", "Four\n");  in string_stream_auto_newline_test()
445 	string_stream_add(stream, "Five\n%s", "Six");  in string_stream_auto_newline_test()
446 	string_stream_add(stream, "Seven\n\n");  in string_stream_auto_newline_test()
447 	string_stream_add(stream, "Eight");  in string_stream_auto_newline_test()
448 	KUNIT_EXPECT_STREQ(test, get_concatenated_string(test, stream),  in string_stream_auto_newline_test()
459 	struct string_stream *stream;  in string_stream_performance_test()  local
465 	stream = kunit_alloc_string_stream(test, GFP_KERNEL);  in string_stream_performance_test()
466 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, stream);  in string_stream_performance_test()
474 		string_stream_add(stream, "%s: %d\n", &test_line[offset], i);  in string_stream_performance_test()
482 	bytes_requested = sizeof(*stream);  in string_stream_performance_test()
483 	actual_bytes_used = ksize(stream);  in string_stream_performance_test()
486 	list_for_each_entry(frag_container, &stream->fragments, node) {  in string_stream_performance_test()
533 	.name = "string-stream-test",