|  | @@ -0,0 +1,37 @@
 | 
	
		
			
			|  | 1 | +package request
 | 
	
		
			
			|  | 2 | +
 | 
	
		
			
			|  | 3 | +import (
 | 
	
		
			
			|  | 4 | +	"context"
 | 
	
		
			
			|  | 5 | +	"io"
 | 
	
		
			
			|  | 6 | +	"net/http"
 | 
	
		
			
			|  | 7 | +	"net/http/httptest"
 | 
	
		
			
			|  | 8 | +)
 | 
	
		
			
			|  | 9 | +
 | 
	
		
			
			|  | 10 | +// HttpRequestConfig http request config
 | 
	
		
			
			|  | 11 | +type HttpRequestConfig struct {
 | 
	
		
			
			|  | 12 | +	Method  string
 | 
	
		
			
			|  | 13 | +	URL     string
 | 
	
		
			
			|  | 14 | +	Param   io.Reader
 | 
	
		
			
			|  | 15 | +	Header  map[string]string
 | 
	
		
			
			|  | 16 | +	Context context.Context
 | 
	
		
			
			|  | 17 | +}
 | 
	
		
			
			|  | 18 | +
 | 
	
		
			
			|  | 19 | +// AssertHttpRequest check http request
 | 
	
		
			
			|  | 20 | +func AssertHttpRequest(handler http.Handler, config HttpRequestConfig) *httptest.ResponseRecorder {
 | 
	
		
			
			|  | 21 | +	r := httptest.NewRequest(config.Method, config.URL, config.Param)
 | 
	
		
			
			|  | 22 | +	w := httptest.NewRecorder()
 | 
	
		
			
			|  | 23 | +
 | 
	
		
			
			|  | 24 | +	if config.Header != nil {
 | 
	
		
			
			|  | 25 | +		for k, v := range config.Header {
 | 
	
		
			
			|  | 26 | +			r.Header.Set(k, v)
 | 
	
		
			
			|  | 27 | +		}
 | 
	
		
			
			|  | 28 | +	}
 | 
	
		
			
			|  | 29 | +
 | 
	
		
			
			|  | 30 | +	if config.Context != nil {
 | 
	
		
			
			|  | 31 | +		r.WithContext(config.Context)
 | 
	
		
			
			|  | 32 | +	}
 | 
	
		
			
			|  | 33 | +
 | 
	
		
			
			|  | 34 | +	handler.ServeHTTP(w, r)
 | 
	
		
			
			|  | 35 | +
 | 
	
		
			
			|  | 36 | +	return w
 | 
	
		
			
			|  | 37 | +}
 |