Merge pull request #77 from chriskempf-wf/path_trim

RM-16227 Add path trimming middleware with unit tests

Author: colefeisthamel-wf

rest/middleware/path_trim.go

@@ -0,0 +1,18 @@
+package middleware
+
+import(
+	"net/http"
+	"strings"
+
+	"github.com/Workiva/go-rest/rest"
+)
+
+// NewPathTrimMiddleware returns a Middleware which inspects request paths
+// and removes the content of trimPortion from the prefix. With a path of /foo/bar
+// and a trimpPortion of /foo you would be left with a path of /bar.
+func NewPathTrimMiddleware(trimPortion string) rest.Middleware {
+	return func(w http.ResponseWriter, r *http.Request) *rest.MiddlewareError {
+		r.URL.Path = strings.TrimPrefix(r.URL.Path, trimPortion)
+		return nil
+	}
+}

rest/middleware/path_trim_test.go

@@ -0,0 +1,30 @@
+package middleware
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// Ensure that PathTrimMiddleware is correctly trimming the request path
+func TestPathTrimMiddlewareTrimPrefix(t *testing.T) {
+	assert := assert.New(t)
+	req, _ := http.NewRequest("GET", "http://example.com/foo/bar", nil)
+	w := httptest.NewRecorder()
+
+	assert.Nil(NewPathTrimMiddleware("/foo")(w, req))
+	assert.Equal(req.URL.Path, "/bar")
+}
+
+// Ensure that PathTrimMiddleware correclty passes a path that doesn't need 
+// to be trimmed
+func TestPathTrimMiddlewareNoTrimPrefix(t *testing.T) {
+	assert := assert.New(t)
+	req, _ := http.NewRequest("GET", "http://example.com/foo/bar", nil)
+	w := httptest.NewRecorder()
+
+	assert.Nil(NewPathTrimMiddleware("/baz")(w, req))
+	assert.Equal(req.URL.Path, "/foo/bar")
+}