added feature for extra delay for grpc apis

Author: ns-jjiang

stub/duration.go

@@ -0,0 +1,37 @@
+package stub
+
+import (
+	"encoding/json"
+	"errors"
+	"time"
+)
+
+type Duration struct {
+	time.Duration
+}
+
+func (d Duration) MarshalJSON() ([]byte, error) {
+	return json.Marshal(d.String())
+}
+
+func (d *Duration) UnmarshalJSON(b []byte) error {
+	var v interface{}
+	if err := json.Unmarshal(b, &v); err != nil {
+		return err
+	}
+	switch value := v.(type) {
+	case float64:
+		d.Duration = time.Duration(value)
+		return nil
+	case string:
+		var err error
+		d.Duration, err = time.ParseDuration(value)
+		if err != nil {
+			return err
+		}
+		return nil
+	default:
+		return errors.New("invalid duration")
+	}
+}
+

stub/storage.go

@@ -8,6 +8,7 @@ import (
 	"reflect"
 	"regexp"
 	"sync"
+	"time"
 
 	"github.com/lithammer/fuzzysearch/fuzzy"
 )
@@ -105,6 +106,7 @@ func findStub(stub *findStubPayload) (*Output, error) {
 			closestMatch = append(closestMatch, closeMatch{"equals", expect})
 			if equals(stub.Data, expect) {
 				stubs[idx].timesCalled++
+				time.Sleep(stubrange.Output.Delay.Duration)
 				return &stubrange.Output, nil
 			}
 		}
@@ -113,6 +115,7 @@ func findStub(stub *findStubPayload) (*Output, error) {
 			closestMatch = append(closestMatch, closeMatch{"contains", expect})
 			if contains(stubrange.Input.Contains, stub.Data) {
 				stubs[idx].timesCalled++
+				time.Sleep(stubrange.Output.Delay.Duration)
 				return &stubrange.Output, nil
 			}
 		}
@@ -121,6 +124,7 @@ func findStub(stub *findStubPayload) (*Output, error) {
 			closestMatch = append(closestMatch, closeMatch{"matches", expect})
 			if matches(stubrange.Input.Matches, stub.Data) {
 				stubs[idx].timesCalled++
+				time.Sleep(stubrange.Output.Delay.Duration)
 				return &stubrange.Output, nil
 			}
 		}

stub/stub.go

@@ -64,6 +64,7 @@ type Input struct {
 }
 
 type Output struct {
+	Delay Duration               `json:"delay"`
 	Data  map[string]interface{} `json:"data"`
 	Error string                 `json:"error"`
 	Code  *codes.Code            `json:"code,omitempty"`

stub/stub_test.go

@@ -31,9 +31,11 @@ func TestStub(t *testing.T) {
 							}
 						},
 						"output":{
+                            "delay": "1s",
 							"data":{
 								"Hello":"World"
 							}
+
 						}
 					}`
 				read := bytes.NewReader([]byte(payload))
@@ -48,7 +50,7 @@ func TestStub(t *testing.T) {
 				return httptest.NewRequest("GET", "/", nil)
 			},
 			handler: listStub,
-			expect:  "{\"Testing\":{\"TestMethod\":[{\"Input\":{\"equals\":{\"Hola\":\"Mundo\"},\"contains\":null,\"matches\":null},\"Output\":{\"data\":{\"Hello\":\"World\"},\"error\":\"\"}}]}}\n",
+			expect:  "{\"Testing\":{\"TestMethod\":[{\"Input\":{\"equals\":{\"Hola\":\"Mundo\"},\"contains\":null,\"matches\":null},\"Output\":{\"delay\":\"1s\",\"data\":{\"Hello\":\"World\"},\"error\":\"\"}}]}}\n",
 		},
 		{
 			name: "find stub equals",
@@ -57,7 +59,7 @@ func TestStub(t *testing.T) {
 				return httptest.NewRequest("POST", "/find", bytes.NewReader([]byte(payload)))
 			},
 			handler: handleFindStub,
-			expect:  "{\"data\":{\"Hello\":\"World\"},\"error\":\"\"}\n",
+			expect:  "{\"delay\":\"1s\",\"data\":{\"Hello\":\"World\"},\"error\":\"\"}\n",
 		},
 		{
 			name: "add nested stub equals",
@@ -290,7 +292,7 @@ func TestStub(t *testing.T) {
 				return httptest.NewRequest("GET", "/find", bytes.NewReader([]byte(payload)))
 			},
 			handler: handleFindStub,
-			expect:  "{\"data\":{\"hello\":\"world\"},\"error\":\"\"}\n",
+			expect:  "{\"delay\":\"0s\",\"data\":{\"hello\":\"world\"},\"error\":\"\"}\n",
 		},
 		{
 			name: "add stub matches regex",
@@ -379,7 +381,7 @@ func TestStub(t *testing.T) {
 				return httptest.NewRequest("GET", "/find", bytes.NewReader([]byte(payload)))
 			},
 			handler: handleFindStub,
-			expect:  "{\"data\":{\"reply\":\"OK\"},\"error\":\"\"}\n",
+			expect:  "{\"delay\":\"0s\",\"data\":{\"reply\":\"OK\"},\"error\":\"\"}\n",
 		},
 		{
 			name: "error find stub contains",