ut.c 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  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. #include <test/test.h>
  11. #include <test/ut.h>
  12. DECLARE_GLOBAL_DATA_PTR;
  13. void ut_fail(struct unit_test_state *uts, const char *fname, int line,
  14. const char *func, const char *cond)
  15. {
  16. gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
  17. printf("%s:%d, %s(): %s\n", fname, line, func, cond);
  18. uts->fail_count++;
  19. }
  20. void ut_failf(struct unit_test_state *uts, const char *fname, int line,
  21. const char *func, const char *cond, const char *fmt, ...)
  22. {
  23. va_list args;
  24. gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
  25. printf("%s:%d, %s(): %s: ", fname, line, func, cond);
  26. va_start(args, fmt);
  27. vprintf(fmt, args);
  28. va_end(args);
  29. putc('\n');
  30. uts->fail_count++;
  31. }
  32. ulong ut_check_free(void)
  33. {
  34. struct mallinfo info = mallinfo();
  35. return info.uordblks;
  36. }
  37. long ut_check_delta(ulong last)
  38. {
  39. return ut_check_free() - last;
  40. }
  41. int ut_check_console_line(struct unit_test_state *uts, const char *fmt, ...)
  42. {
  43. va_list args;
  44. va_start(args, fmt);
  45. vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args);
  46. va_end(args);
  47. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  48. return strcmp(uts->expect_str, uts->actual_str);
  49. }
  50. int ut_check_console_linen(struct unit_test_state *uts, const char *fmt, ...)
  51. {
  52. va_list args;
  53. va_start(args, fmt);
  54. vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args);
  55. va_end(args);
  56. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  57. return strncmp(uts->expect_str, uts->actual_str,
  58. strlen(uts->expect_str));
  59. }
  60. int ut_check_skipline(struct unit_test_state *uts)
  61. {
  62. if (!console_record_avail())
  63. return -ENFILE;
  64. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  65. return 0;
  66. }
  67. int ut_check_console_end(struct unit_test_state *uts)
  68. {
  69. if (!console_record_avail())
  70. return 0;
  71. console_record_readline(uts->actual_str, sizeof(uts->actual_str));
  72. return 1;
  73. }
  74. int ut_check_console_dump(struct unit_test_state *uts, int total_bytes)
  75. {
  76. char *str = uts->actual_str;
  77. int upto;
  78. /* Handle empty dump */
  79. if (!total_bytes)
  80. return 0;
  81. for (upto = 0; upto < total_bytes;) {
  82. int len;
  83. int bytes;
  84. len = console_record_readline(str, sizeof(uts->actual_str));
  85. if (str[8] != ':' || str[9] != ' ')
  86. return 1;
  87. bytes = len - 8 - 2 - 3 * 16 - 4;
  88. upto += bytes;
  89. }
  90. return upto == total_bytes ? 0 : 1;
  91. }