tests: benchmarks: Add test for s2ram with uart automatic usage

Test UARTE driver power management without manual enable/disable

Signed-off-by: Bartosz Miller <[email protected]>
(cherry picked from commit d318939)

Author: nordic-bami

tests/benchmarks/multicore/idle_uarte/Kconfig

@@ -0,0 +1,5 @@
+source "Kconfig.zephyr"
+
+config TEST_AUTOMATIC_POWER_CONTROL
+	default n
+	bool "Enable testing without manual UART disable/enable"

tests/benchmarks/multicore/idle_uarte/prj.conf

@@ -12,6 +12,7 @@ CONFIG_POWEROFF=y
 
 CONFIG_BOOT_BANNER=n
 CONFIG_NRFS_MRAM_SERVICE_ENABLED=n
+CONFIG_ASSERT=y
 
 # Enable for debugging purposes only
 CONFIG_PRINTK=n

tests/benchmarks/multicore/idle_uarte/src/main.c

@@ -46,6 +46,8 @@ static void async_uart_callback(const struct device *dev, struct uart_event *evt
 		break;
 	case UART_TX_ABORTED:
 		printk("UART_TX_ABORTED\n");
+		printk("Callback should not enter here\n");
+		__ASSERT_NO_MSG(1 == 0);
 		break;
 	case UART_RX_RDY:
 		printk("UART_RX_RDY\n");
@@ -74,6 +76,33 @@ static void async_uart_callback(const struct device *dev, struct uart_event *evt
 	}
 }
 
+/* Helper function for enabling UART RX */
+void enable_uart_rx(void)
+{
+	int err;
+
+	printk("Enable UART RX\n");
+	err = uart_rx_enable(uart_dev, test_buffer, TEST_BUFFER_LEN,
+			     5 * UART_ACTION_BASE_TIMEOUT_US);
+	if (err != 0) {
+		printk("Unexpected error when enabling UART RX: %d\n", err);
+	}
+}
+
+/* Helper function for disabling UART RX */
+void disable_uart_rx(void)
+{
+	int err;
+
+	k_msleep(250);
+	printk("Disable UART RX\n");
+	err = uart_rx_disable(uart_dev);
+	if (err != 0) {
+		printk("Unexpected error when disabling RX: %d\n", err);
+	}
+	k_msleep(250);
+}
+
 int main(void)
 {
 	int err;
@@ -112,23 +141,27 @@ int main(void)
 		pm_device_runtime_enable(console_dev);
 	}
 
+#if defined(CONFIG_TEST_AUTOMATIC_POWER_CONTROL)
+	enable_uart_rx();
+#endif
+
 	while (1) {
 		printk("Hello\n");
 
-		err = uart_rx_enable(uart_dev, test_buffer, TEST_BUFFER_LEN,
-				     5 * UART_ACTION_BASE_TIMEOUT_US);
-		if (err != 0) {
-			printk("Unexpected error when enabling UART RX: %d\n", err);
-			return -1;
-		}
+#if !defined(CONFIG_TEST_AUTOMATIC_POWER_CONTROL)
+		enable_uart_rx();
+#endif
+
+		printk("UART test transmission\n");
 		err = uart_tx(uart_dev, test_pattern, TEST_BUFFER_LEN, UART_ACTION_BASE_TIMEOUT_US);
 		if (err != 0) {
 			printk("Unexpected error when sending UART TX data: %d\n", err);
 			return -1;
 		}
-		k_msleep(250);
-		uart_rx_disable(uart_dev);
-		k_msleep(250);
+
+#if !defined(CONFIG_TEST_AUTOMATIC_POWER_CONTROL)
+		disable_uart_rx();
+#endif
 
 		printk("Good night\n");
 		k_msleep(2000);

tests/benchmarks/multicore/idle_uarte/testcase.yaml

@@ -31,3 +31,18 @@ tests:
       fixture: gpio_loopback
       pytest_root:
         - "${CUSTOM_ROOT_TEST_DIR}/test_measure_power_consumption.py::test_measure_and_data_dump_power_consumption_uarte"
+
+  benchmarks.multicore.idle_uarte.automatic_pm.nrf54h20dk_cpuapp_cpurad.s2ram:
+    harness: pytest
+    platform_allow:
+      - nrf54h20dk/nrf54h20/cpuapp
+    integration_platforms:
+      - nrf54h20dk/nrf54h20/cpuapp
+    extra_args:
+      - SB_CONF_FILE=sysbuild/nrf54h20dk_nrf54h20_cpurad.conf
+      - DTC_OVERLAY_FILE="boards/nrf54h20dk_nrf54h20_cpuapp_normal.overlay"
+      - CONFIG_TEST_AUTOMATIC_POWER_CONTROL=y
+    harness_config:
+      fixture: gpio_loopback
+      pytest_root:
+        - "${CUSTOM_ROOT_TEST_DIR}/test_measure_power_consumption.py::test_measure_and_data_dump_power_consumption_uarte"