udpate test

Author: Ryan19929

fe/fe-core/src/main/java/org/apache/doris/catalog/DiskInfo.java

@@ -18,6 +18,7 @@
 package org.apache.doris.catalog;
 
 import org.apache.doris.common.Config;
+import org.apache.doris.common.util.DebugPointUtil;
 import org.apache.doris.thrift.TStorageMedium;
 
 import com.google.gson.annotations.SerializedName;
@@ -164,8 +165,47 @@ public void setStorageMedium(TStorageMedium storageMedium) {
      * Check if this disk's capacity reach the limit. Return true if yes.
      * if floodStage is true, use floodStage threshold to check.
      *      floodStage threshold means a loosely limit, and we use 'AND' to give a more loosely limit.
+     *
+     * Debug points:
+     * - DiskInfo.exceedLimit.ssd.alwaysTrue: Force SSD disks to return true (exceed limit)
+     * - DiskInfo.exceedLimit.hdd.alwaysTrue: Force HDD disks to return true (exceed limit)
+     * - DiskInfo.exceedLimit.ssd.alwaysFalse: Force SSD disks to return false (available)
+     * - DiskInfo.exceedLimit.hdd.alwaysFalse: Force HDD disks to return false (available)
      */
     public boolean exceedLimit(boolean floodStage) {
+        // Debug point: Force specific medium to report as exceed limit
+        if (storageMedium == TStorageMedium.SSD) {
+            if (DebugPointUtil.isEnable("DiskInfo.exceedLimit.ssd.alwaysTrue")) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("Debug point active: DiskInfo.exceedLimit.ssd.alwaysTrue, "
+                            + "forcing SSD disk {} to report exceed limit", rootPath);
+                }
+                return true;
+            }
+            if (DebugPointUtil.isEnable("DiskInfo.exceedLimit.ssd.alwaysFalse")) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("Debug point active: DiskInfo.exceedLimit.ssd.alwaysFalse, "
+                            + "forcing SSD disk {} to report available", rootPath);
+                }
+                return false;
+            }
+        } else if (storageMedium == TStorageMedium.HDD) {
+            if (DebugPointUtil.isEnable("DiskInfo.exceedLimit.hdd.alwaysTrue")) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("Debug point active: DiskInfo.exceedLimit.hdd.alwaysTrue, "
+                            + "forcing HDD disk {} to report exceed limit", rootPath);
+                }
+                return true;
+            }
+            if (DebugPointUtil.isEnable("DiskInfo.exceedLimit.hdd.alwaysFalse")) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("Debug point active: DiskInfo.exceedLimit.hdd.alwaysFalse, "
+                            + "forcing HDD disk {} to report available", rootPath);
+                }
+                return false;
+            }
+        }
+
         if (LOG.isDebugEnabled()) {
             LOG.debug("flood stage: {}, diskAvailableCapacityB: {}, totalCapacityB: {}",
                     floodStage, diskAvailableCapacityB, totalCapacityB);

regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Syncer.groovy

@@ -631,6 +631,10 @@ class Syncer {
     }
 
     Boolean restoreSnapshot(boolean forCCR = false) {
+        return restoreSnapshot(forCCR, null, null)
+    }
+    
+    Boolean restoreSnapshot(boolean forCCR, String storageMedium, String mediumAllocationMode) {
         logger.info("Restore snapshot ${context.labelName}")
         FrontendClientImpl clientImpl = context.getSourceFrontClient()
 
@@ -648,6 +652,10 @@ class Syncer {
         logger.info("json map ${jsonMap}.")
         context.getSnapshotResult.setJobInfo(gson.toJson(jsonMap).getBytes())
 
+        // Set storage_medium and medium_allocation_mode for testing
+        context.storageMedium = storageMedium
+        context.mediumAllocationMode = mediumAllocationMode
+
         // step 2: restore
         TRestoreSnapshotResult result = SyncerUtils.restoreSnapshot(clientImpl, context, forCCR)
         return checkRestoreSnapshot(result)

regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/SyncerContext.groovy

@@ -133,6 +133,8 @@ class SyncerContext {
     public String tableName
     public TGetSnapshotResult getSnapshotResult
     public String token
+    public String storageMedium
+    public String mediumAllocationMode
 
     public Config config
     public String user

regression-test/framework/src/main/groovy/org/apache/doris/regression/util/SyncerUtils.groovy

@@ -131,6 +131,16 @@ class SyncerUtils {
         request.setProperties(properties)
         request.setMeta(context.getSnapshotResult.getMeta())
         request.setJobInfo(context.getSnapshotResult.getJobInfo())
+        
+        // Set storage_medium and medium_allocation_mode if specified
+        // These will test the new code in FrontendServiceImpl.restoreSnapshot
+        if (context.storageMedium != null) {
+            request.setStorageMedium(context.storageMedium)
+        }
+        if (context.mediumAllocationMode != null) {
+            request.setMediumAllocationMode(context.mediumAllocationMode)
+        }
+        
         return clientImpl.client.restoreSnapshot(request)
     }