diff --git a/src/main/java/com/iflytop/gd/app/controller/TestController.java b/src/main/java/com/iflytop/gd/app/controller/TestController.java
index f056839..ec201ee 100644
--- a/src/main/java/com/iflytop/gd/app/controller/TestController.java
+++ b/src/main/java/com/iflytop/gd/app/controller/TestController.java
@@ -113,4 +113,11 @@ public class TestController {
         testService.setTemperature(code, temperature, heatingType);
         return Result.success();
     }
+
+    @Operation(summary = "设置加热模块状态")
+    @PostMapping("/set-heat-module-state")
+    public Result<?> setHeatModuleState(HeatModuleCode code, Integer trayStatus) throws Exception {
+        testService.setHeatModuleState(code, trayStatus);
+        return Result.success();
+    }
 }
diff --git a/src/main/java/com/iflytop/gd/app/service/api/TestService.java b/src/main/java/com/iflytop/gd/app/service/api/TestService.java
index f967393..477f41b 100644
--- a/src/main/java/com/iflytop/gd/app/service/api/TestService.java
+++ b/src/main/java/com/iflytop/gd/app/service/api/TestService.java
@@ -52,4 +52,8 @@ public class TestService {
         deviceStateService.getDeviceState().getHeatModuleByCode(code).setTemperature(temperature);
     }
 
+    public void setHeatModuleState(HeatModuleCode code, int trayStatus) {
+        deviceStateService.getDeviceState().getHeatModuleByCode(code).setTrayStatus(trayStatus);
+    }
+
 }