added integration tests

rudderlabs · Jan 12, 2024 · d5145af · d5145af
1 parent bf5f703
commit d5145af
Showing 1 changed file with 138 additions and 3 deletions.
diff --git a/integration_test/docker_test/docker_test.go b/integration_test/docker_test/docker_test.go
@@ -98,7 +98,7 @@ func TestMainFlow(t *testing.T) {
 	sendEventsToGateway(t)
 	t.Run("webhook", func(t *testing.T) {
 		require.Eventually(t, func() bool {
-			return webhook.RequestsCount() == 10
+			return webhook.RequestsCount() == 11
 		}, time.Minute, 300*time.Millisecond)
 
 		i := -1
@@ -224,7 +224,7 @@ func TestMainFlow(t *testing.T) {
 
 		var (
 			msgCount      = 0 // Count how many message processed
-			expectedCount = 10
+			expectedCount = 15
 			timeout       = time.After(2 * time.Minute)
 		)
 
@@ -532,6 +532,125 @@ func sendEventsToGateway(t *testing.T) {
 	}`)
 	sendEvent(t, payloadGroup, "group", writeKey)
 	sendPixelEvents(t, writeKey)
+
+	payloadRetlWebhook := strings.NewReader(`{
+		"batch":
+		[
+			{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			}
+		]
+	}`)
+	sendEvent(t, payloadRetlWebhook, "retl", writeKey, withHeader("X-Rudder-Source-Id", "xxxyyyzzEaEurW247ad9WYZLUyk"), withHeader("X-Rudder-Destination-Id", "xxxyyyzzP9kQfzOoKd1tuxchYAG"), withInternalEndpoint())
+
+	payloadRetlKafka := strings.NewReader(`{
+		"batch":
+		[
+			{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"messageId":"messageId_11",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			},{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			},{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			},{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			},{
+				"userId": "identified_user_id",
+				"anonymousId": "anonymousId_1",
+				"type": "identify",
+				"context":
+				{
+					"traits":
+					{
+						"trait1": "new-val"
+					},
+					"ip": "14.5.67.21",
+					"library":
+					{
+						"name": "http"
+					}
+				},
+				"timestamp": "2020-02-02T00:23:09.544Z"
+			}
+		]
+	}`)
+	sendEvent(t, payloadRetlKafka, "retl", writeKey, withHeader("X-Rudder-Source-Id", "xxxyyyzzEaEurW247ad9WYZLUyk"), withHeader("X-Rudder-Destination-Id", "xxxyyyzzhyrw8v0CrTMrDZ4ovej"), withInternalEndpoint())
 }
 
 func blockOnHold(t *testing.T) {
@@ -597,7 +716,19 @@ func sendPixelEvents(t *testing.T, writeKey string) {
 	}
 }
 
-func sendEvent(t *testing.T, payload *strings.Reader, callType, writeKey string) {
+func withHeader(key, value string) func(r *http.Request) {
+	return func(req *http.Request) {
+		req.Header.Add(key, value)
+	}
+}
+
+func withInternalEndpoint() func(r *http.Request) {
+	return func(req *http.Request) {
+		req.URL.Path = fmt.Sprintf("/internal%s", req.URL.Path)
+	}
+}
+
+func sendEvent(t *testing.T, payload *strings.Reader, callType, writeKey string, reqOptions ...func(r *http.Request)) {
 	t.Helper()
 	t.Logf("Sending %s Event", callType)
 
@@ -618,6 +749,10 @@ func sendEvent(t *testing.T, payload *strings.Reader, callType, writeKey string)
 		[]byte(fmt.Sprintf("%s:", writeKey)),
 	)))
 
+	for _, reqOption := range reqOptions {
+		reqOption(req)
+	}
+
 	res, err := httpClient.Do(req)
 	if err != nil {
 		t.Logf("sendEvent error: %v", err)