From 21e412393bfbee88d1e04d5003a53548758bc2af Mon Sep 17 00:00:00 2001
From: kkop <45192870+cjj2010@users.noreply.github.com>
Date: Fri, 8 Mar 2024 19:30:43 +0800
Subject: [PATCH] [enhancement] add_method_for_schemachange (#31849)

---
 .../org/apache/doris/regression/suite/Suite.groovy    | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
index 109b1c2a76..6e32bbfd60 100644
--- a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
+++ b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
@@ -546,6 +546,17 @@ class Suite implements GroovyInterceptable {
         }
     }
 
+
+    void expectException(Closure userFunction, String tableName, String errorMessage = null) {
+        try {
+            userFunction()
+        } catch (Exception e) {
+            if (e.getMessage()!= errorMessage) {
+                throw e
+            }
+        }
+    }
+
     String getBrokerName() {
         String brokerName = context.config.otherConfigs.get("brokerName")
         return brokerName