add unit tests

Author: LinuxSuRen

README.md

@@ -16,6 +16,7 @@ This is a API testing tool.
 *   Output reference between TestCase
 *   Run in server mode, and provide the [gRPC endpoint](pkg/server/server.proto)
 *   [VS Code extension](https://github.com/LinuxSuRen/vscode-api-testing) support
+*   [HTTP API record](extensions/collector)
 
 ## Get started
 

extensions/collector/README.md

@@ -0,0 +1,12 @@
+HTTP API record tool.
+
+## Usage
+
+```shell
+atest-collector --filter-path /answer/api/v1
+```
+
+It will start a HTTP proxy server, and set the server address to your browser proxy (such as: [SwitchyOmega](https://github.com/FelisCatus/SwitchyOmega)).
+
+`atest-collector` will record all HTTP requests which has prefix `/answer/api/v1`, and 
+save it to file `sample.yaml` once you close the server.

extensions/collector/cmd/collect.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 
 	"github.com/elazarl/goproxy"
@@ -37,19 +38,32 @@ func NewRootCmd() (c *cobra.Command) {
 	return
 }
 
+type responseFilter struct {
+	urlFilter *filter.URLPathFilter
+	collects  *pkg.Collects
+}
+
+func (f *responseFilter) filter(resp *http.Response, ctx *goproxy.ProxyCtx) *http.Response {
+	contentType := resp.Header.Get("Content-Type")
+	if !strings.Contains(contentType, "application/json") {
+		return resp
+	}
+
+	req := resp.Request
+	if f.urlFilter.Filter(req.URL) {
+		f.collects.Add(req.Clone(context.TODO()))
+	}
+	return resp
+}
+
 func (o *option) runE(cmd *cobra.Command, args []string) (err error) {
 	urlFilter := &filter.URLPathFilter{PathPrefix: o.filterPath}
 	collects := pkg.NewCollects()
+	responseFilter := &responseFilter{urlFilter: urlFilter, collects: collects}
 
 	proxy := goproxy.NewProxyHttpServer()
 	proxy.Verbose = true
-	proxy.OnRequest().DoFunc(
-		func(r *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
-			if urlFilter.Filter(r.URL) {
-				collects.Add(r.Clone(context.TODO()))
-			}
-			return r, nil
-		})
+	proxy.OnResponse().DoFunc(responseFilter.filter)
 
 	exporter := pkg.NewSampleExporter()
 	collects.AddEvent(exporter.Add)

extensions/collector/cmd/collect_test.go

@@ -1,12 +1,36 @@
 package cmd
 
 import (
-	"github.com/stretchr/testify/assert"
+	"net/http"
+	"net/url"
 	"testing"
+
+	"github.com/linuxsuren/api-testing/extensions/collector/pkg"
+	"github.com/linuxsuren/api-testing/extensions/collector/pkg/filter"
+	"github.com/stretchr/testify/assert"
 )
 
 func TestNewRootCmd(t *testing.T) {
 	c := NewRootCmd()
 	assert.NotNil(t, c)
 	assert.Equal(t, "atest-collector", c.Use)
 }
+
+func TestResponseFilter(t *testing.T) {
+	resp := &http.Response{
+		Header: http.Header{
+			"Content-Type": []string{"application/json; charset=utf-8"},
+		},
+		Request: &http.Request{
+			URL: &url.URL{},
+		},
+	}
+	emptyResp := &http.Response{}
+
+	filter := &responseFilter{
+		urlFilter: &filter.URLPathFilter{},
+		collects:  pkg.NewCollects(),
+	}
+	filter.filter(emptyResp, nil)
+	filter.filter(resp, nil)
+}