github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-22 14:32:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix rest of the failing acceptance tests

Browse Source
This commit is contained in:
M. Mert Yildiran 2021-11-10 01:05:43 +03:00
parent f4438bd61e
commit df7911aeee
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
1 changed files with 18 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
acceptanceTests/tap_test.go
View File

@ -56,7 +56,9 @@ func checkDBHasEntries(t *testing.T, timestamp int64, limit int) (entries []map[
err = json.Unmarshal([]byte(message), &data) err = json.Unmarshal([]byte(message), &data)
assert.Nil(t, err) assert.Nil(t, err)
entries = append(entries, data) if data["messageType"] == "entry" {
entries = append(entries, data)
}
} }
} }
@ -502,19 +504,10 @@ 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)
} }
data := requestResult.(map[string]interface{})["data"].(map[string]interface{}) entry := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryJson := data["entry"].(string) request := entry["request"].(map[string]interface{})
var entry map[string]interface{} headers := request["_headers"].([]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" {
@ -527,7 +520,7 @@ func TestTapRedact(t *testing.T) {
} }
} }
postData := entryDetails["postData"].(map[string]interface{}) postData := request["postData"].(map[string]interface{})
textDataStr := postData["text"].(string) textDataStr := postData["text"].(string)
var textData map[string]string var textData map[string]string
@ -607,19 +600,10 @@ 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)
} }
data := requestResult.(map[string]interface{})["data"].(map[string]interface{}) entry := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryJson := data["entry"].(string) request := entry["request"].(map[string]interface{})
var entry map[string]interface{} headers := request["_headers"].([]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" {
@ -632,7 +616,7 @@ func TestTapNoRedact(t *testing.T) {
} }
} }
postData := entryDetails["postData"].(map[string]interface{}) postData := request["postData"].(map[string]interface{})
textDataStr := postData["text"].(string) textDataStr := postData["text"].(string)
var textData map[string]string var textData map[string]string
@ -712,19 +696,10 @@ 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)
} }
data := requestResult.(map[string]interface{})["data"].(map[string]interface{}) entry := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryJson := data["entry"].(string) request := entry["request"].(map[string]interface{})
var entry map[string]interface{} postData := request["postData"].(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{})
postData := entryDetails["postData"].(map[string]interface{})
textData := postData["text"].(string) textData := postData["text"].(string)
if textData != "[REDACTED]" { if textData != "[REDACTED]" {
@ -809,20 +784,11 @@ func TestTapIgnoredUserAgents(t *testing.T) {
return fmt.Errorf("failed to get entry, err: %v", requestErr) return fmt.Errorf("failed to get entry, err: %v", requestErr)
} }
data := requestResult.(map[string]interface{})["data"].(map[string]interface{}) entry := requestResult.(map[string]interface{})["data"].(map[string]interface{})
entryJson := data["entry"].(string) request := entry["request"].(map[string]interface{})
var entry map[string]interface{} headers := request["_headers"].([]interface{})
if parseErr := json.Unmarshal([]byte(entryJson), &entry); parseErr != nil { for _, headerInterface := range headers {
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{})
entryHeaders := entryDetails["_headers"].([]interface{})
for _, headerInterface := range entryHeaders {
header := headerInterface.(map[string]interface{}) header := headerInterface.(map[string]interface{})
if header["name"].(string) != ignoredUserAgentCustomHeader { if header["name"].(string) != ignoredUserAgentCustomHeader {
continue continue