executor_test.c 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * KUnit test for the KUnit executor.
  4. *
  5. * Copyright (C) 2021, Google LLC.
  6. * Author: Daniel Latypov <[email protected]>
  7. */
  8. #include <kunit/test.h>
  9. static void kfree_at_end(struct kunit *test, const void *to_free);
  10. static struct kunit_suite *alloc_fake_suite(struct kunit *test,
  11. const char *suite_name,
  12. struct kunit_case *test_cases);
  13. static void dummy_test(struct kunit *test) {}
  14. static struct kunit_case dummy_test_cases[] = {
  15. /* .run_case is not important, just needs to be non-NULL */
  16. { .name = "test1", .run_case = dummy_test },
  17. { .name = "test2", .run_case = dummy_test },
  18. {},
  19. };
  20. static void parse_filter_test(struct kunit *test)
  21. {
  22. struct kunit_test_filter filter = {NULL, NULL};
  23. kunit_parse_filter_glob(&filter, "suite");
  24. KUNIT_EXPECT_STREQ(test, filter.suite_glob, "suite");
  25. KUNIT_EXPECT_FALSE(test, filter.test_glob);
  26. kfree(filter.suite_glob);
  27. kfree(filter.test_glob);
  28. kunit_parse_filter_glob(&filter, "suite.test");
  29. KUNIT_EXPECT_STREQ(test, filter.suite_glob, "suite");
  30. KUNIT_EXPECT_STREQ(test, filter.test_glob, "test");
  31. kfree(filter.suite_glob);
  32. kfree(filter.test_glob);
  33. }
  34. static void filter_suites_test(struct kunit *test)
  35. {
  36. struct kunit_suite *subsuite[3] = {NULL, NULL};
  37. struct suite_set suite_set = {.start = subsuite, .end = &subsuite[2]};
  38. struct suite_set got;
  39. int err = 0;
  40. subsuite[0] = alloc_fake_suite(test, "suite1", dummy_test_cases);
  41. subsuite[1] = alloc_fake_suite(test, "suite2", dummy_test_cases);
  42. /* Want: suite1, suite2, NULL -> suite2, NULL */
  43. got = kunit_filter_suites(&suite_set, "suite2", &err);
  44. KUNIT_ASSERT_NOT_ERR_OR_NULL(test, got.start);
  45. KUNIT_ASSERT_EQ(test, err, 0);
  46. kfree_at_end(test, got.start);
  47. /* Validate we just have suite2 */
  48. KUNIT_ASSERT_NOT_ERR_OR_NULL(test, got.start[0]);
  49. KUNIT_EXPECT_STREQ(test, (const char *)got.start[0]->name, "suite2");
  50. /* Contains one element (end is 1 past end) */
  51. KUNIT_ASSERT_EQ(test, got.end - got.start, 1);
  52. }
  53. static void filter_suites_test_glob_test(struct kunit *test)
  54. {
  55. struct kunit_suite *subsuite[3] = {NULL, NULL};
  56. struct suite_set suite_set = {.start = subsuite, .end = &subsuite[2]};
  57. struct suite_set got;
  58. int err = 0;
  59. subsuite[0] = alloc_fake_suite(test, "suite1", dummy_test_cases);
  60. subsuite[1] = alloc_fake_suite(test, "suite2", dummy_test_cases);
  61. /* Want: suite1, suite2, NULL -> suite2 (just test1), NULL */
  62. got = kunit_filter_suites(&suite_set, "suite2.test2", &err);
  63. KUNIT_ASSERT_NOT_ERR_OR_NULL(test, got.start);
  64. KUNIT_ASSERT_EQ(test, err, 0);
  65. kfree_at_end(test, got.start);
  66. /* Validate we just have suite2 */
  67. KUNIT_ASSERT_NOT_ERR_OR_NULL(test, got.start[0]);
  68. KUNIT_EXPECT_STREQ(test, (const char *)got.start[0]->name, "suite2");
  69. KUNIT_ASSERT_EQ(test, got.end - got.start, 1);
  70. /* Now validate we just have test2 */
  71. KUNIT_ASSERT_NOT_ERR_OR_NULL(test, got.start[0]->test_cases);
  72. KUNIT_EXPECT_STREQ(test, (const char *)got.start[0]->test_cases[0].name, "test2");
  73. KUNIT_EXPECT_FALSE(test, got.start[0]->test_cases[1].name);
  74. }
  75. static void filter_suites_to_empty_test(struct kunit *test)
  76. {
  77. struct kunit_suite *subsuite[3] = {NULL, NULL};
  78. struct suite_set suite_set = {.start = subsuite, .end = &subsuite[2]};
  79. struct suite_set got;
  80. int err = 0;
  81. subsuite[0] = alloc_fake_suite(test, "suite1", dummy_test_cases);
  82. subsuite[1] = alloc_fake_suite(test, "suite2", dummy_test_cases);
  83. got = kunit_filter_suites(&suite_set, "not_found", &err);
  84. KUNIT_ASSERT_EQ(test, err, 0);
  85. kfree_at_end(test, got.start); /* just in case */
  86. KUNIT_EXPECT_PTR_EQ_MSG(test, got.start, got.end,
  87. "should be empty to indicate no match");
  88. }
  89. static struct kunit_case executor_test_cases[] = {
  90. KUNIT_CASE(parse_filter_test),
  91. KUNIT_CASE(filter_suites_test),
  92. KUNIT_CASE(filter_suites_test_glob_test),
  93. KUNIT_CASE(filter_suites_to_empty_test),
  94. {}
  95. };
  96. static struct kunit_suite executor_test_suite = {
  97. .name = "kunit_executor_test",
  98. .test_cases = executor_test_cases,
  99. };
  100. kunit_test_suites(&executor_test_suite);
  101. /* Test helpers */
  102. static void kfree_res_free(struct kunit_resource *res)
  103. {
  104. kfree(res->data);
  105. }
  106. /* Use the resource API to register a call to kfree(to_free).
  107. * Since we never actually use the resource, it's safe to use on const data.
  108. */
  109. static void kfree_at_end(struct kunit *test, const void *to_free)
  110. {
  111. /* kfree() handles NULL already, but avoid allocating a no-op cleanup. */
  112. if (IS_ERR_OR_NULL(to_free))
  113. return;
  114. kunit_alloc_resource(test, NULL, kfree_res_free, GFP_KERNEL,
  115. (void *)to_free);
  116. }
  117. static struct kunit_suite *alloc_fake_suite(struct kunit *test,
  118. const char *suite_name,
  119. struct kunit_case *test_cases)
  120. {
  121. struct kunit_suite *suite;
  122. /* We normally never expect to allocate suites, hence the non-const cast. */
  123. suite = kunit_kzalloc(test, sizeof(*suite), GFP_KERNEL);
  124. strncpy((char *)suite->name, suite_name, sizeof(suite->name) - 1);
  125. suite->test_cases = test_cases;
  126. return suite;
  127. }