Skip to content
GitLab
Select Git revision
  • rb_enableResponseHeaders
  • pre-dev
  • staging default
  • devbranch
  • ps_redisOtpFunctionality
  • SG_Conflict_new
  • SG_ConflictResolved
  • rb_conflictsResolved
  • SG_Conflict_resolved
  • ps_redisOtpAndAuth
  • jwtkidfeature
  • jwt-key-id-prajwalp
  • SG_mongoFunctions
  • SG_Mysql_TestCaseMongo
  • SG_Go1.21.10
  • ps_redisAuthCorepkg
  • rb_jwtClaimsChange
  • SG_AggregateOptions
  • coreimmudb-connection-failure-logger
  • SG_Findops
  • v1.2.17
  • v1.2.16
  • v1.2.15
  • v1.2.14
  • v1.2.13
  • v1.2.12
  • v1.2.11
  • v1.2.10
  • v1.2.9
  • v1.2.8
  • v1.2.7
  • v1.2.6
  • v1.2.5
  • v1.2.4
  • v1.2.3
  • v1.2.2
  • v1.2.1
  • v1.1.3
  • v1.2.0
  • v2.0.0-beta.1
Find file Blame History Permalink
httpclientmdl_test.go 6.52 KiB
1 
package httpclientmdl
2
3 
import (
4 
	"bytes"
5 
	"encoding/json"
6 
	"io/ioutil"
7 
	"net/http"
8 
	"testing"
9 
	"time"
10
11 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/errormdl"
12 
	"github.com/gin-gonic/gin"
13 
	"github.com/stretchr/testify/assert"
14 
)
15
16 
func TestGethttpclientwithconfigNotPanic(t *testing.T) {
17 
	httpTransport := HTTPTransport{}
18 
	httpTransport.MaxIdleConns = 10
19 
	httpTransport.MaxIdleConnsPerHost = 20
20 
	httpTransport.IdleConnTimeout = 90
21 
	assert.NotPanics(t, func() { GetHTTPClientWithConfig(httpTransport) }, "The code did  panic")
22 
}
23
24 
func TestGethttpclientwithconfigDefaults(t *testing.T) {
25 
	httpTransport := HTTPTransport{}
26 
	httpTransport.MaxIdleConns = 0
27 
	httpTransport.MaxIdleConnsPerHost = 0
28 
	httpTransport.IdleConnTimeout = 0
29 
	assert.NotPanics(t, func() { GetHTTPClientWithConfig(httpTransport) }, "The code did  panic")
30 
}
31 
func TestGethttpclientwithconfigNil(t *testing.T) {
32 
	httpTransport := HTTPTransport{}
33 
	httpTransport.MaxIdleConns = 10
34 
	httpTransport.MaxIdleConnsPerHost = 20
35 
	httpTransport.IdleConnTimeout = 90
36 
	assert.NotNil(t, func() { GetHTTPClientWithConfig(httpTransport) }, "Do not return nil")
37 
}
38 
func TestGethttpclientNil(t *testing.T) {
39 
	assert.NotNil(t, func() { GetHTTPClient() }, "Do not return nil")
40 
}
41 
func TestGethttpclientnotpanic(t *testing.T) {
42 
	assert.NotPanics(t, func() { GetHTTPClient() }, "The code did  panic")
43 
}
44
45 
func BenchmarkGethttpclient(b *testing.B) {
46 
	for i := 0; i < b.N; i++ {
47 
		GetHTTPClient()
48 
	}
49 
}
50
51 
func BenchmarkGethttpclientDefault(b *testing.B) {
52 
	for i := 0; i < b.N; i++ {
53 
		httpTransport := HTTPTransport{}
54 
		httpTransport.MaxIdleConns = 10
55 
		httpTransport.MaxIdleConnsPerHost = 20
56 
		httpTransport.IdleConnTimeout = 90
57 
		GetHTTPClientWithConfig(httpTransport)
58 
	}
59 
}
60
61 
var (
62 
	attempts  int
63 
	sleeptime time.Duration
64 
)
65
66 
func TestDoHttpWithRetryGET200(t *testing.T) {
67 
	attempts = 2
68 
	sleeptime = 2 * time.Second
69 
	request, _ := http.NewRequest("GET", "http://www.mkcl.org/about-mkcl", nil)
70 
	resp, err := DoHTTPWithRetry(attempts, sleeptime, request)
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
assert.NoError(t, err, "This should not return error") assert.Equal(t, resp.StatusCode, 200) } func TestDoHttpWithRetryGETError(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second request, _ := http.NewRequest("GET", "http://www.servernotrunning.org", nil) _, err := DoHTTPWithRetry(attempts, sleeptime, request) assert.Error(t, err, "This should return error") } func TestDoHttpWithRetryGETAttempt(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second request, _ := http.NewRequest("GET", "http://www.mkcl.org/about-mkcl-error", nil) resp, err := DoHTTPWithRetry(attempts, sleeptime, request) assert.NoError(t, err, "This should not return error") assert.Equal(t, resp.StatusCode, 404) } func TestDoHttpWithRetryGETAttempt2(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second request, _ := http.NewRequest("GET", "http://www.mkcl.org/about-mkcl", nil) errormdl.IsTestingNegetiveCaseOnCheckBool = true resp, err := DoHTTPWithRetry(attempts, sleeptime, request) errormdl.IsTestingNegetiveCaseOnCheckBool = false assert.NoError(t, err, "This should not return error") assert.Equal(t, resp.StatusCode, 200) } // Person struct type Person struct { Name string `json:"name"` Age string `json:"age"` } func addPerson(c *gin.Context) { person := Person{} defer c.Request.Body.Close() byteArr, readErr := ioutil.ReadAll(c.Request.Body) if readErr != nil { c.String(http.StatusInternalServerError, "") return } unmarshallErr := json.Unmarshal(byteArr, &person) if unmarshallErr != nil { c.String(http.StatusInternalServerError, "") return } c.String(http.StatusOK, "person1 is added successfully!") return } var router *gin.Engine func Init(o *gin.RouterGroup) { o.POST("/addPerson", addPerson) } func init() { router = gin.New() o := router.Group("/o") Init(o) router.Run("localhost:8081") } func TestDoHttpWithRetryPOST200(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second var jsonStr = []byte(`{"name":"Arnav","age":"4"}`)
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205
// request, _ := http.NewRequest("POST", "http://10.4.0.104:3001/o/addPerson", bytes.NewBuffer(jsonStr)) request, _ := http.NewRequest("POST", "http://localhost:8081/o/addPerson", bytes.NewBuffer(jsonStr)) resp, err := DoHTTPWithRetry(attempts, sleeptime, request) assert.NoError(t, err, "This should not return error") assert.Equal(t, resp.StatusCode, 200) } func TestDoHttpWithRetryPOSTError(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second var jsonStr = []byte(`{"name":"Arnav","age":"4"}`) request, _ := http.NewRequest("POST", "http://www.servernotrunning.org", bytes.NewBuffer(jsonStr)) _, err := DoHTTPWithRetry(attempts, sleeptime, request) assert.Error(t, err, "This should return error") } func TestDoHttpWithRetryPOSTAttempt(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second var jsonStr = []byte(`{"name":"Arnav","age":"4"}`) request, _ := http.NewRequest("POST", "http://www.mkcl.org/about-mkcl-error", bytes.NewBuffer(jsonStr)) resp, _ := DoHTTPWithRetry(attempts, sleeptime, request) assert.Equal(t, resp.StatusCode, 404) } // func TestDoHttpWithRetryPOSTAttempt2(t *testing.T) { // attempts = 2 // sleeptime = 2 * time.Second // var jsonStr = []byte(`{"name":"Arnav","age":"4"}`) // request, _ := http.NewRequest("POST", "http://10.4.0.104:3001/o/addPerson", bytes.NewBuffer(jsonStr)) // errormdl.IsTestingNegetiveCaseOnCheckBool = true // resp, err := DoHTTPWithRetry(attempts, sleeptime, request) // errormdl.IsTestingNegetiveCaseOnCheckBool = false // assert.NoError(t, err, "This should not return error") // assert.Equal(t, resp.StatusCode, 200) // } func TestDoHttpWithRetryPOSTReadAllError(t *testing.T) { attempts = 2 sleeptime = 2 * time.Second var jsonStr = []byte(`{"name":"Arnav","age":"4"}`) request, _ := http.NewRequest("POST", "http://www.mkcl.org/about-mkcl-error", bytes.NewBuffer(jsonStr)) errormdl.IsTestingNegetiveCaseOnCheckBool = true _, err := DoHTTPWithRetry(attempts, sleeptime, request) errormdl.IsTestingNegetiveCaseOnCheckBool = false assert.Error(t, err, "This should return error") } func BenchmarkDoHttpWithRetry(b *testing.B) { for i := 0; i < b.N; i++ { attempts := 2 sleeptime := 2 * time.Second request, _ := http.NewRequest("GET", "http://www.mkcl.org/about-mkcl-error", nil) DoHTTPWithRetry(attempts, sleeptime, request) } } func BenchmarkDoHttpWithRetryPOST(b *testing.B) { for i := 0; i < b.N; i++ { attempts := 2 sleeptime := 2 * time.Second var jsonStr = []byte(`{"name":"Arnav","age":"4"}`) request, _ := http.NewRequest("POST", "http://www.mkcl.org/about-mkcl-error", bytes.NewBuffer(jsonStr)) DoHTTPWithRetry(attempts, sleeptime, request) } }