feat: support to download atest self (#180)

Author: LinuxSuRen

cmd/server.go

@@ -125,6 +125,7 @@ func (o *serverOption) runE(cmd *cobra.Command, args []string) (err error) {
 		mux.HandlePath(http.MethodGet, "/", frontEndHandlerWithLocation(o.consolePath))
 		mux.HandlePath(http.MethodGet, "/assets/{asset}", frontEndHandlerWithLocation(o.consolePath))
 		mux.HandlePath(http.MethodGet, "/healthz", frontEndHandlerWithLocation(o.consolePath))
+		mux.HandlePath(http.MethodGet, "/get", o.getAtestBinary)
 		debugHandler(mux)
 		o.httpServer.WithHandler(mux)
 		log.Printf("HTTP server listening at %v", httplis.Addr())
@@ -186,6 +187,25 @@ func debugHandler(mux *runtime.ServeMux) {
 	})
 }
 
+func (o *serverOption) getAtestBinary(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
+	w.Header().Set("Content-Disposition", "attachment; filename=atest")
+	w.Header().Set("Content-Type", "application/octet-stream")
+	w.Header().Set("Content-Transfer-Encoding", "binary")
+	w.Header().Set("Expires", "0")
+	w.Header().Set("Pragma", "no-cache")
+	w.Header().Set("Cache-Control", "no-cache")
+
+	var data []byte
+	if atestPath, err := o.execer.LookPath("atest"); err == nil {
+		if data, err = os.ReadFile(atestPath); err != nil {
+			data = []byte(fmt.Sprintf("failed to read atest: %v", err))
+		}
+	} else {
+		data = []byte("not found atest")
+	}
+	w.Write(data)
+}
+
 type gRPCServer interface {
 	Serve(lis net.Listener) error
 	grpc.ServiceRegistrar

cmd/server_test.go

@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
 	"net"
 	"net/http"
@@ -137,6 +138,39 @@ func TestFrontEndHandlerWithLocation(t *testing.T) {
 			assert.Equal(t, http.StatusOK, resp.StatusCode)
 		}
 	})
+
+	t.Run("download atest", func(t *testing.T) {
+		opt := &serverOption{
+			execer: fakeruntime.FakeExecer{
+				ExpectOS:            "linux",
+				ExpectLookPathError: errors.New("fake"),
+			},
+		}
+
+		req, err := http.NewRequest(http.MethodGet, "/get", nil)
+		assert.NoError(t, err)
+
+		resp := newFakeResponseWriter()
+
+		opt.getAtestBinary(resp, req, map[string]string{})
+		assert.Equal(t, "not found atest", resp.GetBody().String())
+	})
+
+	t.Run("download atest, failed to read", func(t *testing.T) {
+		opt := &serverOption{
+			execer: fakeruntime.FakeExecer{
+				ExpectOS: "linux",
+			},
+		}
+
+		req, err := http.NewRequest(http.MethodGet, "/get", nil)
+		assert.NoError(t, err)
+
+		resp := newFakeResponseWriter()
+
+		opt.getAtestBinary(resp, req, map[string]string{})
+		assert.Equal(t, "failed to read atest: open : no such file or directory", resp.GetBody().String())
+	})
 }
 
 type fakeResponseWriter struct {

docs/README.md

@@ -74,6 +74,28 @@ jmeter -n -t bin/gitee.jmx
 
 Please feel free to bring more test tool converters.
 
+## Run in Jenkins
+You can run the API testings in Jenkins, see also the following example:
+
+```Jenkinsfile
+pipeline {
+    agent any
+    
+    stages() {
+        stage('test') {
+            steps {
+                sh '''
+                curl http://localhost:9090/get -o atest
+                chmod u+x atest
+                
+                ./atest run -p http://localhost:9090/server.Runner/ConvertTestSuite?suite=api-testing
+                '''
+            }
+        }
+    }
+}
+```
+
 ## Storage
 There are multiple storage backends supported. See the status from the list: