github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-07-21 19:43:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed redact acceptance test (#472)

Browse Source
This commit is contained in:
RoyUP9 2021-11-16 15:49:08 +02:00 committed by GitHub
parent 89836d8d75
commit 1c75ce314b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
acceptanceTests/tap_test.go
View File

@ -427,9 +427,10 @@ func TestTapRedact(t *testing.T) {
} }
proxyUrl := getProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := getProxyUrl(defaultNamespaceName, defaultServiceName)
requestHeaders := map[string]string{"User-Header": "Mizu"}
requestBody := map[string]string{"User": "Mizu"} requestBody := map[string]string{"User": "Mizu"}
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < defaultEntriesCount; i++ {
if _, requestErr := executeHttpPostRequest(fmt.Sprintf("%v/post", proxyUrl), requestBody); requestErr != nil { if _, requestErr := executeHttpPostRequestWithHeaders(fmt.Sprintf("%v/post", proxyUrl), requestHeaders, requestBody); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
return return
} }
@ -460,12 +461,12 @@ func TestTapRedact(t *testing.T) {
headers := request["_headers"].([]interface{}) headers := request["_headers"].([]interface{})
for _, headerInterface := range headers { for _, headerInterface := range headers {
header := headerInterface.(map[string]interface{}) header := headerInterface.(map[string]interface{})
if header["name"].(string) != "User-Agent" { if header["name"].(string) != "User-Header" {
continue continue
} }
userAgent := header["value"].(string) userHeader := header["value"].(string)
if userAgent != "[REDACTED]" { if userHeader != "[REDACTED]" {
return fmt.Errorf("unexpected result - user agent is not redacted") return fmt.Errorf("unexpected result - user agent is not redacted")
} }
} }
@ -530,9 +531,10 @@ func TestTapNoRedact(t *testing.T) {
} }
proxyUrl := getProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := getProxyUrl(defaultNamespaceName, defaultServiceName)
requestHeaders := map[string]string{"User-Header": "Mizu"}
requestBody := map[string]string{"User": "Mizu"} requestBody := map[string]string{"User": "Mizu"}
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < defaultEntriesCount; i++ {
if _, requestErr := executeHttpPostRequest(fmt.Sprintf("%v/post", proxyUrl), requestBody); requestErr != nil { if _, requestErr := executeHttpPostRequestWithHeaders(fmt.Sprintf("%v/post", proxyUrl), requestHeaders, requestBody); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
return return
} }
@ -563,12 +565,12 @@ func TestTapNoRedact(t *testing.T) {
headers := request["_headers"].([]interface{}) headers := request["_headers"].([]interface{})
for _, headerInterface := range headers { for _, headerInterface := range headers {
header := headerInterface.(map[string]interface{}) header := headerInterface.(map[string]interface{})
if header["name"].(string) != "User-Agent" { if header["name"].(string) != "User-Header" {
continue continue
} }
userAgent := header["value"].(string) userHeader := header["value"].(string)
if userAgent == "[REDACTED]" { if userHeader == "[REDACTED]" {
return fmt.Errorf("unexpected result - user agent is redacted") return fmt.Errorf("unexpected result - user agent is redacted")
} }
} }

15
acceptanceTests/testsUtils.go
View File

@ -240,13 +240,24 @@ func executeHttpGetRequest(url string) (interface{}, error) {
return executeHttpRequest(response, requestErr) return executeHttpRequest(response, requestErr)
} }
func executeHttpPostRequest(url string, body interface{}) (interface{}, error) { func executeHttpPostRequestWithHeaders(url string, headers map[string]string, body interface{}) (interface{}, error) {
requestBody, jsonErr := json.Marshal(body) requestBody, jsonErr := json.Marshal(body)
if jsonErr != nil { if jsonErr != nil {
return nil, jsonErr return nil, jsonErr
} }
response, requestErr := http.Post(url, "application/json", bytes.NewBuffer(requestBody)) request, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer(requestBody))
if err != nil {
return nil, err
}
request.Header.Add("Content-Type", "application/json")
for headerKey, headerValue := range headers {
request.Header.Add(headerKey, headerValue)
}
client := &http.Client{}
response, requestErr := client.Do(request)
return executeHttpRequest(response, requestErr) return executeHttpRequest(response, requestErr)
} }