tests: pm: Test the SoC State Change Power Domain driver

The SoC State Change Power Domain driver issues TURN_ON/
TURN_OFF actions to all devices registered with it for
certain power states that can be specified via device tree.
This test exercises the functionality of this driver.

Signed-off-by: Mahesh Mahadevan <[email protected]>

Author: mmahadevan108

tests/subsys/pm/power_domain_soc_state_change/CMakeLists.txt

@@ -0,0 +1,8 @@
+# Copyright 2025 NXP
+# SPDX-License-Identifier: Apache-2.0
+
+cmake_minimum_required(VERSION 3.20.0)
+find_package(Zephyr REQUIRED HINTS $ENV{ZEPHYR_BASE})
+project(power_domain_soc_state_change)
+
+target_sources(app PRIVATE src/main.c)

tests/subsys/pm/power_domain_soc_state_change/Kconfig

@@ -0,0 +1,11 @@
+# Copyright (c) 2023 Nordic Semiconductor ASA
+# SPDX-License-Identifier: Apache-2.0
+
+menu "Zephyr"
+source "Kconfig.zephyr"
+endmenu
+
+config TEST_PROVIDE_PM_HOOKS
+	bool "Provide PM hooks for test purposes"
+	default y
+	select HAS_PM

tests/subsys/pm/power_domain_soc_state_change/app.overlay

@@ -0,0 +1,46 @@
+/*
+ * Copyright 2025 NXP
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/ {
+	power-states {
+		state0: state0 {
+			compatible = "zephyr,power-state";
+			power-state-name = "runtime-idle";
+			min-residency-us = <10000>;
+			exit-latency-us = <100>;
+		};
+
+		state1: state1 {
+			compatible = "zephyr,power-state";
+			power-state-name = "suspend-to-idle";
+			min-residency-us = <50000>;
+			exit-latency-us = <500>;
+		};
+
+		state2: state2 {
+			compatible = "zephyr,power-state";
+			power-state-name = "standby";
+			min-residency-us = <100000>;
+			exit-latency-us = <1000>;
+		};
+	};
+
+	test_soc_state_domain: test_soc_state_domain {
+		compatible = "power-domain-soc-state-change";
+		status = "okay";
+		onoff-power-states = <&state1 &state2>;
+		#power-domain-cells = <0>;
+	};
+
+	test_dev_soc_state_change: test_dev_soc_state_change {
+		compatible = "test-device-pm";
+		status = "okay";
+		power-domains = <&test_soc_state_domain>;
+	};
+};
+
+&cpu0 {
+	cpu-power-states = <&state0 &state1 &state2>;
+};

tests/subsys/pm/power_domain_soc_state_change/dts/bindings/test-device-pm.yaml

@@ -0,0 +1,10 @@
+# Copyright 2025 NXP
+# SPDX-License-Identifier: Apache-2.0
+
+include: [base.yaml, pm.yaml]
+
+description: |
+    This binding provides resources required to build and run the
+    tests/subsys/pm/power_domain_soc_state_change test in Zephyr.
+
+compatible: "test-device-pm"

tests/subsys/pm/power_domain_soc_state_change/prj.conf

@@ -0,0 +1,7 @@
+# Copyright 2025 NXP
+# SPDX-License-Identifier: Apache-2.0
+
+CONFIG_ZTEST=y
+CONFIG_PM=y
+CONFIG_PM_DEVICE=y
+CONFIG_POWER_DOMAIN=y

tests/subsys/pm/power_domain_soc_state_change/src/main.c

@@ -0,0 +1,93 @@
+/*
+ * Copyright 2025 NXP
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#include <zephyr/ztest.h>
+#include <zephyr/pm/device.h>
+
+#define TEST_DEV DT_NODELABEL(test_dev_soc_state_change)
+
+static int testing_domain_on_notitication;
+static int testing_domain_off_notitication;
+
+void pm_state_set(enum pm_state state, uint8_t substate_id)
+{
+	ARG_UNUSED(substate_id);
+
+	switch (testing_domain_off_notitication) {
+	case 1:
+		zassert_equal(state, PM_STATE_STANDBY, "Wrong system state %d", state);
+		break;
+	case 2:
+		zassert_true(
+			((state == PM_STATE_SUSPEND_TO_IDLE) || (state == PM_STATE_RUNTIME_IDLE)),
+			"Wrong system state %d", state);
+		break;
+	default:
+		break;
+	}
+
+	k_cpu_idle();
+}
+
+void pm_state_exit_post_ops(enum pm_state state, uint8_t substate_id)
+{
+	ARG_UNUSED(state);
+	ARG_UNUSED(substate_id);
+
+	irq_unlock(0);
+}
+
+static int dev_pm_action(const struct device *dev, enum pm_device_action pm_action)
+{
+	ARG_UNUSED(dev);
+
+	if (pm_action == PM_DEVICE_ACTION_TURN_ON) {
+		testing_domain_on_notitication++;
+	}
+
+	if (pm_action == PM_DEVICE_ACTION_TURN_OFF) {
+		testing_domain_off_notitication++;
+	}
+
+	return 0;
+}
+
+PM_DEVICE_DT_DEFINE(TEST_DEV, dev_pm_action);
+DEVICE_DT_DEFINE(TEST_DEV, NULL, PM_DEVICE_DT_GET(TEST_DEV), NULL, NULL, POST_KERNEL, 20, NULL);
+
+ZTEST(power_domain_soc_state_change_1cpu, test_power_domain_soc_state_change)
+{
+	const struct pm_state_info *cpu_states, *state;
+
+	pm_state_cpu_get_all(_current_cpu->id, &cpu_states);
+
+	state = &cpu_states[2];
+	/* Sleep to transition to STATE: STANDBY */
+	k_usleep(state->min_residency_us + state->exit_latency_us);
+
+	zassert_equal(testing_domain_on_notitication, 1);
+	zassert_equal(testing_domain_off_notitication, 1);
+
+	state = &cpu_states[1];
+	/* Sleep to transition to STATE: SUSPEND-TO-IDLE */
+	k_usleep(state->min_residency_us + state->exit_latency_us);
+
+	zassert_equal(testing_domain_on_notitication, 2);
+	zassert_equal(testing_domain_off_notitication, 2);
+
+	state = &cpu_states[0];
+	/* Sleep to transition to STATE: RUNTIME-IDLE */
+	k_usleep(state->min_residency_us + state->exit_latency_us);
+
+	/* The domain notification flags should remain the same as RUNTIME-IDLE
+	 * is not listed as an ON/OFF power state in device-tree.
+	 */
+	zassert_equal(testing_domain_on_notitication, 2);
+	zassert_equal(testing_domain_off_notitication, 2);
+}
+
+ZTEST_SUITE(power_domain_soc_state_change_1cpu, NULL, NULL, ztest_simple_1cpu_before,
+	    ztest_simple_1cpu_after, NULL);

tests/subsys/pm/power_domain_soc_state_change/testcase.yaml

@@ -0,0 +1,10 @@
+# Copyright 2025 NXP
+# SPDX-License-Identifier: Apache-2.0
+
+tests:
+  pm.power_domain_soc_state_change:
+    platform_allow:
+      - native_sim
+    integration_platforms:
+      - native_sim
+    tags: pm