From c97ab134ef568440e9cf98a0db7243e8e1b31ea5 Mon Sep 17 00:00:00 2001
From: wuzhenghui <wuzhenghui@espressif.com>
Date: Fri, 19 Apr 2024 13:13:44 +0800
Subject: [PATCH] ci(esp_pm): add pd_top auto lightsleep test case for esp_pm

---
 .../cpu_retention/port/esp32p4/sleep_cpu.c      |  4 ++--
 .../esp_pm/test_apps/.build-test-rules.yml      |  1 +
 .../esp_pm/test_apps/esp_pm/pytest_esp_pm.py    | 17 ++++++++++++++++-
 .../esp_pm/sdkconfig.ci.pm_pd_top_sleep         |  1 +
 4 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 components/esp_pm/test_apps/esp_pm/sdkconfig.ci.pm_pd_top_sleep

diff --git a/components/esp_hw_support/lowpower/cpu_retention/port/esp32p4/sleep_cpu.c b/components/esp_hw_support/lowpower/cpu_retention/port/esp32p4/sleep_cpu.c
index cefc1f70cb..097e6b9b50 100644
--- a/components/esp_hw_support/lowpower/cpu_retention/port/esp32p4/sleep_cpu.c
+++ b/components/esp_hw_support/lowpower/cpu_retention/port/esp32p4/sleep_cpu.c
@@ -410,7 +410,7 @@ static IRAM_ATTR esp_err_t do_cpu_retention(sleep_cpu_entry_cb_t goto_sleep,
         esp_sleep_execute_event_callbacks(SLEEP_EVENT_SW_CPU_TO_MEM_END, (void *)0);
 #if CONFIG_PM_CHECK_SLEEP_RETENTION_FRAME
         /* Minus 2 * sizeof(long) is for bypass `pmufunc` and `frame_crc` field */
-        update_retention_frame_crc((uint32_t*)frame, RV_SLEEP_CTX_FRMSZ - 2 * sizeof(long), (uint32_t *)(&frame->frame_crc));
+        update_retention_frame_crc((uint32_t*)frame, RV_SLEEP_CTX_SZ1 - 2 * sizeof(long), (uint32_t *)(&frame->frame_crc));
 #endif
         REG_WRITE(RTC_SLEEP_WAKE_STUB_ADDR_REG, (uint32_t)rv_core_critical_regs_restore);
 
@@ -425,7 +425,7 @@ static IRAM_ATTR esp_err_t do_cpu_retention(sleep_cpu_entry_cb_t goto_sleep,
     }
 #if CONFIG_PM_CHECK_SLEEP_RETENTION_FRAME
     else {
-        validate_retention_frame_crc((uint32_t*)frame, RV_SLEEP_CTX_FRMSZ - 2 * sizeof(long), (uint32_t *)(&frame->frame_crc));
+        validate_retention_frame_crc((uint32_t*)frame, RV_SLEEP_CTX_SZ1 - 2 * sizeof(long), (uint32_t *)(&frame->frame_crc));
     }
 #endif
 
diff --git a/components/esp_pm/test_apps/.build-test-rules.yml b/components/esp_pm/test_apps/.build-test-rules.yml
index 8aece557bd..67cf500b6c 100644
--- a/components/esp_pm/test_apps/.build-test-rules.yml
+++ b/components/esp_pm/test_apps/.build-test-rules.yml
@@ -4,6 +4,7 @@ components/esp_pm/test_apps:
   enable:
     - if: INCLUDE_DEFAULT == 1
   disable:
+    - if: CONFIG_NAME == "pm_pd_top_sleep" and IDF_TARGET not in ["esp32c6", "esp32h2", "esp32p4"]
     - if: IDF_TARGET == "esp32c5"
       temporary: true
       reason: not support yet # TODO: [ESP32C5] IDF-8643
diff --git a/components/esp_pm/test_apps/esp_pm/pytest_esp_pm.py b/components/esp_pm/test_apps/esp_pm/pytest_esp_pm.py
index 9d7ff9eaff..a2c2fc0f76 100644
--- a/components/esp_pm/test_apps/esp_pm/pytest_esp_pm.py
+++ b/components/esp_pm/test_apps/esp_pm/pytest_esp_pm.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2022-2023 Espressif Systems (Shanghai) CO LTD
+# SPDX-FileCopyrightText: 2024 Espressif Systems (Shanghai) CO LTD
 # SPDX-License-Identifier: Apache-2.0
 import pytest
 from pytest_embedded import Dut
@@ -42,3 +42,18 @@ def test_esp_attr_xip_psram_esp32s2(dut: Dut) -> None:
 )
 def test_esp_attr_xip_psram_esp32s3(dut: Dut) -> None:
     dut.run_all_single_board_cases()
+
+
+# power down CPU and TOP domain in auto-lightsleep
+@pytest.mark.esp32c6
+@pytest.mark.esp32h2
+@pytest.mark.esp32p4
+@pytest.mark.generic
+@pytest.mark.parametrize(
+    'config',
+    [
+        'pm_pd_top_sleep'
+    ],
+)
+def test_esp_pd_top_and_cpu_sleep(dut: Dut) -> None:
+    dut.run_all_single_board_cases()
diff --git a/components/esp_pm/test_apps/esp_pm/sdkconfig.ci.pm_pd_top_sleep b/components/esp_pm/test_apps/esp_pm/sdkconfig.ci.pm_pd_top_sleep
new file mode 100644
index 0000000000..5658858b85
--- /dev/null
+++ b/components/esp_pm/test_apps/esp_pm/sdkconfig.ci.pm_pd_top_sleep
@@ -0,0 +1 @@
+CONFIG_PM_POWER_DOWN_PERIPHERAL_IN_LIGHT_SLEEP=y