feat: reset all func

Author: slzhffktm

go.mod

@@ -8,6 +8,6 @@ require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/julienschmidt/httprouter v1.3.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
-	github.com/slzhffktm/httprouter v0.0.2 // indirect
+	github.com/slzhffktm/httprouter v0.0.3 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

go.sum

@@ -6,6 +6,8 @@ github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZb
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/slzhffktm/httprouter v0.0.2 h1:alUI45TxyQkJpLOL0EmYmO+9KzH8L5JV5yyq/nbxFEA=
 github.com/slzhffktm/httprouter v0.0.2/go.mod h1:WbWqTm2WQoxF5XQm7wLrLgGEWCSBrvLv0x3CF3THxnk=
+github.com/slzhffktm/httprouter v0.0.3 h1:GJPKfirTnAqe+L/yzFW/fMRvlykbqhbRMjFwzs0qRbM=
+github.com/slzhffktm/httprouter v0.0.3/go.mod h1:WbWqTm2WQoxF5XQm7wLrLgGEWCSBrvLv0x3CF3THxnk=
 github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
 github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=

httptest.go

@@ -65,7 +65,12 @@ func (s *Server) Close() error {
 
 // GetNCalls returns the number of calls for a path.
 func (s *Server) GetNCalls(method, path string) int {
-	return s.calls[method][path]
+	calls, ok := s.calls[method][path]
+	if !ok {
+		return 0
+	}
+
+	return calls
 }
 
 // ResetNCalls resets the number of calls for all paths.
@@ -92,3 +97,9 @@ func (s *Server) RegisterHandler(method string, path string, handler ServerHandl
 		})
 	})
 }
+
+// ResetAll resets all the calls and handlers.
+func (s *Server) ResetAll() {
+	s.router.ClearHandlers()
+	s.calls = map[string]map[string]int{}
+}

httptest_test.go

@@ -267,3 +267,32 @@ func (s *serverTestSuite) TestSimulateTimeout() {
 
 	s.Equal(1, server.GetNCalls(http.MethodGet, path))
 }
+
+func (s *serverTestSuite) TestResetAll() {
+	server, err := httptest.NewServer(address, httptest.ServerConfig{})
+	s.NoError(err)
+	defer server.Close()
+
+	path := "/some-path/subpath"
+	expectedResBody := []byte(`{"res":"ponse"}`)
+
+	server.RegisterHandler(http.MethodGet, path, func(w httptest.ResponseWriter, r *httptest.Request) {
+		w.SetStatusCode(http.StatusOK)
+		w.SetBodyBytes(expectedResBody)
+	})
+
+	server.ResetAll()
+
+	res, _, err := s.httpClient.Do(
+		ctx,
+		http.MethodGet,
+		path,
+		nil,
+		nil,
+		nil,
+	)
+	s.NoError(err)
+	s.Equal(http.StatusNotFound, res.StatusCode)
+
+	s.Equal(0, server.GetNCalls(http.MethodGet, path))
+}