]> www.infradead.org Git - users/jedix/linux-maple.git/commit
rust: kunit: allow to know if we are in a test
authorJosé Expósito <jose.exposito89@gmail.com>
Fri, 7 Mar 2025 09:00:58 +0000 (17:00 +0800)
committerMiguel Ojeda <ojeda@kernel.org>
Thu, 20 Mar 2025 11:27:05 +0000 (12:27 +0100)
commit100af58c8d5822750ef9ba65f5d5ea3367c669de
treec593c1a68c9c193f5ac575260d8eb00006e02849
parentc0010452893e07e032427e88f6b7b4bf7ac42e95
rust: kunit: allow to know if we are in a test

In some cases, we need to call test-only code from outside the test
case, for example, to mock a function or a module.

In order to check whether we are in a test or not, we need to test if
`CONFIG_KUNIT` is set.
Unfortunately, we cannot rely only on this condition because:
- a test could be running in another thread,
- some distros compile KUnit in production kernels, so checking at runtime
  that `current->kunit_test != NULL` is required.

Forturately, KUnit provides an optimised check in
`kunit_get_current_test()`, which checks CONFIG_KUNIT, a global static
key, and then the current thread's running KUnit test.

Add a safe wrapper function around this to know whether or not we are in
a KUnit test and examples showing how to mock a function and a module.

Signed-off-by: José Expósito <jose.exposito89@gmail.com>
Co-developed-by: Miguel Ojeda <ojeda@kernel.org>
Signed-off-by: Miguel Ojeda <ojeda@kernel.org>
Co-developed-by: David Gow <davidgow@google.com>
Signed-off-by: David Gow <davidgow@google.com>
Link: https://lore.kernel.org/r/20250307090103.918788-4-davidgow@google.com
Signed-off-by: Miguel Ojeda <ojeda@kernel.org>
rust/kernel/kunit.rs