refactor: move LoadImage

uixt/ai/planner_test.go

@@ -1,18 +1,12 @@
 package ai
 
 import (
-	"bytes"
 	"context"
-	"encoding/base64"
-	"fmt"
-	"image"
-	"image/jpeg"
-	"image/png"
-	"os"
 	"testing"
 
 	"github.com/cloudwego/eino/schema"
 	"github.com/httprunner/httprunner/v5/code"
+	"github.com/httprunner/httprunner/v5/internal/builtin"
 	"github.com/httprunner/httprunner/v5/uixt/option"
 	"github.com/httprunner/httprunner/v5/uixt/types"
 	"github.com/stretchr/testify/assert"
@@ -20,7 +14,7 @@ import (
 )
 
 func TestVLMPlanning(t *testing.T) {
-	imageBase64, size, err := loadImage("testdata/llk_1.png")
+	imageBase64, size, err := builtin.LoadImage("testdata/llk_1.png")
 	require.NoError(t, err)
 
 	userInstruction := `连连看是一款经典的益智消除类小游戏，通常以图案或图标为主要元素。以下是连连看的基本规则说明：
@@ -104,7 +98,7 @@ func TestVLMPlanning(t *testing.T) {
 }
 
 func TestXHSPlanning(t *testing.T) {
-	imageBase64, size, err := loadImage("testdata/xhs-feed.jpeg")
+	imageBase64, size, err := builtin.LoadImage("testdata/xhs-feed.jpeg")
 	require.NoError(t, err)
 
 	userInstruction := "点击第二个帖子的作者头像"
@@ -177,7 +171,7 @@ func TestXHSPlanning(t *testing.T) {
 }
 
 func TestChatList(t *testing.T) {
-	imageBase64, size, err := loadImage("testdata/chat_list.jpeg")
+	imageBase64, size, err := builtin.LoadImage("testdata/chat_list.jpeg")
 	require.NoError(t, err)
 
 	userInstruction := "请结合图片的文字信息，请告诉我一共有多少个群聊，哪些群聊右下角有绿点"
@@ -232,7 +226,7 @@ func TestHandleSwitch(t *testing.T) {
 	}
 
 	for _, tc := range testCases {
-		imageBase64, size, err := loadImage(tc.imageFile)
+		imageBase64, size, err := builtin.LoadImage(tc.imageFile)
 		require.NoError(t, err)
 
 		opts := &PlanningOptions{
@@ -263,7 +257,7 @@ func TestHandleSwitch(t *testing.T) {
 }
 
 func TestValidateInput(t *testing.T) {
-	imageBase64, size, err := loadImage("testdata/popup_risk_warning.png")
+	imageBase64, size, err := builtin.LoadImage("testdata/popup_risk_warning.png")
 	require.NoError(t, err)
 
 	tests := []struct {
@@ -388,87 +382,18 @@ func TestProcessVLMResponse(t *testing.T) {
 	}
 }
 
-func TestSavePositionImg(t *testing.T) {
-	imageBase64, _, err := loadImage("testdata/popup_risk_warning.png")
-	require.NoError(t, err)
-
-	params := struct {
-		InputImgBase64 string
-		Rect           struct {
-			X float64
-			Y float64
-		}
-		OutputPath string
-	}{
-		InputImgBase64: imageBase64,
-		Rect: struct {
-			X float64
-			Y float64
-		}{
-			X: 100,
-			Y: 100,
-		},
-		OutputPath: "testdata/output.png",
-	}
-
-	err = SavePositionImg(params)
-	require.NoError(t, err)
-
-	// cleanup
-	defer os.Remove(params.OutputPath)
-}
-
 func TestLoadImage(t *testing.T) {
 	// Test PNG image
-	pngBase64, pngSize, err := loadImage("testdata/llk_1.png")
+	pngBase64, pngSize, err := builtin.LoadImage("testdata/llk_1.png")
 	require.NoError(t, err)
 	assert.NotEmpty(t, pngBase64)
 	assert.Greater(t, pngSize.Width, 0)
 	assert.Greater(t, pngSize.Height, 0)
 
 	// Test JPEG image
-	jpegBase64, jpegSize, err := loadImage("testdata/xhs-feed.jpeg")
+	jpegBase64, jpegSize, err := builtin.LoadImage("testdata/xhs-feed.jpeg")
 	require.NoError(t, err)
 	assert.NotEmpty(t, jpegBase64)
 	assert.Greater(t, jpegSize.Width, 0)
 	assert.Greater(t, jpegSize.Height, 0)
 }
-
-// loadImage loads image and returns base64 encoded string
-func loadImage(imagePath string) (base64Str string, size types.Size, err error) {
-	// Read the image file
-	imageFile, err := os.OpenFile(imagePath, os.O_RDONLY, 0o600)
-	if err != nil {
-		return "", types.Size{}, fmt.Errorf("failed to open image file: %w", err)
-	}
-	defer imageFile.Close()
-
-	// Decode the image to get its resolution
-	imageData, format, err := image.Decode(imageFile)
-	if err != nil {
-		return "", types.Size{}, fmt.Errorf("failed to decode image: %w", err)
-	}
-
-	// Get the resolution of the image
-	width := imageData.Bounds().Dx()
-	height := imageData.Bounds().Dy()
-	size = types.Size{Width: width, Height: height}
-
-	// Convert image to base64
-	buf := new(bytes.Buffer)
-	// 根据图像格式选择正确的编码器
-	if format == "jpeg" || format == "jpg" {
-		if err := jpeg.Encode(buf, imageData, nil); err != nil {
-			return "", types.Size{}, fmt.Errorf("failed to encode image to buffer: %w", err)
-		}
-	} else {
-		// 默认使用 PNG 编码
-		if err := png.Encode(buf, imageData); err != nil {
-			return "", types.Size{}, fmt.Errorf("failed to encode image to buffer: %w", err)
-		}
-	}
-	base64Str = fmt.Sprintf("data:image/%s;base64,%s", format,
-		base64.StdEncoding.EncodeToString(buf.Bytes()))
-
-	return base64Str, size, nil
-}