From 0a98fdcf9a1b8ec815e4aefe64fe41f4eb0ba5b3 Mon Sep 17 00:00:00 2001
From: Steve Zhang <huailong.zhang@intel.com>
Date: Wed, 4 Sep 2024 18:55:17 +0800
Subject: [PATCH] add unit test for new function. (#390)

Signed-off-by: zhlsunshine <huailong.zhang@intel.com>
---
 .../cmd/router/main_test.go                   | 74 +++++++++++++++++++
 1 file changed, 74 insertions(+)

diff --git a/microservices-connector/cmd/router/main_test.go b/microservices-connector/cmd/router/main_test.go
index 6f2d7e54b..6b8d3f88c 100644
--- a/microservices-connector/cmd/router/main_test.go
+++ b/microservices-connector/cmd/router/main_test.go
@@ -16,6 +16,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"os"
+	"reflect"
 	"strings"
 	"testing"
 	"time"
@@ -934,3 +935,76 @@ func TestMcDataHandler(t *testing.T) {
 			rr.Body.String(), expected)
 	}
 }
+
+func TestMergeRequests(t *testing.T) {
+	// Define the test cases
+	tests := []struct {
+		name        string
+		respReq     []byte
+		initReqData map[string]interface{}
+		expected    []byte
+		expectErr   bool
+	}{
+		{
+			name:    "Merge with Parameters",
+			respReq: []byte(`{"key1":"value1","key2":"value2"}`),
+			initReqData: map[string]interface{}{
+				"parameters": map[string]interface{}{
+					"key2": "newValue2",
+					"key3": "value3",
+				},
+			},
+			expected:  []byte(`{"key1":"value1","key2":"newValue2","key3":"value3"}`),
+			expectErr: false,
+		},
+		{
+			name:        "No Parameters in initReqData",
+			respReq:     []byte(`{"key1":"value1"}`),
+			initReqData: map[string]interface{}{},
+			expected:    []byte(`{"key1":"value1"}`),
+			expectErr:   false,
+		},
+		{
+			name: "Invalid JSON in respReq",
+			// Invalid JSON
+			respReq: []byte(`{"key1":value1}`),
+			initReqData: map[string]interface{}{
+				"parameters": map[string]interface{}{
+					"key2": "value2",
+				},
+			},
+			expected:  nil,
+			expectErr: true,
+		},
+	}
+
+	// Run the test cases
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := mergeRequests(tt.respReq, tt.initReqData)
+
+			if tt.expectErr {
+				if result != nil {
+					t.Errorf("Expected error, but got result: %s", string(result))
+				}
+				return
+			}
+
+			// Unmarshal the JSON result to compare
+			var resultMap map[string]interface{}
+			if err := json.Unmarshal(result, &resultMap); err != nil {
+				t.Fatalf("Failed to unmarshal result: %v", err)
+			}
+
+			var expectedMap map[string]interface{}
+			if err := json.Unmarshal(tt.expected, &expectedMap); err != nil {
+				t.Fatalf("Failed to unmarshal expected result: %v", err)
+			}
+
+			// Compare the actual result with the expected result
+			if !reflect.DeepEqual(resultMap, expectedMap) {
+				t.Errorf("Test %s failed: expected %v, got %v", tt.name, expectedMap, resultMap)
+			}
+		})
+	}
+}