Add support for expectations, which allow properties to be specified and
then verified in tests.

Signed-off-by: Brendan Higgins <brendanhigg...@google.com>
Reviewed-by: Greg Kroah-Hartman <gre...@linuxfoundation.org>
Reviewed-by: Logan Gunthorpe <log...@deltatee.com>
---
 include/kunit/test.h | 525 +++++++++++++++++++++++++++++++++++++++++++
 kunit/test.c         |  66 ++++++
 2 files changed, 591 insertions(+)

diff --git a/include/kunit/test.h b/include/kunit/test.h
index a96a166c4808c..e1729f19174c1 100644
--- a/include/kunit/test.h
+++ b/include/kunit/test.h
@@ -9,6 +9,7 @@
 #ifndef _KUNIT_TEST_H
 #define _KUNIT_TEST_H
 
+#include <linux/kernel.h>
 #include <linux/types.h>
 #include <linux/slab.h>
 #include <kunit/kunit-stream.h>
@@ -293,4 +294,528 @@ void __printf(3, 4) kunit_printk(const char *level,
 #define kunit_err(test, fmt, ...) \
                kunit_printk(KERN_ERR, test, fmt, ##__VA_ARGS__)
 
+/*
+ * Generates a compile-time warning in case of comparing incompatible types.
+ */
+#define __kunit_typecheck(lhs, rhs) \
+       ((void) __typecheck(lhs, rhs))
+
+static inline struct kunit_stream *kunit_expect_start(struct kunit *test,
+                                                     const char *file,
+                                                     const char *line)
+{
+       struct kunit_stream *stream = alloc_kunit_stream(test, KERN_ERR);
+
+       kunit_stream_add(stream, "EXPECTATION FAILED at %s:%s\n\t", file, line);
+
+       return stream;
+}
+
+static inline void kunit_expect_end(struct kunit *test,
+                                   bool success,
+                                   struct kunit_stream *stream)
+{
+       if (!success)
+               kunit_fail(test, stream);
+       else
+               kunit_stream_clear(stream);
+}
+
+#define KUNIT_EXPECT_START(test) \
+               kunit_expect_start(test, __FILE__, __stringify(__LINE__))
+
+#define KUNIT_EXPECT_END(test, success, stream) \
+               kunit_expect_end(test, success, stream)
+
+#define KUNIT_EXPECT_MSG(test, success, message, fmt, ...) do {                
       \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+                                                                              \
+       kunit_stream_add(__stream, message);                                   \
+       kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                        \
+       KUNIT_EXPECT_END(test, success, __stream);                             \
+} while (0)
+
+#define KUNIT_EXPECT(test, success, message) do {                             \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+                                                                              \
+       kunit_stream_add(__stream, message);                                   \
+       KUNIT_EXPECT_END(test, success, __stream);                             \
+} while (0)
+
+/**
+ * KUNIT_SUCCEED() - A no-op expectation. Only exists for code clarity.
+ * @test: The test context object.
+ *
+ * The opposite of KUNIT_FAIL(), it is an expectation that cannot fail. In 
other
+ * words, it does nothing and only exists for code clarity. See
+ * KUNIT_EXPECT_TRUE() for more information.
+ */
+#define KUNIT_SUCCEED(test) do {} while (0)
+
+/**
+ * KUNIT_FAIL() - Always causes a test to fail when evaluated.
+ * @test: The test context object.
+ * @fmt: an informational message to be printed when the assertion is made.
+ * @...: string format arguments.
+ *
+ * The opposite of KUNIT_SUCCEED(), it is an expectation that always fails. In
+ * other words, it always results in a failed expectation, and consequently
+ * always causes the test case to fail when evaluated. See KUNIT_EXPECT_TRUE()
+ * for more information.
+ */
+#define KUNIT_FAIL(test, fmt, ...) do {                                        
       \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+                                                                              \
