Add KUnit tests for the KUnit test abort mechanism (see preceding
commit). Add tests both for general try catch mechanism as well as
non-architecture specific mechanism.

Signed-off-by: Brendan Higgins <brendanhigg...@google.com>
Reviewed-by: Greg Kroah-Hartman <gre...@linuxfoundation.org>
Reviewed-by: Logan Gunthorpe <log...@deltatee.com>
Reviewed-by: Stephen Boyd <sb...@kernel.org>
---
 kunit/Makefile    |   3 +-
 kunit/test-test.c | 106 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 108 insertions(+), 1 deletion(-)
 create mode 100644 kunit/test-test.c

diff --git a/kunit/Makefile b/kunit/Makefile
index c9176c9c578c6..769d9402b5d3a 100644
--- a/kunit/Makefile
+++ b/kunit/Makefile
@@ -3,6 +3,7 @@ obj-$(CONFIG_KUNIT) +=                  test.o \
                                        assert.o \
                                        try-catch.o
 
-obj-$(CONFIG_KUNIT_TEST) +=            string-stream-test.o
+obj-$(CONFIG_KUNIT_TEST) +=            test-test.o \
+                                       string-stream-test.o
 
 obj-$(CONFIG_KUNIT_EXAMPLE_TEST) +=    example-test.o
diff --git a/kunit/test-test.c b/kunit/test-test.c
new file mode 100644
index 0000000000000..06d34d36b1038
--- /dev/null
+++ b/kunit/test-test.c
@@ -0,0 +1,106 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * KUnit test for core test infrastructure.
+ *
+ * Copyright (C) 2019, Google LLC.
+ * Author: Brendan Higgins <brendanhigg...@google.com>
+ */
+#include <kunit/test.h>
+
+struct kunit_try_catch_test_context {
+       struct kunit_try_catch *try_catch;
+       bool function_called;
+};
+
+static void kunit_test_successful_try(void *data)
+{
+       struct kunit *test = data;
+       struct kunit_try_catch_test_context *ctx = test->priv;
+
+       ctx->function_called = true;
+}
+
+static void kunit_test_no_catch(void *data)
+{
+       struct kunit *test = data;
+
+       KUNIT_FAIL(test, "Catch should not be called\n");
+}
+
+static void kunit_test_try_catch_successful_try_no_catch(struct kunit *test)
+{
+       struct kunit_try_catch_test_context *ctx = test->priv;
+       struct kunit_try_catch *try_catch = ctx->try_catch;
+
+       kunit_try_catch_init(try_catch,
+                            test,
+                            kunit_test_successful_try,
+                            kunit_test_no_catch);
+       kunit_try_catch_run(try_catch, test);
+
+       KUNIT_EXPECT_TRUE(test, ctx->function_called);
+}
+
+static void kunit_test_unsuccessful_try(void *data)
+{
+       struct kunit *test = data;
+       struct kunit_try_catch_test_context *ctx = test->priv;
+       struct kunit_try_catch *try_catch = ctx->try_catch;
+
+       kunit_try_catch_throw(try_catch);
+       KUNIT_FAIL(test, "This line should never be reached\n");
+}
+
+static void kunit_test_catch(void *data)
+{
+       struct kunit *test = data;
+       struct kunit_try_catch_test_context *ctx = test->priv;
+
+       ctx->function_called = true;
+}
+
+static void kunit_test_try_catch_unsuccessful_try_does_catch(struct kunit 
*test)
+{
+       struct kunit_try_catch_test_context *ctx = test->priv;
+       struct kunit_try_catch *try_catch = ctx->try_catch;
+
+       kunit_try_catch_init(try_catch,
+                            test,
+                            kunit_test_unsuccessful_try,
+                            kunit_test_catch);
+       kunit_try_catch_run(try_catch, test);
+
+       KUNIT_EXPECT_TRUE(test, ctx->function_called);
+}
+
+static int kunit_try_catch_test_init(struct kunit *test)
+{
+       struct kunit_try_catch_test_context *ctx;
+
+       ctx = kunit_kzalloc(test, sizeof(*ctx), GFP_KERNEL);
+       if (!ctx)
+               return -ENOMEM;
+
+       test->priv = ctx;
+
+       ctx->try_catch = kunit_kmalloc(test,
+                                      sizeof(*ctx->try_catch),
+                                      GFP_KERNEL);
+       if (!ctx->try_catch)
+               return -ENOMEM;
+
+       return 0;
+}
+
+static struct kunit_case kunit_try_catch_test_cases[] = {
+       KUNIT_CASE(kunit_test_try_catch_successful_try_no_catch),
+       KUNIT_CASE(kunit_test_try_catch_unsuccessful_try_does_catch),
+       {}
+};
+
+static struct kunit_suite kunit_try_catch_test_suite = {
+       .name = "kunit-try-catch-test",
+       .init = kunit_try_catch_test_init,
+       .test_cases = kunit_try_catch_test_cases,
+};
+kunit_test_suite(kunit_try_catch_test_suite);
-- 
2.23.0.rc1.153.gdeed80330f-goog

_______________________________________________
Linux-nvdimm mailing list
Linux-nvdimm@lists.01.org
https://lists.01.org/mailman/listinfo/linux-nvdimm

Reply via email to