From 59ca322ddaa8e73022da7b0991a10e0a63e82645 Mon Sep 17 00:00:00 2001
From: "lilong.129" <lilong.129@bytedance.com>
Date: Mon, 30 Jun 2025 19:50:21 +0800
Subject: [PATCH] change: update example yanglegeyang

---
 .../game/yanglegeyang/game_yanglegeyang.json  | 21 ++++++++++++++++---
 examples/game/yanglegeyang/main_test.go       | 12 ++++++++---
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/examples/game/yanglegeyang/game_yanglegeyang.json b/examples/game/yanglegeyang/game_yanglegeyang.json
index 9c72a02a..8ce49876 100644
--- a/examples/game/yanglegeyang/game_yanglegeyang.json
+++ b/examples/game/yanglegeyang/game_yanglegeyang.json
@@ -1,6 +1,9 @@
 {
     "config": {
         "name": "羊了个羊小游戏自动化测试",
+        "variables": {
+            "package_name": "com.ss.android.ugc.aweme"
+        },
         "ai_options": {
             "llm_service": "doubao-1.5-thinking-vision-pro-250428"
         }
@@ -13,7 +16,7 @@
                 "actions": [
                     {
                         "method": "app_launch",
-                        "params": "com.ss.android.ugc.aweme"
+                        "params": "$package_name"
                     },
                     {
                         "method": "sleep",
@@ -25,8 +28,8 @@
                 {
                     "check": "ui_foreground_app",
                     "assert": "equal",
-                    "expect": "com.ss.android.ugc.aweme",
-                    "msg": "app [com.ss.android.ugc.aweme] should be in foreground"
+                    "expect": "$package_name",
+                    "msg": "app [$package_name] should be in foreground"
                 }
             ]
         },
@@ -68,6 +71,18 @@
                     }
                 ]
             }
+        },
+        {
+            "name": "退出抖音 app",
+            "android": {
+                "os_type": "android",
+                "actions": [
+                    {
+                        "method": "app_terminate",
+                        "params": "$package_name"
+                    }
+                ]
+            }
         }
     ]
 }
diff --git a/examples/game/yanglegeyang/main_test.go b/examples/game/yanglegeyang/main_test.go
index a6372347..691c749c 100644
--- a/examples/game/yanglegeyang/main_test.go
+++ b/examples/game/yanglegeyang/main_test.go
@@ -28,14 +28,17 @@ func TestGameYanglegeyang(t *testing.T) {
 
 	testCase := &hrp.TestCase{
 		Config: hrp.NewConfig("羊了个羊小游戏自动化测试").
-			SetLLMService(option.DOUBAO_1_5_THINKING_VISION_PRO_250428),
+			SetLLMService(option.DOUBAO_1_5_THINKING_VISION_PRO_250428).
+			WithVariables(map[string]interface{}{
+				"package_name": "com.ss.android.ugc.aweme",
+			}),
 		TestSteps: []hrp.IStep{
 			hrp.NewStep("启动抖音 app").
 				Android().
-				AppLaunch("com.ss.android.ugc.aweme").
+				AppLaunch("$package_name").
 				Sleep(5).
 				Validate().
-				AssertAppInForeground("com.ss.android.ugc.aweme"),
+				AssertAppInForeground("$package_name"),
 			hrp.NewStep("进入「羊了个羊」小游戏").
 				Android().
 				StartToGoal("搜索「羊了个羊星球」，进入小程序，加入羊群进入游戏",
@@ -47,6 +50,9 @@ func TestGameYanglegeyang(t *testing.T) {
 				StartToGoal(userInstruction,
 					option.WithPreMarkOperation(true),
 					option.WithTimeout(300)), // 5 minutes
+			hrp.NewStep("退出抖音 app").
+				Android().
+				AppTerminate("$package_name"),
 		},
 	}
 	err := testCase.Dump2JSON("game_yanglegeyang.json")