ut.c 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Simple unit test library
  4. *
  5. * Copyright (c) 2013 Google, Inc
  6. */
  7. #include <common.h>
  8. #include <console.h>
  9. #include <malloc.h>
  10. #ifdef CONFIG_SANDBOX
  11. #include <asm/state.h>
  12. #endif
  13. #include <test/test.h>
  14. #include <test/ut.h>
  15. DECLARE_GLOBAL_DATA_PTR;
  16. void ut_fail(struct unit_test_state *uts, const char *fname, int line,
  17. const char *func, const char *cond)
  18. {
  19. gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
  20. printf("%s:%d, %s(): %s\n", fname, line, func, cond);
  21. uts->fail_count++;
  22. }
  23. void ut_failf(struct unit_test_state *uts, const char *fname, int line,
  24. const char *func, const char *cond, const char *fmt, ...)
  25. {
  26. va_list args;
  27. gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
  28. printf("%s:%d, %s(): %s: ", fname, line, func, cond);
  29. va_start(args, fmt);
  30. vprintf(fmt, args);
  31. va_end(args);
  32. putc('\n');
  33. uts->fail_count++;
  34. }
  35. ulong ut_check_free(void)
  36. {
  37. struct mallinfo info = mallinfo();
  38. return info.uordblks;
  39. }
  40. long ut_check_delta(ulong last)
  41. {
  42. return ut_check_free() - last;
  43. }
  44. int ut_check_console_line(struct unit_test_state *uts, const char *fmt, ...)
  45. {
  46. va_list args;
  47. va_start(args, fmt);
  48. vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args);
  49. va_end(args);
  50. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  51. return strcmp(uts->expect_str, uts->actual_str);
  52. }
  53. int ut_check_console_linen(struct unit_test_state *uts, const char *fmt, ...)
  54. {
  55. va_list args;
  56. va_start(args, fmt);
  57. vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args);
  58. va_end(args);
  59. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  60. return strncmp(uts->expect_str, uts->actual_str,
  61. strlen(uts->expect_str));
  62. }
  63. int ut_check_skipline(struct unit_test_state *uts)
  64. {
  65. if (!console_record_avail())
  66. return -ENFILE;
  67. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  68. return 0;
  69. }
  70. int ut_check_console_end(struct unit_test_state *uts)
  71. {
  72. if (!console_record_avail())
  73. return 0;
  74. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  75. return 1;
  76. }
  77. int ut_check_console_dump(struct unit_test_state *uts, int total_bytes)
  78. {
  79. char *str = uts->actual_str;
  80. int upto;
  81. /* Handle empty dump */
  82. if (!total_bytes)
  83. return 0;
  84. for (upto = 0; upto < total_bytes;) {
  85. int len;
  86. int bytes;
  87. len = console_record_readline(str, sizeof(uts->actual_str));
  88. if (str[8] != ':' || str[9] != ' ')
  89. return 1;
  90. bytes = len - 8 - 2 - 3 * 16 - 4;
  91. upto += bytes;
  92. }
  93. return upto == total_bytes ? 0 : 1;
  94. }
  95. void ut_silence_console(struct unit_test_state *uts)
  96. {
  97. #ifdef CONFIG_SANDBOX
  98. struct sandbox_state *state = state_get_current();
  99. if (!state->show_test_output)
  100. gd->flags |= GD_FLG_SILENT;
  101. #endif
  102. }
  103. void ut_unsilence_console(struct unit_test_state *uts)
  104. {
  105. gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
  106. }