repo_description: LinDB is a scalable, high performance, high availability distributed time series database.
repo_topics: cross-datacenter-native database distributed-database go high-performance metrics monitoring timeseries
code: // Licensed to LinDB under one or more contributor
  // license agreements. See the NOTICE file distributed with
  // this work for additional information regarding copyright
  // ownership. LinDB licenses this file to you under
  // the Apache License, Version 2.0 (the "License"); you may
  // not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //     http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  
  package mock
  
  import (
  	"bytes"
  	"context"
  	"io"
  	"net/http"
  	"net/http/httptest"
  	"testing"
  
  	"github.com/gin-gonic/gin"
  )
  
  func init() {
  	gin.SetMode(gin.ReleaseMode)
  }
  
  type closeNotifyingRecorder struct {
  	*httptest.ResponseRecorder
  	closed chan bool
  }
  
  func newCloseNotifyingRecorder() *closeNotifyingRecorder {
  	return &closeNotifyingRecorder{
  		httptest.NewRecorder(),
  		make(chan bool, 1),
  	}
  }
  
  func (c *closeNotifyingRecorder) CloseNotify() <-chan bool {
  	return c.closed
  }
  
  // DoRequest does http request for test.
  func DoRequest(t *testing.T, r *gin.Engine, method, path, reqBody string, headers ...http.Header) *httptest.ResponseRecorder {
  	t.Helper()
  
  	var body io.Reader
  	if reqBody != "" {
  		body = bytes.NewBufferString(reqBody)
  	}
  	req, _ := http.NewRequestWithContext(context.TODO(), method, path, body)
  	if len(headers) == 0 {
  		req.Header.Set("content-type", "application/json")
  	} else {
  		for _, header := range headers {
  			req.Header = header
  		}
  	}
  	resp := newCloseNotifyingRecorder()
  	r.ServeHTTP(resp, req)
  	return resp.ResponseRecorder
  }
  