mirror of
https://github.com/openwrt/openwrt.git
synced 2024-12-30 18:47:06 +00:00
76 lines
2.3 KiB
Diff
76 lines
2.3 KiB
Diff
|
From 0f354629c10f457751c20a230b13b1b7ac2f1fbc Mon Sep 17 00:00:00 2001
|
||
|
From: David Gow <davidgow@google.com>
|
||
|
Date: Fri, 25 Nov 2022 16:43:04 +0800
|
||
|
Subject: [PATCH] kunit: Provide a static key to check if KUnit is
|
||
|
actively running tests
|
||
|
|
||
|
KUnit does a few expensive things when enabled. This hasn't been a
|
||
|
problem because KUnit was only enabled on test kernels, but with a few
|
||
|
people enabling (but not _using_) KUnit on production systems, we need a
|
||
|
runtime way of handling this.
|
||
|
|
||
|
Provide a 'kunit_running' static key (defaulting to false), which allows
|
||
|
us to hide any KUnit code behind a static branch. This should reduce the
|
||
|
performance impact (on other code) of having KUnit enabled to a single
|
||
|
NOP when no tests are running.
|
||
|
|
||
|
Note that, while it looks unintuitive, tests always run entirely within
|
||
|
__kunit_test_suites_init(), so it's safe to decrement the static key at
|
||
|
the end of this function, rather than in __kunit_test_suites_exit(),
|
||
|
which is only there to clean up results in debugfs.
|
||
|
|
||
|
Signed-off-by: David Gow <davidgow@google.com>
|
||
|
Reviewed-by: Daniel Latypov <dlatypov@google.com>
|
||
|
Signed-off-by: Shuah Khan <skhan@linuxfoundation.org>
|
||
|
---
|
||
|
include/kunit/test.h | 4 ++++
|
||
|
lib/kunit/test.c | 6 ++++++
|
||
|
2 files changed, 10 insertions(+)
|
||
|
|
||
|
--- a/include/kunit/test.h
|
||
|
+++ b/include/kunit/test.h
|
||
|
@@ -16,6 +16,7 @@
|
||
|
#include <linux/container_of.h>
|
||
|
#include <linux/err.h>
|
||
|
#include <linux/init.h>
|
||
|
+#include <linux/jump_label.h>
|
||
|
#include <linux/kconfig.h>
|
||
|
#include <linux/kref.h>
|
||
|
#include <linux/list.h>
|
||
|
@@ -27,6 +28,9 @@
|
||
|
|
||
|
#include <asm/rwonce.h>
|
||
|
|
||
|
+/* Static key: true if any KUnit tests are currently running */
|
||
|
+DECLARE_STATIC_KEY_FALSE(kunit_running);
|
||
|
+
|
||
|
struct kunit;
|
||
|
|
||
|
/* Size of log associated with test. */
|
||
|
--- a/lib/kunit/test.c
|
||
|
+++ b/lib/kunit/test.c
|
||
|
@@ -20,6 +20,8 @@
|
||
|
#include "string-stream.h"
|
||
|
#include "try-catch-impl.h"
|
||
|
|
||
|
+DEFINE_STATIC_KEY_FALSE(kunit_running);
|
||
|
+
|
||
|
#if IS_BUILTIN(CONFIG_KUNIT)
|
||
|
/*
|
||
|
* Fail the current test and print an error message to the log.
|
||
|
@@ -622,10 +624,14 @@ int __kunit_test_suites_init(struct kuni
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
+ static_branch_inc(&kunit_running);
|
||
|
+
|
||
|
for (i = 0; i < num_suites; i++) {
|
||
|
kunit_init_suite(suites[i]);
|
||
|
kunit_run_tests(suites[i]);
|
||
|
}
|
||
|
+
|
||
|
+ static_branch_dec(&kunit_running);
|
||
|
return 0;
|
||
|
}
|
||
|
EXPORT_SYMBOL_GPL(__kunit_test_suites_init);
|