1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * KUnit test for core test infrastructure. 4 * 5 * Copyright (C) 2019, Google LLC. 6 * Author: Brendan Higgins <brendanhiggins@google.com> 7 */ 8 #include <kunit/test.h> 9 #include <kunit/test-bug.h> 10 11 #include "string-stream.h" 12 #include "try-catch-impl.h" 13 14 struct kunit_try_catch_test_context { 15 struct kunit_try_catch *try_catch; 16 bool function_called; 17 }; 18 19 static void kunit_test_successful_try(void *data) 20 { 21 struct kunit *test = data; 22 struct kunit_try_catch_test_context *ctx = test->priv; 23 24 ctx->function_called = true; 25 } 26 27 static void kunit_test_no_catch(void *data) 28 { 29 struct kunit *test = data; 30 31 KUNIT_FAIL(test, "Catch should not be called\n"); 32 } 33 34 static void kunit_test_try_catch_successful_try_no_catch(struct kunit *test) 35 { 36 struct kunit_try_catch_test_context *ctx = test->priv; 37 struct kunit_try_catch *try_catch = ctx->try_catch; 38 39 kunit_try_catch_init(try_catch, 40 test, 41 kunit_test_successful_try, 42 kunit_test_no_catch); 43 kunit_try_catch_run(try_catch, test); 44 45 KUNIT_EXPECT_TRUE(test, ctx->function_called); 46 } 47 48 static void kunit_test_unsuccessful_try(void *data) 49 { 50 struct kunit *test = data; 51 struct kunit_try_catch_test_context *ctx = test->priv; 52 struct kunit_try_catch *try_catch = ctx->try_catch; 53 54 kunit_try_catch_throw(try_catch); 55 KUNIT_FAIL(test, "This line should never be reached\n"); 56 } 57 58 static void kunit_test_catch(void *data) 59 { 60 struct kunit *test = data; 61 struct kunit_try_catch_test_context *ctx = test->priv; 62 63 ctx->function_called = true; 64 } 65 66 static void kunit_test_try_catch_unsuccessful_try_does_catch(struct kunit *test) 67 { 68 struct kunit_try_catch_test_context *ctx = test->priv; 69 struct kunit_try_catch *try_catch = ctx->try_catch; 70 71 kunit_try_catch_init(try_catch, 72 test, 73 kunit_test_unsuccessful_try, 74 kunit_test_catch); 75 kunit_try_catch_run(try_catch, test); 76 77 KUNIT_EXPECT_TRUE(test, ctx->function_called); 78 } 79 80 static int kunit_try_catch_test_init(struct kunit *test) 81 { 82 struct kunit_try_catch_test_context *ctx; 83 84 ctx = kunit_kzalloc(test, sizeof(*ctx), GFP_KERNEL); 85 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx); 86 test->priv = ctx; 87 88 ctx->try_catch = kunit_kmalloc(test, 89 sizeof(*ctx->try_catch), 90 GFP_KERNEL); 91 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx->try_catch); 92 93 return 0; 94 } 95 96 static struct kunit_case kunit_try_catch_test_cases[] = { 97 KUNIT_CASE(kunit_test_try_catch_successful_try_no_catch), 98 KUNIT_CASE(kunit_test_try_catch_unsuccessful_try_does_catch), 99 {} 100 }; 101 102 static struct kunit_suite kunit_try_catch_test_suite = { 103 .name = "kunit-try-catch-test", 104 .init = kunit_try_catch_test_init, 105 .test_cases = kunit_try_catch_test_cases, 106 }; 107 108 /* 109 * Context for testing test managed resources 110 * is_resource_initialized is used to test arbitrary resources 111 */ 112 struct kunit_test_resource_context { 113 struct kunit test; 114 bool is_resource_initialized; 115 int allocate_order[2]; 116 int free_order[4]; 117 }; 118 119 static int fake_resource_init(struct kunit_resource *res, void *context) 120 { 121 struct kunit_test_resource_context *ctx = context; 122 123 res->data = &ctx->is_resource_initialized; 124 ctx->is_resource_initialized = true; 125 return 0; 126 } 127 128 static void fake_resource_free(struct kunit_resource *res) 129 { 130 bool *is_resource_initialized = res->data; 131 132 *is_resource_initialized = false; 133 } 134 135 static void kunit_resource_test_init_resources(struct kunit *test) 136 { 137 struct kunit_test_resource_context *ctx = test->priv; 138 139 kunit_init_test(&ctx->test, "testing_test_init_test", NULL); 140 141 KUNIT_EXPECT_TRUE(test, list_empty(&ctx->test.resources)); 142 } 143 144 static void kunit_resource_test_alloc_resource(struct kunit *test) 145 { 146 struct kunit_test_resource_context *ctx = test->priv; 147 struct kunit_resource *res; 148 kunit_resource_free_t free = fake_resource_free; 149 150 res = kunit_alloc_and_get_resource(&ctx->test, 151 fake_resource_init, 152 fake_resource_free, 153 GFP_KERNEL, 154 ctx); 155 156 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, res); 157 KUNIT_EXPECT_PTR_EQ(test, 158 &ctx->is_resource_initialized, 159 (bool *)res->data); 160 KUNIT_EXPECT_TRUE(test, list_is_last(&res->node, &ctx->test.resources)); 161 KUNIT_EXPECT_PTR_EQ(test, free, res->free); 162 163 kunit_put_resource(res); 164 } 165 166 static inline bool kunit_resource_instance_match(struct kunit *test, 167 struct kunit_resource *res, 168 void *match_data) 169 { 170 return res->data == match_data; 171 } 172 173 /* 174 * Note: tests below use kunit_alloc_and_get_resource(), so as a consequence 175 * they have a reference to the associated resource that they must release 176 * via kunit_put_resource(). In normal operation, users will only 177 * have to do this for cases where they use kunit_find_resource(), and the 178 * kunit_alloc_resource() function will be used (which does not take a 179 * resource reference). 180 */ 181 static void kunit_resource_test_destroy_resource(struct kunit *test) 182 { 183 struct kunit_test_resource_context *ctx = test->priv; 184 struct kunit_resource *res = kunit_alloc_and_get_resource( 185 &ctx->test, 186 fake_resource_init, 187 fake_resource_free, 188 GFP_KERNEL, 189 ctx); 190 191 kunit_put_resource(res); 192 193 KUNIT_ASSERT_FALSE(test, 194 kunit_destroy_resource(&ctx->test, 195 kunit_resource_instance_match, 196 res->data)); 197 198 KUNIT_EXPECT_FALSE(test, ctx->is_resource_initialized); 199 KUNIT_EXPECT_TRUE(test, list_empty(&ctx->test.resources)); 200 } 201 202 static void kunit_resource_test_remove_resource(struct kunit *test) 203 { 204 struct kunit_test_resource_context *ctx = test->priv; 205 struct kunit_resource *res = kunit_alloc_and_get_resource( 206 &ctx->test, 207 fake_resource_init, 208 fake_resource_free, 209 GFP_KERNEL, 210 ctx); 211 212 /* The resource is in the list */ 213 KUNIT_EXPECT_FALSE(test, list_empty(&ctx->test.resources)); 214 215 /* Remove the resource. The pointer is still valid, but it can't be 216 * found. 217 */ 218 kunit_remove_resource(test, res); 219 KUNIT_EXPECT_TRUE(test, list_empty(&ctx->test.resources)); 220 /* We haven't been freed yet. */ 221 KUNIT_EXPECT_TRUE(test, ctx->is_resource_initialized); 222 223 /* Removing the resource multiple times is valid. */ 224 kunit_remove_resource(test, res); 225 KUNIT_EXPECT_TRUE(test, list_empty(&ctx->test.resources)); 226 /* Despite having been removed twice (from only one reference), the 227 * resource still has not been freed. 228 */ 229 KUNIT_EXPECT_TRUE(test, ctx->is_resource_initialized); 230 231 /* Free the resource. */ 232 kunit_put_resource(res); 233 KUNIT_EXPECT_FALSE(test, ctx->is_resource_initialized); 234 } 235 236 static void kunit_resource_test_cleanup_resources(struct kunit *test) 237 { 238 int i; 239 struct kunit_test_resource_context *ctx = test->priv; 240 struct kunit_resource *resources[5]; 241 242 for (i = 0; i < ARRAY_SIZE(resources); i++) { 243 resources[i] = kunit_alloc_and_get_resource(&ctx->test, 244 fake_resource_init, 245 fake_resource_free, 246 GFP_KERNEL, 247 ctx); 248 kunit_put_resource(resources[i]); 249 } 250 251 kunit_cleanup(&ctx->test); 252 253 KUNIT_EXPECT_TRUE(test, list_empty(&ctx->test.resources)); 254 } 255 256 static void kunit_resource_test_mark_order(int order_array[], 257 size_t order_size, 258 int key) 259 { 260 int i; 261 262 for (i = 0; i < order_size && order_array[i]; i++) 263 ; 264 265 order_array[i] = key; 266 } 267 268 #define KUNIT_RESOURCE_TEST_MARK_ORDER(ctx, order_field, key) \ 269 kunit_resource_test_mark_order(ctx->order_field, \ 270 ARRAY_SIZE(ctx->order_field), \ 271 key) 272 273 static int fake_resource_2_init(struct kunit_resource *res, void *context) 274 { 275 struct kunit_test_resource_context *ctx = context; 276 277 KUNIT_RESOURCE_TEST_MARK_ORDER(ctx, allocate_order, 2); 278 279 res->data = ctx; 280 281 return 0; 282 } 283 284 static void fake_resource_2_free(struct kunit_resource *res) 285 { 286 struct kunit_test_resource_context *ctx = res->data; 287 288 KUNIT_RESOURCE_TEST_MARK_ORDER(ctx, free_order, 2); 289 } 290 291 static int fake_resource_1_init(struct kunit_resource *res, void *context) 292 { 293 struct kunit_test_resource_context *ctx = context; 294 struct kunit_resource *res2; 295 296 res2 = kunit_alloc_and_get_resource(&ctx->test, 297 fake_resource_2_init, 298 fake_resource_2_free, 299 GFP_KERNEL, 300 ctx); 301 302 KUNIT_RESOURCE_TEST_MARK_ORDER(ctx, allocate_order, 1); 303 304 res->data = ctx; 305 306 kunit_put_resource(res2); 307 308 return 0; 309 } 310 311 static void fake_resource_1_free(struct kunit_resource *res) 312 { 313 struct kunit_test_resource_context *ctx = res->data; 314 315 KUNIT_RESOURCE_TEST_MARK_ORDER(ctx, free_order, 1); 316 } 317 318 /* 319 * TODO(brendanhiggins@google.com): replace the arrays that keep track of the 320 * order of allocation and freeing with strict mocks using the IN_SEQUENCE macro 321 * to assert allocation and freeing order when the feature becomes available. 322 */ 323 static void kunit_resource_test_proper_free_ordering(struct kunit *test) 324 { 325 struct kunit_test_resource_context *ctx = test->priv; 326 struct kunit_resource *res; 327 328 /* fake_resource_1 allocates a fake_resource_2 in its init. */ 329 res = kunit_alloc_and_get_resource(&ctx->test, 330 fake_resource_1_init, 331 fake_resource_1_free, 332 GFP_KERNEL, 333 ctx); 334 335 /* 336 * Since fake_resource_2_init calls KUNIT_RESOURCE_TEST_MARK_ORDER 337 * before returning to fake_resource_1_init, it should be the first to 338 * put its key in the allocate_order array. 339 */ 340 KUNIT_EXPECT_EQ(test, ctx->allocate_order[0], 2); 341 KUNIT_EXPECT_EQ(test, ctx->allocate_order[1], 1); 342 343 kunit_put_resource(res); 344 345 kunit_cleanup(&ctx->test); 346 347 /* 348 * Because fake_resource_2 finishes allocation before fake_resource_1, 349 * fake_resource_1 should be freed first since it could depend on 350 * fake_resource_2. 351 */ 352 KUNIT_EXPECT_EQ(test, ctx->free_order[0], 1); 353 KUNIT_EXPECT_EQ(test, ctx->free_order[1], 2); 354 } 355 356 static void kunit_resource_test_static(struct kunit *test) 357 { 358 struct kunit_test_resource_context ctx; 359 struct kunit_resource res; 360 361 KUNIT_EXPECT_EQ(test, kunit_add_resource(test, NULL, NULL, &res, &ctx), 362 0); 363 364 KUNIT_EXPECT_PTR_EQ(test, res.data, (void *)&ctx); 365 366 kunit_cleanup(test); 367 368 KUNIT_EXPECT_TRUE(test, list_empty(&test->resources)); 369 } 370 371 static void kunit_resource_test_named(struct kunit *test) 372 { 373 struct kunit_resource res1, res2, *found = NULL; 374 struct kunit_test_resource_context ctx; 375 376 KUNIT_EXPECT_EQ(test, 377 kunit_add_named_resource(test, NULL, NULL, &res1, 378 "resource_1", &ctx), 379 0); 380 KUNIT_EXPECT_PTR_EQ(test, res1.data, (void *)&ctx); 381 382 KUNIT_EXPECT_EQ(test, 383 kunit_add_named_resource(test, NULL, NULL, &res1, 384 "resource_1", &ctx), 385 -EEXIST); 386 387 KUNIT_EXPECT_EQ(test, 388 kunit_add_named_resource(test, NULL, NULL, &res2, 389 "resource_2", &ctx), 390 0); 391 392 found = kunit_find_named_resource(test, "resource_1"); 393 394 KUNIT_EXPECT_PTR_EQ(test, found, &res1); 395 396 if (found) 397 kunit_put_resource(&res1); 398 399 KUNIT_EXPECT_EQ(test, kunit_destroy_named_resource(test, "resource_2"), 400 0); 401 402 kunit_cleanup(test); 403 404 KUNIT_EXPECT_TRUE(test, list_empty(&test->resources)); 405 } 406 407 static void increment_int(void *ctx) 408 { 409 int *i = (int *)ctx; 410 (*i)++; 411 } 412 413 static void kunit_resource_test_action(struct kunit *test) 414 { 415 int num_actions = 0; 416 417 kunit_add_action(test, increment_int, &num_actions); 418 KUNIT_EXPECT_EQ(test, num_actions, 0); 419 kunit_cleanup(test); 420 KUNIT_EXPECT_EQ(test, num_actions, 1); 421 422 /* Once we've cleaned up, the action queue is empty. */ 423 kunit_cleanup(test); 424 KUNIT_EXPECT_EQ(test, num_actions, 1); 425 426 /* Check the same function can be deferred multiple times. */ 427 kunit_add_action(test, increment_int, &num_actions); 428 kunit_add_action(test, increment_int, &num_actions); 429 kunit_cleanup(test); 430 KUNIT_EXPECT_EQ(test, num_actions, 3); 431 } 432 static void kunit_resource_test_remove_action(struct kunit *test) 433 { 434 int num_actions = 0; 435 436 kunit_add_action(test, increment_int, &num_actions); 437 KUNIT_EXPECT_EQ(test, num_actions, 0); 438 439 kunit_remove_action(test, increment_int, &num_actions); 440 kunit_cleanup(test); 441 KUNIT_EXPECT_EQ(test, num_actions, 0); 442 } 443 static void kunit_resource_test_release_action(struct kunit *test) 444 { 445 int num_actions = 0; 446 447 kunit_add_action(test, increment_int, &num_actions); 448 KUNIT_EXPECT_EQ(test, num_actions, 0); 449 /* Runs immediately on trigger. */ 450 kunit_release_action(test, increment_int, &num_actions); 451 KUNIT_EXPECT_EQ(test, num_actions, 1); 452 453 /* Doesn't run again on test exit. */ 454 kunit_cleanup(test); 455 KUNIT_EXPECT_EQ(test, num_actions, 1); 456 } 457 static void action_order_1(void *ctx) 458 { 459 struct kunit_test_resource_context *res_ctx = (struct kunit_test_resource_context *)ctx; 460 461 KUNIT_RESOURCE_TEST_MARK_ORDER(res_ctx, free_order, 1); 462 kunit_log(KERN_INFO, current->kunit_test, "action_order_1"); 463 } 464 static void action_order_2(void *ctx) 465 { 466 struct kunit_test_resource_context *res_ctx = (struct kunit_test_resource_context *)ctx; 467 468 KUNIT_RESOURCE_TEST_MARK_ORDER(res_ctx, free_order, 2); 469 kunit_log(KERN_INFO, current->kunit_test, "action_order_2"); 470 } 471 static void kunit_resource_test_action_ordering(struct kunit *test) 472 { 473 struct kunit_test_resource_context *ctx = test->priv; 474 475 kunit_add_action(test, action_order_1, ctx); 476 kunit_add_action(test, action_order_2, ctx); 477 kunit_add_action(test, action_order_1, ctx); 478 kunit_add_action(test, action_order_2, ctx); 479 kunit_remove_action(test, action_order_1, ctx); 480 kunit_release_action(test, action_order_2, ctx); 481 kunit_cleanup(test); 482 483 /* [2 is triggered] [2], [(1 is cancelled)] [1] */ 484 KUNIT_EXPECT_EQ(test, ctx->free_order[0], 2); 485 KUNIT_EXPECT_EQ(test, ctx->free_order[1], 2); 486 KUNIT_EXPECT_EQ(test, ctx->free_order[2], 1); 487 } 488 489 static int kunit_resource_test_init(struct kunit *test) 490 { 491 struct kunit_test_resource_context *ctx = 492 kzalloc(sizeof(*ctx), GFP_KERNEL); 493 494 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx); 495 496 test->priv = ctx; 497 498 kunit_init_test(&ctx->test, "test_test_context", NULL); 499 500 return 0; 501 } 502 503 static void kunit_resource_test_exit(struct kunit *test) 504 { 505 struct kunit_test_resource_context *ctx = test->priv; 506 507 kunit_cleanup(&ctx->test); 508 kfree(ctx); 509 } 510 511 static struct kunit_case kunit_resource_test_cases[] = { 512 KUNIT_CASE(kunit_resource_test_init_resources), 513 KUNIT_CASE(kunit_resource_test_alloc_resource), 514 KUNIT_CASE(kunit_resource_test_destroy_resource), 515 KUNIT_CASE(kunit_resource_test_remove_resource), 516 KUNIT_CASE(kunit_resource_test_cleanup_resources), 517 KUNIT_CASE(kunit_resource_test_proper_free_ordering), 518 KUNIT_CASE(kunit_resource_test_static), 519 KUNIT_CASE(kunit_resource_test_named), 520 KUNIT_CASE(kunit_resource_test_action), 521 KUNIT_CASE(kunit_resource_test_remove_action), 522 KUNIT_CASE(kunit_resource_test_release_action), 523 KUNIT_CASE(kunit_resource_test_action_ordering), 524 {} 525 }; 526 527 static struct kunit_suite kunit_resource_test_suite = { 528 .name = "kunit-resource-test", 529 .init = kunit_resource_test_init, 530 .exit = kunit_resource_test_exit, 531 .test_cases = kunit_resource_test_cases, 532 }; 533 534 /* 535 * Log tests call string_stream functions, which aren't exported. So only 536 * build this code if this test is built-in. 537 */ 538 #if IS_BUILTIN(CONFIG_KUNIT_TEST) 539 540 /* This avoids a cast warning if kfree() is passed direct to kunit_add_action(). */ 541 static void kfree_wrapper(void *p) 542 { 543 kfree(p); 544 } 545 546 static void kunit_log_test(struct kunit *test) 547 { 548 struct kunit_suite suite; 549 #ifdef CONFIG_KUNIT_DEBUGFS 550 char *full_log; 551 #endif 552 suite.log = kunit_alloc_string_stream(test, GFP_KERNEL); 553 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, suite.log); 554 string_stream_set_append_newlines(suite.log, true); 555 556 kunit_log(KERN_INFO, test, "put this in log."); 557 kunit_log(KERN_INFO, test, "this too."); 558 kunit_log(KERN_INFO, &suite, "add to suite log."); 559 kunit_log(KERN_INFO, &suite, "along with this."); 560 561 #ifdef CONFIG_KUNIT_DEBUGFS 562 KUNIT_EXPECT_TRUE(test, test->log->append_newlines); 563 564 full_log = string_stream_get_string(test->log); 565 kunit_add_action(test, kfree_wrapper, full_log); 566 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, 567 strstr(full_log, "put this in log.")); 568 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, 569 strstr(full_log, "this too.")); 570 571 full_log = string_stream_get_string(suite.log); 572 kunit_add_action(test, kfree_wrapper, full_log); 573 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, 574 strstr(full_log, "add to suite log.")); 575 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, 576 strstr(full_log, "along with this.")); 577 #else 578 KUNIT_EXPECT_NULL(test, test->log); 579 #endif 580 } 581 582 static void kunit_log_newline_test(struct kunit *test) 583 { 584 char *full_log; 585 586 kunit_info(test, "Add newline\n"); 587 if (test->log) { 588 full_log = string_stream_get_string(test->log); 589 kunit_add_action(test, kfree_wrapper, full_log); 590 KUNIT_ASSERT_NOT_NULL_MSG(test, strstr(full_log, "Add newline\n"), 591 "Missing log line, full log:\n%s", full_log); 592 KUNIT_EXPECT_NULL(test, strstr(full_log, "Add newline\n\n")); 593 } else { 594 kunit_skip(test, "only useful when debugfs is enabled"); 595 } 596 } 597 #else 598 static void kunit_log_test(struct kunit *test) 599 { 600 kunit_skip(test, "Log tests only run when built-in"); 601 } 602 603 static void kunit_log_newline_test(struct kunit *test) 604 { 605 kunit_skip(test, "Log tests only run when built-in"); 606 } 607 #endif /* IS_BUILTIN(CONFIG_KUNIT_TEST) */ 608 609 static struct kunit_case kunit_log_test_cases[] = { 610 KUNIT_CASE(kunit_log_test), 611 KUNIT_CASE(kunit_log_newline_test), 612 {} 613 }; 614 615 static struct kunit_suite kunit_log_test_suite = { 616 .name = "kunit-log-test", 617 .test_cases = kunit_log_test_cases, 618 }; 619 620 static void kunit_status_set_failure_test(struct kunit *test) 621 { 622 struct kunit fake; 623 624 kunit_init_test(&fake, "fake test", NULL); 625 626 KUNIT_EXPECT_EQ(test, fake.status, (enum kunit_status)KUNIT_SUCCESS); 627 kunit_set_failure(&fake); 628 KUNIT_EXPECT_EQ(test, fake.status, (enum kunit_status)KUNIT_FAILURE); 629 } 630 631 static void kunit_status_mark_skipped_test(struct kunit *test) 632 { 633 struct kunit fake; 634 635 kunit_init_test(&fake, "fake test", NULL); 636 637 /* Before: Should be SUCCESS with no comment. */ 638 KUNIT_EXPECT_EQ(test, fake.status, KUNIT_SUCCESS); 639 KUNIT_EXPECT_STREQ(test, fake.status_comment, ""); 640 641 /* Mark the test as skipped. */ 642 kunit_mark_skipped(&fake, "Accepts format string: %s", "YES"); 643 644 /* After: Should be SKIPPED with our comment. */ 645 KUNIT_EXPECT_EQ(test, fake.status, (enum kunit_status)KUNIT_SKIPPED); 646 KUNIT_EXPECT_STREQ(test, fake.status_comment, "Accepts format string: YES"); 647 } 648 649 static struct kunit_case kunit_status_test_cases[] = { 650 KUNIT_CASE(kunit_status_set_failure_test), 651 KUNIT_CASE(kunit_status_mark_skipped_test), 652 {} 653 }; 654 655 static struct kunit_suite kunit_status_test_suite = { 656 .name = "kunit_status", 657 .test_cases = kunit_status_test_cases, 658 }; 659 660 static void kunit_current_test(struct kunit *test) 661 { 662 /* Check results of both current->kunit_test and 663 * kunit_get_current_test() are equivalent to current test. 664 */ 665 KUNIT_EXPECT_PTR_EQ(test, test, current->kunit_test); 666 KUNIT_EXPECT_PTR_EQ(test, test, kunit_get_current_test()); 667 } 668 669 static void kunit_current_fail_test(struct kunit *test) 670 { 671 struct kunit fake; 672 673 kunit_init_test(&fake, "fake test", NULL); 674 KUNIT_EXPECT_EQ(test, fake.status, KUNIT_SUCCESS); 675 676 /* Set current->kunit_test to fake test. */ 677 current->kunit_test = &fake; 678 679 kunit_fail_current_test("This should make `fake` test fail."); 680 KUNIT_EXPECT_EQ(test, fake.status, (enum kunit_status)KUNIT_FAILURE); 681 kunit_cleanup(&fake); 682 683 /* Reset current->kunit_test to current test. */ 684 current->kunit_test = test; 685 } 686 687 static struct kunit_case kunit_current_test_cases[] = { 688 KUNIT_CASE(kunit_current_test), 689 KUNIT_CASE(kunit_current_fail_test), 690 {} 691 }; 692 693 static struct kunit_suite kunit_current_test_suite = { 694 .name = "kunit_current", 695 .test_cases = kunit_current_test_cases, 696 }; 697 698 kunit_test_suites(&kunit_try_catch_test_suite, &kunit_resource_test_suite, 699 &kunit_log_test_suite, &kunit_status_test_suite, 700 &kunit_current_test_suite); 701 702 MODULE_LICENSE("GPL v2"); 703