Mirrors/VILLASweb-backend-go
1
0
Fork 0
mirror of https://git.rwth-aachen.de/acs/public/villas/web-backend-go/ synced 2025-03-30 00:00:12 +01:00
Code Issues Releases Wiki Activity

Improves LengthOfResponse()

Browse source
This commit is contained in:
smavros 2019-08-17 20:53:33 +02:00
parent ff19fdcdaf
commit d3aa873732
1 changed files with 15 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
common/utilities.go
View file

@ -41,15 +41,18 @@ func LengthOfResponse(router *gin.Engine, token string, url string,
w := httptest.NewRecorder() w := httptest.NewRecorder()
if body != nil { req, err := http.NewRequest(method, url, nil)
req, _ := http.NewRequest(method, url, bytes.NewBuffer(body)) if err != nil {
return 0, fmt.Errorf("Failed to create new request: %v", err)
}
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Add("Authorization", "Bearer "+token) req.Header.Add("Authorization", "Bearer "+token)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
} else {
req, _ := http.NewRequest(method, url, nil) // HTTP Code of response must be 200
req.Header.Add("Authorization", "Bearer "+token) if w.Code != 200 {
router.ServeHTTP(w, req) return 0, fmt.Errorf("HTTP Code: Expected \"200\". Got \"%v\""+
".\nResponse message:\n%v", w.Code, w.Body.String())
} }
// Convert the response in array of bytes // Convert the response in array of bytes
@ -58,11 +61,11 @@ func LengthOfResponse(router *gin.Engine, token string, url string,
// First we are trying to unmarshal the response into an array of // First we are trying to unmarshal the response into an array of
// general type variables ([]interface{}). If this fails we will try // general type variables ([]interface{}). If this fails we will try
// to unmarshal into a single general type variable (interface{}). // to unmarshal into a single general type variable (interface{}).
// If that also fails we will return -1. // If that also fails we will return 0.
// Response might be array of objects // Response might be array of objects
var arrayResponse map[string][]interface{} var arrayResponse map[string][]interface{}
err := json.Unmarshal(responseBytes, &arrayResponse) err = json.Unmarshal(responseBytes, &arrayResponse)
if err == nil { if err == nil {
// Get an arbitrary key from tha map. The only key (entry) of // Get an arbitrary key from tha map. The only key (entry) of
@ -80,8 +83,7 @@ func LengthOfResponse(router *gin.Engine, token string, url string,
return 1, nil return 1, nil
} }
// Failed to identify response. It means we got a different HTTP // Failed to identify response.
// code than 200.
return 0, fmt.Errorf("Length of response cannot be detected") return 0, fmt.Errorf("Length of response cannot be detected")
} }