support to download bianry files on ui

Author: LinuxSuRen

console/atest-ui/src/views/TestCase.vue

@@ -1335,10 +1335,10 @@ const renameTestCase = (name: string) => {
             <Codemirror v-if="!isResponseFile" v-model="testResult.bodyText"/>
             <div v-if="isResponseFile" style="padding-top: 10px;">
             <el-row>
-              <el-col :span="8">
+              <el-col :span="10">
                 <div>Response body is too large, please download to view.</div>
               </el-col>
-              <el-col :span="4">
+              <el-col :span="2">
                 <el-button type="primary" @click="downloadResponseFile">Download</el-button>
               </el-col>
             </el-row>

pkg/runner/http.go

@@ -22,6 +22,8 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"os"
+	"path/filepath"
 	"strconv"
 	"strings"
 	"time"
@@ -225,13 +227,49 @@ func (r *simpleTestCaseRunner) RunTestCase(testcase *testing.TestCase, dataConte
 
 		err = errors.Join(err, jsonSchemaValidation(testcase.Expect.Schema, responseBodyData))
 	} else {
+		switch respType {
+		case util.OctetStream, util.Image:
+			var data []byte
+			if data, err = io.ReadAll(resp.Body); err == nil {
+				r.simpleResponse.RawBody = data
+				r.simpleResponse, err = HandleLargeResponseBody(r.simpleResponse, testcase.Group, testcase.Name)
+			}
+		}
 		r.log.Debug("skip to read the body due to it is not struct content: %q\n", respType)
 	}
 
 	r.cookies = append(r.cookies, resp.Cookies()...)
 	return
 }
 
+func HandleLargeResponseBody(resp SimpleResponse, suite string, caseName string) (SimpleResponse, error) {
+	const maxSize = 5120
+	prefix := "isFilePath-" + strings.Join([]string{suite, caseName}, "-")
+	if len(resp.Body) > 0 {
+		resp.RawBody = []byte(resp.Body)
+	}
+
+	if len(resp.RawBody) > maxSize {
+		fmt.Println("response body is too large, will be saved to file", "size", len(resp.RawBody))
+		tmpFile, err := os.CreateTemp("", prefix+"-")
+		defer tmpFile.Close()
+		if err != nil {
+			return resp, fmt.Errorf("failed to create file: %w", err)
+		}
+
+		if _, err = tmpFile.Write(resp.RawBody); err != nil {
+			return resp, fmt.Errorf("failed to write response body to file: %w", err)
+		}
+		absFilePath, err := filepath.Abs(tmpFile.Name())
+		if err != nil {
+			return resp, fmt.Errorf("failed to get absolute file path: %w", err)
+		}
+		resp.Body = filepath.Base(absFilePath)
+		return resp, nil
+	}
+	return resp, nil
+}
+
 func ammendHeaders(headers http.Header, body []byte) {
 	// add content-length if it's missing
 	if val := headers.Get(util.ContentLength); val == "" {

pkg/runner/runner.go

@@ -47,6 +47,7 @@ type ResponseRecord interface {
 type SimpleResponse struct {
 	Header     map[string]string
 	Body       string
+	RawBody    []byte
 	StatusCode int
 }
 

pkg/server/remote_server.go

@@ -262,7 +262,7 @@ func (s *server) Run(ctx context.Context, task *TestTask) (reply *TestResult, er
 		output, testErr := suiteRunner.RunTestCase(&testCase, dataContext, ctx)
 		if getter, ok := suiteRunner.(runner.ResponseRecord); ok {
 			resp := getter.GetResponseRecord()
-			resp, err = handleLargeResponseBody(resp, suite.Name, testCase.Name)
+			resp, err = runner.HandleLargeResponseBody(resp, suite.Name, testCase.Name)
 			reply.TestCaseResult = append(reply.TestCaseResult, &TestCaseResult{
 				StatusCode: int32(resp.StatusCode),
 				Body:       resp.Body,
@@ -343,32 +343,6 @@ func (s *server) BatchRun(srv Runner_BatchRunServer) (err error) {
 			}
 		}
 	}
-	return
-}
-
-func handleLargeResponseBody(resp runner.SimpleResponse, suite string, caseName string) (reply runner.SimpleResponse, err error) {
-	const maxSize = 5120
-	prefix := "isFilePath-" + strings.Join([]string{suite, caseName}, "-")
-
-	if len(resp.Body) > maxSize {
-		remoteServerLogger.Logger.Info("response body is too large, will be saved to file", "size", len(resp.Body))
-		tmpFile, err := os.CreateTemp("", prefix+"-")
-		defer tmpFile.Close()
-		if err != nil {
-			return resp, fmt.Errorf("failed to create file: %w", err)
-		}
-
-		if _, err = tmpFile.Write([]byte(resp.Body)); err != nil {
-			return resp, fmt.Errorf("failed to write response body to file: %w", err)
-		}
-		absFilePath, err := filepath.Abs(tmpFile.Name())
-		if err != nil {
-			return resp, fmt.Errorf("failed to get absolute file path: %w", err)
-		}
-		resp.Body = filepath.Base(absFilePath)
-		return resp, nil
-	}
-	return resp, nil
 }
 
 func (s *server) DownloadResponseFile(ctx context.Context, in *TestCase) (reply *FileData, err error) {

pkg/util/default.go

@@ -78,6 +78,7 @@ const (
 	YAML               = "application/yaml"
 	ZIP                = "application/zip"
 	OctetStream        = "application/octet-stream"
+	Image              = "image/jpeg"
 	Plain              = "text/plain"
 	Authorization      = "Authorization"
 )