github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-07-31 08:20:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Exported test consts (#870)

Browse Source
This commit is contained in:
RoyUP9 2022-03-02 10:02:21 +02:00 committed by GitHub
parent 346e904e77
commit b9f5475e3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 47 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
acceptanceTests/extensions_test.go
View File

@ -40,7 +40,7 @@ func TestRedis(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -55,7 +55,7 @@ func TestRedis(t *testing.T) {
return return
} }
redisExternalIp, err := kubernetesProvider.GetServiceExternalIp(ctx, defaultNamespaceName, "redis") redisExternalIp, err := kubernetesProvider.GetServiceExternalIp(ctx, DefaultNamespaceName, "redis")
if err != nil { if err != nil {
t.Errorf("failed to get redis external ip, err: %v", err) t.Errorf("failed to get redis external ip, err: %v", err)
return return
@ -65,7 +65,7 @@ func TestRedis(t *testing.T) {
Addr: fmt.Sprintf("%v:6379", redisExternalIp), Addr: fmt.Sprintf("%v:6379", redisExternalIp),
}) })
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
requestErr := rdb.Ping(ctx).Err() requestErr := rdb.Ping(ctx).Err()
if requestErr != nil { if requestErr != nil {
t.Errorf("failed to send redis request, err: %v", requestErr) t.Errorf("failed to send redis request, err: %v", requestErr)
@ -73,7 +73,7 @@ func TestRedis(t *testing.T) {
} }
} }
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
requestErr := rdb.Set(ctx, "key", "value", -1).Err() requestErr := rdb.Set(ctx, "key", "value", -1).Err()
if requestErr != nil { if requestErr != nil {
t.Errorf("failed to send redis request, err: %v", requestErr) t.Errorf("failed to send redis request, err: %v", requestErr)
@ -81,7 +81,7 @@ func TestRedis(t *testing.T) {
} }
} }
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
requestErr := rdb.Exists(ctx, "key").Err() requestErr := rdb.Exists(ctx, "key").Err()
if requestErr != nil { if requestErr != nil {
t.Errorf("failed to send redis request, err: %v", requestErr) t.Errorf("failed to send redis request, err: %v", requestErr)
@ -89,7 +89,7 @@ func TestRedis(t *testing.T) {
} }
} }
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
requestErr := rdb.Get(ctx, "key").Err() requestErr := rdb.Get(ctx, "key").Err()
if requestErr != nil { if requestErr != nil {
t.Errorf("failed to send redis request, err: %v", requestErr) t.Errorf("failed to send redis request, err: %v", requestErr)
@ -97,7 +97,7 @@ func TestRedis(t *testing.T) {
} }
} }
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
requestErr := rdb.Del(ctx, "key").Err() requestErr := rdb.Del(ctx, "key").Err()
if requestErr != nil { if requestErr != nil {
t.Errorf("failed to send redis request, err: %v", requestErr) t.Errorf("failed to send redis request, err: %v", requestErr)
@ -138,7 +138,7 @@ func TestAmqp(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -153,7 +153,7 @@ func TestAmqp(t *testing.T) {
return return
} }
rabbitmqExternalIp, err := kubernetesProvider.GetServiceExternalIp(ctx, defaultNamespaceName, "rabbitmq") rabbitmqExternalIp, err := kubernetesProvider.GetServiceExternalIp(ctx, DefaultNamespaceName, "rabbitmq")
if err != nil { if err != nil {
t.Errorf("failed to get RabbitMQ external ip, err: %v", err) t.Errorf("failed to get RabbitMQ external ip, err: %v", err)
return return
@ -169,7 +169,7 @@ func TestAmqp(t *testing.T) {
// Temporary fix for missing amqp entries // Temporary fix for missing amqp entries
time.Sleep(10 * time.Second) time.Sleep(10 * time.Second)
for i := 0; i < defaultEntriesCount/5; i++ { for i := 0; i < DefaultEntriesCount/5; i++ {
ch, err := conn.Channel() ch, err := conn.Channel()
if err != nil { if err != nil {
t.Errorf("failed to open a channel, err: %v", err) t.Errorf("failed to open a channel, err: %v", err)

4
acceptanceTests/logs_test.go
View File

@ -36,7 +36,7 @@ func TestLogs(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -137,7 +137,7 @@ func TestLogsPath(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)

46
acceptanceTests/tap_test.go
View File

@ -53,14 +53,14 @@ func basicTapTest(t *testing.T, shouldCheckSrcAndDest bool, extraArgs... string)
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
for i := 0; i < entriesCount; i++ { for i := 0; i < entriesCount; i++ {
if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil { if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
@ -127,8 +127,8 @@ func TestTapGuiPort(t *testing.T) {
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < DefaultEntriesCount; i++ {
if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil { if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
return return
@ -175,7 +175,7 @@ func TestTapAllNamespaces(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -224,7 +224,7 @@ func TestTapMultipleNamespaces(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -270,7 +270,7 @@ func TestTapRegex(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -318,7 +318,7 @@ func TestTapDryRun(t *testing.T) {
}() }()
go func() { go func() {
time.Sleep(shortRetriesCount * time.Second) time.Sleep(ShortRetriesCount * time.Second)
resultChannel <- "fail" resultChannel <- "fail"
}() }()
@ -358,17 +358,17 @@ func TestTapRedact(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
requestHeaders := map[string]string{"User-Header": "Mizu"} 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 := ExecuteHttpPostRequestWithHeaders(fmt.Sprintf("%v/post", proxyUrl), requestHeaders, 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
@ -410,17 +410,17 @@ func TestTapNoRedact(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
requestHeaders := map[string]string{"User-Header": "Mizu"} 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 := ExecuteHttpPostRequestWithHeaders(fmt.Sprintf("%v/post", proxyUrl), requestHeaders, 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
@ -462,15 +462,15 @@ func TestTapRegexMasking(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < DefaultEntriesCount; i++ {
response, requestErr := http.Post(fmt.Sprintf("%v/post", proxyUrl), "text/plain", bytes.NewBufferString("Mizu")) response, requestErr := http.Post(fmt.Sprintf("%v/post", proxyUrl), "text/plain", bytes.NewBufferString("Mizu"))
if _, requestErr = ExecuteHttpRequest(response, requestErr); requestErr != nil { if _, requestErr = ExecuteHttpRequest(response, requestErr); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
@ -515,25 +515,25 @@ func TestTapIgnoredUserAgents(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
return return
} }
proxyUrl := GetProxyUrl(defaultNamespaceName, defaultServiceName) proxyUrl := GetProxyUrl(DefaultNamespaceName, DefaultServiceName)
ignoredUserAgentCustomHeader := "Ignored-User-Agent" ignoredUserAgentCustomHeader := "Ignored-User-Agent"
headers := map[string]string{"User-Agent": ignoredUserAgentValue, ignoredUserAgentCustomHeader: ""} headers := map[string]string{"User-Agent": ignoredUserAgentValue, ignoredUserAgentCustomHeader: ""}
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < DefaultEntriesCount; i++ {
if _, requestErr := ExecuteHttpGetRequestWithHeaders(fmt.Sprintf("%v/get", proxyUrl), headers); requestErr != nil { if _, requestErr := ExecuteHttpGetRequestWithHeaders(fmt.Sprintf("%v/get", proxyUrl), headers); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
return return
} }
} }
for i := 0; i < defaultEntriesCount; i++ { for i := 0; i < DefaultEntriesCount; i++ {
if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil { if _, requestErr := ExecuteHttpGetRequest(fmt.Sprintf("%v/get", proxyUrl)); requestErr != nil {
t.Errorf("failed to send proxy request, err: %v", requestErr) t.Errorf("failed to send proxy request, err: %v", requestErr)
return return
@ -569,7 +569,7 @@ func TestTapDumpLogs(t *testing.T) {
return return
} }
apiServerUrl := GetApiServerUrl(defaultApiServerPort) apiServerUrl := GetApiServerUrl(DefaultApiServerPort)
if err := WaitTapPodsReady(apiServerUrl); err != nil { if err := WaitTapPodsReady(apiServerUrl); err != nil {
t.Errorf("failed to start tap pods on time, err: %v", err) t.Errorf("failed to start tap pods on time, err: %v", err)
@ -651,7 +651,7 @@ func TestTapDumpLogs(t *testing.T) {
} }
func TestIpResolving(t *testing.T) { func TestIpResolving(t *testing.T) {
namespace := allNamespaces namespace := AllNamespaces
t.Log("add permissions for ip-resolution for current user") t.Log("add permissions for ip-resolution for current user")
if err := ApplyKubeFilesForTest( if err := ApplyKubeFilesForTest(

24
acceptanceTests/testsUtils.go
View File

@ -24,14 +24,14 @@ import (
) )
const ( const (
longRetriesCount = 100 LongRetriesCount = 100
shortRetriesCount = 10 ShortRetriesCount = 10
defaultApiServerPort = shared.DefaultApiServerPort DefaultApiServerPort = shared.DefaultApiServerPort
defaultNamespaceName = "mizu-tests" DefaultNamespaceName = "mizu-tests"
defaultServiceName = "httpbin" DefaultServiceName = "httpbin"
defaultEntriesCount = 50 DefaultEntriesCount = 50
waitAfterTapPodsReady = 3 * time.Second WaitAfterTapPodsReady = 3 * time.Second
allNamespaces = "" AllNamespaces = ""
) )
type PodDescriptor struct { type PodDescriptor struct {
@ -181,7 +181,7 @@ func ApplyKubeFile(kubeContext string, namespace string, filename string) (error
"--context", kubeContext, "--context", kubeContext,
"-f", filename, "-f", filename,
} }
if namespace != allNamespaces { if namespace != AllNamespaces {
cmdArgs = append(cmdArgs, "-n", namespace) cmdArgs = append(cmdArgs, "-n", namespace)
} }
cmd := exec.Command("kubectl", cmdArgs...) cmd := exec.Command("kubectl", cmdArgs...)
@ -199,7 +199,7 @@ func DeleteKubeFile(kubeContext string, namespace string, filename string) error
"--context", kubeContext, "--context", kubeContext,
"-f", filename, "-f", filename,
} }
if namespace != allNamespaces { if namespace != AllNamespaces {
cmdArgs = append(cmdArgs, "-n", namespace) cmdArgs = append(cmdArgs, "-n", namespace)
} }
cmd := exec.Command("kubectl", cmdArgs...) cmd := exec.Command("kubectl", cmdArgs...)
@ -304,11 +304,11 @@ func WaitTapPodsReady(apiServerUrl string) error {
if connectedTappersCount == 0 { if connectedTappersCount == 0 {
return fmt.Errorf("no connected tappers running") return fmt.Errorf("no connected tappers running")
} }
time.Sleep(waitAfterTapPodsReady) time.Sleep(WaitAfterTapPodsReady)
return nil return nil
} }
return retriesExecute(longRetriesCount, tapPodsReadyFunc) return retriesExecute(LongRetriesCount, tapPodsReadyFunc)
} }
func jsonBytesToInterface(jsonBytes []byte) (interface{}, error) { func jsonBytesToInterface(jsonBytes []byte) (interface{}, error) {