Added AddRouteToContext method (#77)

Co-authored-by: Jacob Stuart <[email protected]>

context.go

@@ -172,14 +172,19 @@ func AddRouteDataToContext(ctx context.Context, data ContextRouteData) context.C
 }
 
 // AddParamsToContext inserts a parameters map into a context using
-// the package's internal context key. This function is deprecated.
-// Use AddRouteDataToContext instead.
+// the package's internal context key.
 func AddParamsToContext(ctx context.Context, params map[string]string) context.Context {
-	data := &contextData{
-		route:  "",
+	return AddRouteDataToContext(ctx, &contextData{
 		params: params,
-	}
-	return AddRouteDataToContext(ctx, data)
+	})
+}
+
+// AddRouteToContext inserts a route into a context using
+// the package's internal context key.
+func AddRouteToContext(ctx context.Context, route string) context.Context {
+	return AddRouteDataToContext(ctx, &contextData{
+		route: route,
+	})
 }
 
 type contextKey int

context_test.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"net/http"
 	"net/http/httptest"
+	"reflect"
 	"testing"
 )
 
@@ -362,3 +363,56 @@ func TestContextMuxSimple(t *testing.T) {
 	t.Log("Testing with DefaultContext")
 	router.ServeHTTP(w, r)
 }
+
+func TestAddDataToContext(t *testing.T) {
+	expectedRoute := "/expected/route"
+	expectedParams := map[string]string{
+		"test": "expected",
+	}
+
+	ctx := AddRouteDataToContext(context.Background(), &contextData{
+		route: expectedRoute,
+		params: expectedParams,
+	})
+
+	if gotData, ok := ctx.Value(contextDataKey).(*contextData); ok && gotData != nil {
+		if gotData.route != expectedRoute {
+			t.Errorf("Did not retrieve the desired route. Expected: %s; Got: %s", expectedRoute, gotData.route)
+		}
+		if !reflect.DeepEqual(expectedParams, gotData.params) {
+			t.Errorf("Did not retrieve the desired parameters. Expected: %#v; Got: %#v", expectedParams, gotData.params)
+		}
+	} else {
+		t.Error("failed to retrieve context data")
+	}
+}
+
+func TestAddParamsToContext(t *testing.T) {
+	expectedParams := map[string]string{
+		"test": "expected",
+	}
+
+	ctx := AddParamsToContext(context.Background(), expectedParams)
+
+	if gotData, ok := ctx.Value(contextDataKey).(*contextData); ok && gotData != nil {
+		if !reflect.DeepEqual(expectedParams, gotData.params) {
+			t.Errorf("Did not retrieve the desired parameters. Expected: %#v; Got: %#v", expectedParams, gotData.params)
+		}
+	} else {
+		t.Error("failed to retrieve context data")
+	}
+}
+
+func TestAddRouteToContext(t *testing.T) {
+	expectedRoute := "/expected/route"
+
+	ctx := AddRouteToContext(context.Background(), expectedRoute)
+
+	if gotData, ok := ctx.Value(contextDataKey).(*contextData); ok && gotData != nil {
+		if gotData.route != expectedRoute {
+			t.Errorf("Did not retrieve the desired route. Expected: %s; Got: %s", expectedRoute, gotData.route)
+		}
+	} else {
+		t.Error("failed to retrieve context data")
+	}
+}