+       kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                        \
+       KUNIT_EXPECT_END(test, false, __stream);                               \
+} while (0)
+
+/**
+ * KUNIT_EXPECT_TRUE() - Causes a test failure when the expression is not true.
+ * @test: The test context object.
+ * @condition: an arbitrary boolean expression. The test fails when this does
+ * not evaluate to true.
+ *
+ * This and expectations of the form `KUNIT_EXPECT_*` will cause the test case
+ * to fail when the specified condition is not met; however, it will not 
prevent
+ * the test case from continuing to run; this is otherwise known as an
+ * *expectation failure*.
+ */
+#define KUNIT_EXPECT_TRUE(test, condition)                                    \
+               KUNIT_EXPECT(test, (condition),                                \
+                      "Expected " #condition " is true, but is false\n")
+
+#define KUNIT_EXPECT_TRUE_MSG(test, condition, fmt, ...)                      \
+               KUNIT_EXPECT_MSG(test, (condition),                            \
+                               "Expected " #condition " is true, but is 
false\n",\
+                               fmt, ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_FALSE() - Makes a test failure when the expression is not 
false.
+ * @test: The test context object.
+ * @condition: an arbitrary boolean expression. The test fails when this does
+ * not evaluate to false.
+ *
+ * Sets an expectation that @condition evaluates to false. See
+ * KUNIT_EXPECT_TRUE() for more information.
+ */
+#define KUNIT_EXPECT_FALSE(test, condition)                                   \
+               KUNIT_EXPECT(test, !(condition),                               \
+                      "Expected " #condition " is false, but is true\n")
+
+#define KUNIT_EXPECT_FALSE_MSG(test, condition, fmt, ...)                     \
+               KUNIT_EXPECT_MSG(test, !(condition),                           \
+                               "Expected " #condition " is false, but is 
true\n",\
+                               fmt, ##__VA_ARGS__)
+
+void kunit_expect_binary_msg(struct kunit *test,
+                            long long left, const char *left_name,
+                            long long right, const char *right_name,
+                            bool compare_result,
+                            const char *compare_name,
+                            const char *file,
+                            const char *line,
+                            const char *fmt, ...);
+
+static inline void kunit_expect_binary(struct kunit *test,
+                                      long long left, const char *left_name,
+                                      long long right, const char *right_name,
+                                      bool compare_result,
+                                      const char *compare_name,
+                                      const char *file,
+                                      const char *line)
+{
+       kunit_expect_binary_msg(test,
+                               left, left_name,
+                               right, right_name,
+                               compare_result,
+                               compare_name,
+                               file,
+                               line,
+                               NULL);
+}
+
+void kunit_expect_ptr_binary_msg(struct kunit *test,
+                                void *left, const char *left_name,
+                                void *right, const char *right_name,
+                                bool compare_result,
+                                const char *compare_name,
+                                const char *file,
+                                const char *line,
+                                const char *fmt, ...);
+
+static inline void kunit_expect_ptr_binary(struct kunit *test,
+                                          void *left, const char *left_name,
+                                          void *right, const char *right_name,
+                                          bool compare_result,
+                                          const char *compare_name,
+                                          const char *file,
+                                          const char *line)
+{
+       kunit_expect_ptr_binary_msg(test,
+                                   left, left_name,
+                                   right, right_name,
+                                   compare_result,
+                                   compare_name,
+                                   file,
+                                   line,
+                                   NULL);
+}
+
+/*
+ * A factory macro for defining the expectations for the basic comparisons
+ * defined for the built in types.
+ *
+ * Unfortunately, there is no common type that all types can be promoted to for
+ * which all the binary operators behave the same way as for the actual types
+ * (for example, there is no type that long long and unsigned long long can
+ * both be cast to where the comparison result is preserved for all values). So
+ * the best we can do is do the comparison in the original types and then 
coerce
+ * everything to long long for printing; this way, the comparison behaves
+ * correctly and the printed out value usually makes sense without
+ * interpretation, but can always be interpretted to figure out the actual
+ * value.
+ */
+#define KUNIT_EXPECT_BINARY(test, left, condition, right) do {                \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+       __kunit_typecheck(__left, __right);                                    \
+       kunit_expect_binary(test,                                              \
+                           (long long) __left, #left,                         \
+                           (long long) __right, #right,                       \
+                           __left condition __right, #condition,              \
+                           __FILE__, __stringify(__LINE__));                  \
+} while (0)
+
+#define KUNIT_EXPECT_BINARY_MSG(test, left, condition, right, fmt, ...) do {   
\
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+       __kunit_typecheck(__left, __right);                                    \
+       kunit_expect_binary_msg(test,                                          \
+                               (long long) __left, #left,                     \
+                               (long long) __right, #right,                   \
+                               __left condition __right, #condition,          \
+                               __FILE__, __stringify(__LINE__),               \
+                               fmt, ##__VA_ARGS__);                           \
+} while (0)
+
+/*
+ * Just like KUNIT_EXPECT_BINARY, but for comparing pointer types.
+ */
+#define KUNIT_EXPECT_PTR_BINARY(test, left, condition, right) do {            \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+       __kunit_typecheck(__left, __right);                                    \
+       kunit_expect_ptr_binary(test,                                          \
+                           (void *) __left, #left,                            \
+                           (void *) __right, #right,                          \
+                           __left condition __right, #condition,              \
+                           __FILE__, __stringify(__LINE__));                  \
+} while (0)
+
+#define KUNIT_EXPECT_PTR_BINARY_MSG(test, left, condition, right, fmt, ...)    
\
+do {                                                                          \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+       __kunit_typecheck(__left, __right);                                    \
+       kunit_expect_ptr_binary_msg(test,                                      \
+                                   (void *) __left, #left,                    \
+                                   (void *) __right, #right,                  \
+                                   __left condition __right, #condition,      \
+                                   __FILE__, __stringify(__LINE__),           \
+                                   fmt, ##__VA_ARGS__);                       \
+} while (0)
+
+/**
+ * KUNIT_EXPECT_EQ() - Sets an expectation that @left and @right are equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are
+ * equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) == (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_EQ(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, ==, right)
+
+#define KUNIT_EXPECT_EQ_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       ==,                                    \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_PTR_EQ() - Expects that pointers @left and @right are equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a pointer.
+ * @right: an arbitrary expression that evaluates to a pointer.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are
+ * equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) == (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_PTR_EQ(test, left, right) \
+               KUNIT_EXPECT_PTR_BINARY(test, left, ==, right)
+
+#define KUNIT_EXPECT_PTR_EQ_MSG(test, left, right, fmt, ...)                  \
+               KUNIT_EXPECT_PTR_BINARY_MSG(test,                              \
+                                           left,                              \
+                                           ==,                                \
+                                           right,                             \
+                                           fmt,                               \
+                                           ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_NE() - An expectation that @left and @right are not equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are 
not
+ * equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) != (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_NE(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, !=, right)
+
+#define KUNIT_EXPECT_NE_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       !=,                                    \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_PTR_NE() - Expects that pointers @left and @right are not 
equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a pointer.
+ * @right: an arbitrary expression that evaluates to a pointer.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are 
not
+ * equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) != (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_PTR_NE(test, left, right) \
+               KUNIT_EXPECT_PTR_BINARY(test, left, !=, right)
+
+#define KUNIT_EXPECT_PTR_NE_MSG(test, left, right, fmt, ...)                  \
+               KUNIT_EXPECT_PTR_BINARY_MSG(test,                              \
+                                           left,                              \
+                                           !=,                                \
+                                           right,                             \
+                                           fmt,                               \
+                                           ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_LT() - An expectation that @left is less than @right.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the value that @left evaluates to is less than the
+ * value that @right evaluates to. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) < (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_LT(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, <, right)
+
+#define KUNIT_EXPECT_LT_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       <,                                     \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_LE() - Expects that @left is less than or equal to @right.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the value that @left evaluates to is less than or
+ * equal to the value that @right evaluates to. Semantically this is equivalent
+ * to KUNIT_EXPECT_TRUE(@test, (@left) <= (@right)). See KUNIT_EXPECT_TRUE() 
for
+ * more information.
+ */
+#define KUNIT_EXPECT_LE(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, <=, right)
+
+#define KUNIT_EXPECT_LE_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       <=,                                    \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_GT() - An expectation that @left is greater than @right.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the value that @left evaluates to is greater than
+ * the value that @right evaluates to. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) > (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_GT(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, >, right)
+
+#define KUNIT_EXPECT_GT_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       >,                                     \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_GE() - Expects that @left is greater than or equal to @right.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a primitive C type.
+ * @right: an arbitrary expression that evaluates to a primitive C type.
+ *
+ * Sets an expectation that the value that @left evaluates to is greater than
+ * the value that @right evaluates to. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, (@left) >= (@right)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_GE(test, left, right) \
+               KUNIT_EXPECT_BINARY(test, left, >=, right)
+
+#define KUNIT_EXPECT_GE_MSG(test, left, right, fmt, ...)                      \
+               KUNIT_EXPECT_BINARY_MSG(test,                                  \
+                                       left,                                  \
+                                       >=,                                    \
+                                       right,                                 \
+                                       fmt,                                   \
+                                       ##__VA_ARGS__)
+
+/**
+ * KUNIT_EXPECT_STREQ() - Expects that strings @left and @right are equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a null terminated string.
+ * @right: an arbitrary expression that evaluates to a null terminated string.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are
+ * equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, !strcmp((@left), (@right))). See 
KUNIT_EXPECT_TRUE()
+ * for more information.
+ */
+#define KUNIT_EXPECT_STREQ(test, left, right) do {                            \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+                                                                              \
+       kunit_stream_add(__stream, "Expected " #left " == " #right ", but\n"); \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #left, __left);           \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #right, __right);         \
+                                                                              \
+       KUNIT_EXPECT_END(test, !strcmp(left, right), __stream);                \
+} while (0)
+
+#define KUNIT_EXPECT_STREQ_MSG(test, left, right, fmt, ...) do {              \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+                                                                              \
+       kunit_stream_add(__stream, "Expected " #left " == " #right ", but\n"); \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #left, __left);           \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #right, __right);         \
+       kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                        \
+                                                                              \
+       KUNIT_EXPECT_END(test, !strcmp(left, right), __stream);                \
+} while (0)
+
+/**
+ * KUNIT_EXPECT_STRNEQ() - Expects that strings @left and @right are not equal.
+ * @test: The test context object.
+ * @left: an arbitrary expression that evaluates to a null terminated string.
+ * @right: an arbitrary expression that evaluates to a null terminated string.
+ *
+ * Sets an expectation that the values that @left and @right evaluate to are
+ * not equal. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, strcmp((@left), (@right))). See KUNIT_EXPECT_TRUE()
+ * for more information.
+ */
+#define KUNIT_EXPECT_STRNEQ(test, left, right) do {                           \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+                                                                              \
+       kunit_stream_add(__stream, "Expected " #left " != " #right ", but\n"); \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #left, __left);           \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #right, __right);         \
+                                                                              \
+       KUNIT_EXPECT_END(test, strcmp(left, right), __stream);                 \
+} while (0)
+
+#define KUNIT_EXPECT_STRNEQ_MSG(test, left, right, fmt, ...) do {             \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(left) __left = (left);                                          \
+       typeof(right) __right = (right);                                       \
+                                                                              \
+       kunit_stream_add(__stream, "Expected " #left " != " #right ", but\n"); \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #left, __left);           \
+       kunit_stream_add(__stream, "\t\t%s == %s\n", #right, __right);         \
+       kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                        \
+                                                                              \
+       KUNIT_EXPECT_END(test, strcmp(left, right), __stream);                 \
+} while (0)
+
+/**
+ * KUNIT_EXPECT_NOT_ERR_OR_NULL() - Expects that @ptr is not null and not err.
+ * @test: The test context object.
+ * @ptr: an arbitrary pointer.
+ *
+ * Sets an expectation that the value that @ptr evaluates to is not null and 
not
+ * an errno stored in a pointer. This is semantically equivalent to
+ * KUNIT_EXPECT_TRUE(@test, !IS_ERR_OR_NULL(@ptr)). See KUNIT_EXPECT_TRUE() for
+ * more information.
+ */
+#define KUNIT_EXPECT_NOT_ERR_OR_NULL(test, ptr) do {                          \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(ptr) __ptr = (ptr);                                             \
+                                                                              \
+       if (!__ptr)                                                            \
+               kunit_stream_add(__stream,                                     \
+                             "Expected " #ptr " is not null, but is\n");      \
+       if (IS_ERR(__ptr))                                                     \
+               kunit_stream_add(__stream,                                     \
+                             "Expected " #ptr " is not error, but is: %ld",   \
+                             PTR_ERR(__ptr));                                 \
+                                                                              \
+       KUNIT_EXPECT_END(test, !IS_ERR_OR_NULL(__ptr), __stream);              \
+} while (0)
+
+#define KUNIT_EXPECT_NOT_ERR_OR_NULL_MSG(test, ptr, fmt, ...) do {            \
+       struct kunit_stream *__stream = KUNIT_EXPECT_START(test);              \
+       typeof(ptr) __ptr = (ptr);                                             \
+                                                                              \
+       if (!__ptr) {                                                          \
+               kunit_stream_add(__stream,                                     \
+                             "Expected " #ptr " is not null, but is\n");      \
+               kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                \
+       }                                                                      \
+       if (IS_ERR(__ptr)) {                                                   \
+               kunit_stream_add(__stream,                                     \
+                             "Expected " #ptr " is not error, but is: %ld",   \
+                             PTR_ERR(__ptr));                                 \
+                                                                              \
+               kunit_stream_add(__stream, fmt, ##__VA_ARGS__);                \
+       }                                                                      \
+       KUNIT_EXPECT_END(test, !IS_ERR_OR_NULL(__ptr), __stream);              \
+} while (0)
+
 #endif /* _KUNIT_TEST_H */
diff --git a/kunit/test.c b/kunit/test.c
index c6a9e89ae3048..cd8ef5ce12e92 100644
--- a/kunit/test.c
+++ b/kunit/test.c
@@ -287,3 +287,69 @@ void kunit_printk(const char *level,
 
        va_end(args);
 }
+
+void kunit_expect_binary_msg(struct kunit *test,
+                            long long left, const char *left_name,
+                            long long right, const char *right_name,
+                            bool compare_result,
+                            const char *compare_name,
+                            const char *file,
+                            const char *line,
+                            const char *fmt, ...)
+{
+       struct kunit_stream *stream = kunit_expect_start(test, file, line);
+       struct va_format vaf;
+       va_list args;
+
+       kunit_stream_add(stream,
+                        "Expected %s %s %s, but\n",
+                        left_name, compare_name, right_name);
+       kunit_stream_add(stream, "\t\t%s == %lld\n", left_name, left);
+       kunit_stream_add(stream, "\t\t%s == %lld", right_name, right);
+
+       if (fmt) {
+               va_start(args, fmt);
+
+               vaf.fmt = fmt;
+               vaf.va = &args;
+
+               kunit_stream_add(stream, "\n%pV", &vaf);
+
+               va_end(args);
+       }
+
+       kunit_expect_end(test, compare_result, stream);
+}
+
+void kunit_expect_ptr_binary_msg(struct kunit *test,
+                                void *left, const char *left_name,
+                                void *right, const char *right_name,
+                                bool compare_result,
+                                const char *compare_name,
+                                const char *file,
+                                const char *line,
+                                const char *fmt, ...)
+{
+       struct kunit_stream *stream = kunit_expect_start(test, file, line);
+       struct va_format vaf;
+       va_list args;
+
+       kunit_stream_add(stream,
+                        "Expected %s %s %s, but\n",
+                        left_name, compare_name, right_name);
+       kunit_stream_add(stream, "\t\t%s == %pK\n", left_name, left);
+       kunit_stream_add(stream, "\t\t%s == %pK", right_name, right);
+
+       if (fmt) {
+               va_start(args, fmt);
+
+               vaf.fmt = fmt;
+               vaf.va = &args;
+
+               kunit_stream_add(stream, "\n%pV", &vaf);
+
+               va_end(args);
+       }
+
+       kunit_expect_end(test, compare_result, stream);
+}
-- 
2.22.0.410.gd8fdbe21b5-goog

Reply via email to