feat: support to generate sample YAML via grpc (#76)

Author: LinuxSuRen

Makefile

@@ -10,7 +10,7 @@ run-image:
 	docker run ghcr.io/linuxsuren/api-testing:dev
 copy: build
 	sudo cp bin/atest /usr/local/bin/
-copy-restart:
+copy-restart: build
 	atest service stop
 	make copy
 	atest service restart

pkg/server/fake_server.go

@@ -39,6 +39,15 @@ func (s *fakeServer) GetVersion(ctx context.Context, in *Empty) (reply *HelloRep
 	return
 }
 
+// Sample returns a sample of the test task
+func (s *fakeServer) Sample(ctx context.Context, in *Empty) (reply *HelloReply, err error) {
+	reply = &HelloReply{
+		Message: "",
+	}
+	err = s.err
+	return
+}
+
 // NewFakeClient creates a fake client
 func NewFakeClient(ctx context.Context, version string, err error) (RunnerClient, func()) {
 	buffer := 101024 * 1024

pkg/server/remote_server.go

@@ -13,6 +13,7 @@ import (
 	"github.com/linuxsuren/api-testing/pkg/runner"
 	"github.com/linuxsuren/api-testing/pkg/testing"
 	"github.com/linuxsuren/api-testing/pkg/version"
+	"github.com/linuxsuren/api-testing/sample"
 )
 
 type server struct {
@@ -140,6 +141,12 @@ func (s *server) GetVersion(ctx context.Context, in *Empty) (reply *HelloReply,
 	return
 }
 
+// Sample returns a sample of the test task
+func (s *server) Sample(ctx context.Context, in *Empty) (reply *HelloReply, err error) {
+	reply = &HelloReply{Message: sample.TestSuiteGitLab}
+	return
+}
+
 func findParentTestCases(testcase *testing.TestCase, suite *testing.TestSuite) (testcases []testing.TestCase) {
 	reg, matchErr := regexp.Compile(`(.*?\{\{.*\.\w*.*?\}\})`)
 	targetReg, targetErr := regexp.Compile(`\.\w*`)

pkg/server/remote_server_test.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/h2non/gock"
 	atesting "github.com/linuxsuren/api-testing/pkg/testing"
+	"github.com/linuxsuren/api-testing/sample"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -61,6 +62,10 @@ func TestRemoteServer(t *testing.T) {
 	ver, err = server.GetVersion(context.TODO(), &Empty{})
 	assert.Empty(t, ver.Message)
 	assert.Nil(t, err)
+
+	ver, err = server.Sample(context.TODO(), &Empty{})
+	assert.Nil(t, err)
+	assert.Equal(t, sample.TestSuiteGitLab, ver.Message)
 }
 
 func TestFindParentTestCases(t *testing.T) {