github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-24 15:24:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed redact and regex masking tests (#284)

Browse Source
This commit is contained in:
RoyUP9 2021-09-19 11:52:11 +03:00 committed by GitHub
parent 9fa9b67328
commit 842d95c836
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 41 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
acceptanceTests/tap_test.go
View File

@ -502,10 +502,19 @@ func TestTapRedact(t *testing.T) {
return fmt.Errorf("failed to get entry, err: %v", requestErr) return fmt.Errorf("failed to get entry, err: %v", requestErr)
} }
entry := requestResult.(map[string]interface{})["entry"].(map[string]interface{}) data := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryRequest := entry["request"].(map[string]interface{}) entryJson := data["entry"].(string)
headers := entryRequest["headers"].([]interface{}) var entry map[string]interface{}
if parseErr := json.Unmarshal([]byte(entryJson), &entry); parseErr != nil {
return fmt.Errorf("failed to parse entry, err: %v", parseErr)
}
entryRequest := entry["request"].(map[string]interface{})
entryPayload := entryRequest["payload"].(map[string]interface{})
entryDetails := entryPayload["details"].(map[string]interface{})
headers := entryDetails["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-Agent" {
@ -518,8 +527,8 @@ func TestTapRedact(t *testing.T) {
} }
} }
data := entryRequest["postData"].(map[string]interface{}) postData := entryDetails["postData"].(map[string]interface{})
textDataStr := data["text"].(string) textDataStr := postData["text"].(string)
var textData map[string]string var textData map[string]string
if parseErr := json.Unmarshal([]byte(textDataStr), &textData); parseErr != nil { if parseErr := json.Unmarshal([]byte(textDataStr), &textData); parseErr != nil {
@ -608,10 +617,19 @@ func TestTapNoRedact(t *testing.T) {
return fmt.Errorf("failed to get entry, err: %v", requestErr) return fmt.Errorf("failed to get entry, err: %v", requestErr)
} }
entry := requestResult.(map[string]interface{})["entry"].(map[string]interface{}) data := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryRequest := entry["request"].(map[string]interface{}) entryJson := data["entry"].(string)
headers := entryRequest["headers"].([]interface{}) var entry map[string]interface{}
if parseErr := json.Unmarshal([]byte(entryJson), &entry); parseErr != nil {
return fmt.Errorf("failed to parse entry, err: %v", parseErr)
}
entryRequest := entry["request"].(map[string]interface{})
entryPayload := entryRequest["payload"].(map[string]interface{})
entryDetails := entryPayload["details"].(map[string]interface{})
headers := entryDetails["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-Agent" {
@ -624,8 +642,8 @@ func TestTapNoRedact(t *testing.T) {
} }
} }
data := entryRequest["postData"].(map[string]interface{}) postData := entryDetails["postData"].(map[string]interface{})
textDataStr := data["text"].(string) textDataStr := postData["text"].(string)
var textData map[string]string var textData map[string]string
if parseErr := json.Unmarshal([]byte(textDataStr), &textData); parseErr != nil { if parseErr := json.Unmarshal([]byte(textDataStr), &textData); parseErr != nil {
@ -714,11 +732,20 @@ func TestTapRegexMasking(t *testing.T) {
return fmt.Errorf("failed to get entry, err: %v", requestErr) return fmt.Errorf("failed to get entry, err: %v", requestErr)
} }
entry := requestResult.(map[string]interface{})["entry"].(map[string]interface{}) data := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryRequest := entry["request"].(map[string]interface{}) entryJson := data["entry"].(string)
data := entryRequest["postData"].(map[string]interface{}) var entry map[string]interface{}
textData := data["text"].(string) if parseErr := json.Unmarshal([]byte(entryJson), &entry); parseErr != nil {
return fmt.Errorf("failed to parse entry, err: %v", parseErr)
}
entryRequest := entry["request"].(map[string]interface{})
entryPayload := entryRequest["payload"].(map[string]interface{})
entryDetails := entryPayload["details"].(map[string]interface{})
postData := entryDetails["postData"].(map[string]interface{})
textData := postData["text"].(string)
if textData != "[REDACTED]" { if textData != "[REDACTED]" {
return fmt.Errorf("unexpected result - body is not redacted") return fmt.Errorf("unexpected result - body is not redacted")