Merge branch 'master' into kkumar-gcc/localization

auth/auth_test.go

@@ -80,6 +80,10 @@ func (mc *Context) Response() http.ContextResponse {
 	return mc.response
 }
 
+func (mc *Context) SetContext(ctx http.Context) {
+	mc.ctx = ctx.Context()
+}
+
 func Background() http.Context {
 	return &Context{
 		ctx:      context.Background(),

contracts/http/context.go

@@ -29,4 +29,6 @@ type Context interface {
 	Request() ContextRequest
 	// Response returns the ContextResponse
 	Response() ContextResponse
+	// SetContext set the custom context
+	SetContext(ctx Context)
 }

http/context.go

@@ -38,3 +38,7 @@ func (r *Context) Request() http.ContextRequest {
 func (r *Context) Response() http.ContextResponse {
 	return nil
 }
+
+func (r *Context) SetContext(ctx http.Context) {
+	r.Ctx = ctx.Context()
+}

http/context_test.go

@@ -36,3 +36,9 @@ func (s *ContextTestSuite) TestRequest() {
 func (s *ContextTestSuite) TestResponse() {
 	s.Nil(s.ctx.Response())
 }
+
+func (s *ContextTestSuite) TestSetContext() {
+	ctx := NewContext()
+	s.ctx.SetContext(ctx)
+	s.Equal(ctx.Context(), s.ctx.Context())
+}

http/middleware/throttle_test.go

@@ -283,6 +283,10 @@ func (r *TestContext) Response() contractshttp.ContextResponse {
 	return r.response
 }
 
+func (r *TestContext) SetContext(ctx contractshttp.Context) {
+	panic("do not need to implement it")
+}
+
 type TestRequest struct{}
 
 func (r *TestRequest) Header(key string, defaultValue ...string) string {