From cc920696edca50989bd04f55276e46256f1f9a66 Mon Sep 17 00:00:00 2001
From: Pieter De Gendt <pieter.degendt@basalte.be>
Date: Thu, 21 Dec 2023 12:46:09 +0100
Subject: [PATCH] tests: unit: util: Add tests for IF_DISABLED

Add a testcase for the IF_DISABLED macro.

Signed-off-by: Pieter De Gendt <pieter.degendt@basalte.be>
---
 tests/unit/util/main.c   |  8 ++++++++
 tests/unit/util/test.inc | 22 ++++++++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/tests/unit/util/main.c b/tests/unit/util/main.c
index bce626f9fe4..3c93968db46 100644
--- a/tests/unit/util/main.c
+++ b/tests/unit/util/main.c
@@ -38,6 +38,10 @@ ZTEST(util_cxx, test_IF_ENABLED) {
 	run_IF_ENABLED();
 }
 
+ZTEST(util_cxx, test_IF_DISABLED) {
+	run_IF_DISABLED();
+}
+
 ZTEST(util_cxx, test_LISTIFY) {
 	run_LISTIFY();
 }
@@ -186,6 +190,10 @@ ZTEST(util_cc, test_IF_ENABLED) {
 	run_IF_ENABLED();
 }
 
+ZTEST(util_cc, test_IF_DISABLED) {
+	run_IF_DISABLED();
+}
+
 ZTEST(util_cc, test_LISTIFY) {
 	run_LISTIFY();
 }
diff --git a/tests/unit/util/test.inc b/tests/unit/util/test.inc
index c47cc7ab7e2..f16f1039d45 100644
--- a/tests/unit/util/test.inc
+++ b/tests/unit/util/test.inc
@@ -162,6 +162,28 @@ skipped:
 	#undef test_IF_ENABLED_FLAG_B
 }
 
+void run_IF_DISABLED(void)
+{
+	#define test_IF_DISABLED_FLAG_A 0
+	#define test_IF_DISABLED_FLAG_B 1
+
+	IF_DISABLED(test_IF_DISABLED_FLAG_A, (goto skipped_a;))
+	/* location should be skipped if IF_DISABLED macro is correct. */
+	zassert_false(true, "location A should be skipped");
+skipped_a:
+	IF_DISABLED(test_IF_DISABLED_FLAG_B, (zassert_false(true, "");))
+
+	IF_DISABLED(test_IF_DISABLED_FLAG_C, (goto skipped_c;))
+	/* location should be skipped if IF_DISABLED macro is correct. */
+	zassert_false(true, "location C should be skipped");
+skipped_c:
+
+	zassert_true(true, "");
+
+	#undef test_IF_DISABLED_FLAG_A
+	#undef test_IF_DISABLED_FLAG_B
+}
+
 void run_LISTIFY(void)
 {
 	int ab0 = 1;