Blame view

include/test/ut.h 3.26 KB
d41ce506b   Eric Lee   Initial Release, ...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
  /*
   * Simple unit test library
   *
   * Copyright (c) 2013 Google, Inc
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #ifndef __TEST_UT_H
  #define __TEST_UT_H
  
  #include <linux/err.h>
  
  struct unit_test_state;
  
  /**
   * ut_fail() - Record failure of a unit test
   *
   * @uts: Test state
   * @fname: Filename where the error occurred
   * @line: Line number where the error occurred
   * @func: Function name where the error occurred
   * @cond: The condition that failed
   */
  void ut_fail(struct unit_test_state *uts, const char *fname, int line,
  	     const char *func, const char *cond);
  
  /**
   * ut_failf() - Record failure of a unit test
   *
   * @uts: Test state
   * @fname: Filename where the error occurred
   * @line: Line number where the error occurred
   * @func: Function name where the error occurred
   * @cond: The condition that failed
   * @fmt: printf() format string for the error, followed by args
   */
  void ut_failf(struct unit_test_state *uts, const char *fname, int line,
  	      const char *func, const char *cond, const char *fmt, ...)
  			__attribute__ ((format (__printf__, 6, 7)));
  
  
  /* Assert that a condition is non-zero */
  #define ut_assert(cond)							\
  	if (!(cond)) {							\
  		ut_fail(uts, __FILE__, __LINE__, __func__, #cond);	\
  		return CMD_RET_FAILURE;					\
  	}
  
  /* Assert that a condition is non-zero, with printf() string */
  #define ut_assertf(cond, fmt, args...)					\
  	if (!(cond)) {							\
  		ut_failf(uts, __FILE__, __LINE__, __func__, #cond,	\
  			 fmt, ##args);					\
  		return CMD_RET_FAILURE;					\
  	}
  
  /* Assert that two int expressions are equal */
  #define ut_asserteq(expr1, expr2) {					\
  	unsigned int val1 = (expr1), val2 = (expr2);			\
  									\
  	if (val1 != val2) {						\
  		ut_failf(uts, __FILE__, __LINE__, __func__,		\
  			 #expr1 " == " #expr2,				\
  			 "Expected %d, got %d", val1, val2);		\
  		return CMD_RET_FAILURE;					\
  	}								\
  }
  
  /* Assert that two string expressions are equal */
  #define ut_asserteq_str(expr1, expr2) {					\
  	const char *val1 = (expr1), *val2 = (expr2);			\
  									\
  	if (strcmp(val1, val2)) {					\
  		ut_failf(uts, __FILE__, __LINE__, __func__,		\
  			 #expr1 " = " #expr2,				\
  			 "Expected \"%s\", got \"%s\"", val1, val2);	\
  		return CMD_RET_FAILURE;					\
  	}								\
  }
  
  /* Assert that two pointers are equal */
  #define ut_asserteq_ptr(expr1, expr2) {					\
  	const void *val1 = (expr1), *val2 = (expr2);			\
  									\
  	if (val1 != val2) {						\
  		ut_failf(uts, __FILE__, __LINE__, __func__,		\
  			 #expr1 " = " #expr2,				\
  			 "Expected %p, got %p", val1, val2);		\
  		return CMD_RET_FAILURE;					\
  	}								\
  }
  
  /* Assert that a pointer is not NULL */
  #define ut_assertnonnull(expr) {					\
  	const void *val = (expr);					\
  									\
  	if (val == NULL) {						\
  		ut_failf(uts, __FILE__, __LINE__, __func__,		\
  			 #expr " = NULL",				\
  			 "Expected non-null, got NULL");		\
  		return CMD_RET_FAILURE;					\
  	}								\
  }
  
  /* Assert that a pointer is not an error pointer */
  #define ut_assertok_ptr(expr) {						\
  	const void *val = (expr);					\
  									\
  	if (IS_ERR(val)) {						\
  		ut_failf(uts, __FILE__, __LINE__, __func__,		\
  			 #expr " = NULL",				\
  			 "Expected pointer, got error %ld",		\
  			 PTR_ERR(val));					\
  		return CMD_RET_FAILURE;					\
  	}								\
  }
  
  /* Assert that an operation succeeds (returns 0) */
  #define ut_assertok(cond)	ut_asserteq(0, cond)
  
  #endif