From 23e7b4cf567585c86ed1d1d14a2ce663d618f03f Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Tue, 11 Oct 2022 21:38:05 +0800
Subject: [PATCH] fix: release UI driver session

---
 hrp/pkg/uixt/demo/main_test.go | 5 +++++
 hrp/runner.go                  | 6 ++++++
 2 files changed, 11 insertions(+)

diff --git a/hrp/pkg/uixt/demo/main_test.go b/hrp/pkg/uixt/demo/main_test.go
index 32dd2d9d..e544eee6 100644
--- a/hrp/pkg/uixt/demo/main_test.go
+++ b/hrp/pkg/uixt/demo/main_test.go
@@ -23,6 +23,11 @@ func TestIOSDemo(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	// release session
+	defer func() {
+		driverExt.Driver.DeleteSession()
+	}()
+
 	// 持续监测手机屏幕，直到出现青少年模式弹窗后，点击「我知道了」
 	for {
 		_, err1 := driverExt.GetTextXY("青少年模式")
diff --git a/hrp/runner.go b/hrp/runner.go
index 9b2380cf..72a20ffe 100644
--- a/hrp/runner.go
+++ b/hrp/runner.go
@@ -211,6 +211,12 @@ func (r *HRPRunner) Run(testcases ...ITestCase) error {
 			log.Error().Err(err).Msg("[Run] init session runner failed")
 			return err
 		}
+		// release UI driver session
+		defer func() {
+			for _, client := range sessionRunner.hrpRunner.uiClients {
+				client.Driver.DeleteSession()
+			}
+		}()
 
 		for it := sessionRunner.parametersIterator; it.HasNext(); {
 			err1 := sessionRunner.Start(it.Next())