test/ut.c

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