Merge pull request #4 from chipim/feat/verify-stub-times-called

Merge: add new endpoint to verify number of calls for stub

Author: ns-jjiang

stub/storage.go

@@ -22,8 +22,9 @@ type matchFunc func(interface{}, interface{}) bool
 var stubStorage = stubMapping{}
 
 type storage struct {
-	Input  Input
-	Output Output
+	Input       Input
+	Output      Output
+	timesCalled int
 }
 
 func storeStub(stub *Stub) error {
@@ -73,24 +74,27 @@ func findStub(stub *findStubPayload) (*Output, error) {
 	}
 
 	closestMatch := []closeMatch{}
-	for _, stubrange := range stubs {
+	for idx, stubrange := range stubs {
 		if expect := stubrange.Input.Equals; expect != nil {
 			closestMatch = append(closestMatch, closeMatch{"equals", expect})
 			if equals(stub.Data, expect) {
+				stubs[idx].timesCalled++
 				return &stubrange.Output, nil
 			}
 		}
 
 		if expect := stubrange.Input.Contains; expect != nil {
 			closestMatch = append(closestMatch, closeMatch{"contains", expect})
 			if contains(stubrange.Input.Contains, stub.Data) {
+				stubs[idx].timesCalled++
 				return &stubrange.Output, nil
 			}
 		}
 
 		if expect := stubrange.Input.Matches; expect != nil {
 			closestMatch = append(closestMatch, closeMatch{"matches", expect})
 			if matches(stubrange.Input.Matches, stub.Data) {
+				stubs[idx].timesCalled++
 				return &stubrange.Output, nil
 			}
 		}
@@ -318,3 +322,33 @@ func (sm *stubMapping) readStubFromFile(path string) {
 		sm.storeStub(stub)
 	}
 }
+
+func getStubTimesCalled(stub *verifyStubCallPayload) (*Output, error) {
+	mx.Lock()
+	defer mx.Unlock()
+
+	if _, ok := stubStorage[stub.Service]; !ok {
+		return nil, fmt.Errorf("Can't find stub for Service: %s", stub.Service)
+	}
+
+	if _, ok := stubStorage[stub.Service][stub.Method]; !ok {
+		return nil, fmt.Errorf("Can't find stub for Service:%s and Method:%s", stub.Service, stub.Method)
+	}
+
+	stubs := stubStorage[stub.Service][stub.Method]
+	if len(stubs) == 0 {
+		return nil, fmt.Errorf("Stub for Service:%s and Method:%s is empty", stub.Service, stub.Method)
+	}
+
+	totalTimesCalled := 0
+
+	for _, stubrange := range stubs {
+		totalTimesCalled += stubrange.timesCalled
+	}
+
+	return &Output{
+		Data: map[string]interface{}{
+			"timesCalled": totalTimesCalled,
+		},
+	}, nil
+}

stub/stub.go

@@ -30,6 +30,7 @@ func RunStubServer(opt Options) {
 	r.Get("/", listStub)
 	r.Post("/find", handleFindStub)
 	r.Get("/clear", handleClearStub)
+	r.Get("/verify", handleVerifyStubCalled)
 
 	if opt.StubPath != "" {
 		readStubFromFile(opt.StubPath)
@@ -165,3 +166,27 @@ func handleClearStub(w http.ResponseWriter, r *http.Request) {
 	clearStorage()
 	w.Write([]byte("OK"))
 }
+
+type verifyStubCallPayload struct {
+	Service string `json:"service"`
+	Method  string `json:"method"`
+}
+
+func handleVerifyStubCalled(w http.ResponseWriter, r *http.Request) {
+	verifyPayload := new(verifyStubCallPayload)
+	err := json.NewDecoder(r.Body).Decode(verifyPayload)
+	if err != nil {
+		responseError(err, w)
+		return
+	}
+
+	output, err := getStubTimesCalled(verifyPayload)
+	if err != nil {
+		log.Println(err)
+		responseError(err, w)
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(output)
+}

stub/stub_test.go

@@ -397,6 +397,15 @@ func TestStub(t *testing.T) {
 			handler: handleFindStub,
 			expect:  "Can't find stub \n\nService: Testing \n\nMethod: TestMethod \n\nInput\n\n{\n\tHola: Dunia\n}\n\nClosest Match \n\nequals:{\n\tHola: Mundo\n}",
 		},
+		{
+			name: "verify stub times called",
+			mock: func() *http.Request {
+				payload := `{"service":"Testing2","method":"TestMethod"}`
+				return httptest.NewRequest("GET", "/verify", bytes.NewReader([]byte(payload)))
+			},
+			handler: handleVerifyStubCalled,
+			expect:  "{\"data\":{\"timesCalled\":1},\"error\":\"\"}\n",
+		},
 	}
 
 	for _, v := range cases {