From 054fb4ab92f659574d6b0dc3b632ad772f2ee46c Mon Sep 17 00:00:00 2001
From: Abhishek Jha <abhishek.jha@uber.com>
Date: Mon, 13 May 2024 09:50:37 -0700
Subject: [PATCH] Addition of test for context_util

---
 .../history/execution/context_util_test.go    | 74 ++++++++++++++++++-
 1 file changed, 71 insertions(+), 3 deletions(-)

diff --git a/service/history/execution/context_util_test.go b/service/history/execution/context_util_test.go
index aa2ac0ef136..de1d1173bcd 100644
--- a/service/history/execution/context_util_test.go
+++ b/service/history/execution/context_util_test.go
@@ -21,18 +21,17 @@
 package execution
 
 import (
-	"testing"
-
 	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/mock"
-
 	"github.com/uber/cadence/common/cache"
 	"github.com/uber/cadence/common/dynamicconfig"
 	"github.com/uber/cadence/common/log"
 	"github.com/uber/cadence/common/metrics"
+	"github.com/uber/cadence/common/metrics/mocks"
 	"github.com/uber/cadence/common/persistence"
 	"github.com/uber/cadence/service/history/config"
 	"github.com/uber/cadence/service/history/shard"
+	"testing"
 )
 
 func createTestConfig() *config.Config {
@@ -127,3 +126,72 @@ func TestEmitLargeWorkflowShardIDStats(t *testing.T) {
 		})
 	}
 }
+
+func TestEmitWorkflowHistoryStats(t *testing.T) {
+
+	mockMetricsClient := new(mocks.Client)
+	mockScope := new(mocks.Scope)
+
+	mockMetricsClient.On("Scope", mock.Anything, mock.Anything).Return(mockScope)
+
+	mockScope.On("RecordTimer", mock.Anything, mock.Anything).Return()
+
+	domainName := "testDomain"
+	historySize := 2048
+	historyCount := 150
+
+	emitWorkflowHistoryStats(mockMetricsClient, domainName, historySize, historyCount)
+
+	mockMetricsClient.AssertExpectations(t)
+	mockScope.AssertExpectations(t)
+}
+
+func TestEmitWorkflowExecutionStats(t *testing.T) {
+	tests := []struct {
+		name        string
+		domainName  string
+		stats       *persistence.MutableStateStats
+		historySize int64
+		expectCalls bool
+	}{
+		{
+			name:       "With valid stats",
+			domainName: "testDomain",
+			stats: &persistence.MutableStateStats{
+				MutableStateSize: 1024,
+				ActivityInfoSize: 256,
+				TimerInfoSize:    128,
+			},
+			historySize: 2048,
+			expectCalls: true,
+		},
+		{
+			name:        "Nil stats",
+			domainName:  "testDomain",
+			stats:       nil,
+			historySize: 2048,
+			expectCalls: false,
+		},
+	}
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+
+			mockMetricsClient := new(mocks.Client)
+			mockScope := new(mocks.Scope)
+
+			if tc.expectCalls {
+				mockMetricsClient.On("Scope", metrics.ExecutionSizeStatsScope, mock.Anything).Return(mockScope)
+				mockMetricsClient.On("Scope", metrics.ExecutionCountStatsScope, mock.Anything).Return(mockScope)
+				mockScope.On("RecordTimer", mock.AnythingOfType("int"), mock.AnythingOfType("time.Duration")).Return().Times(14)
+			} else {
+				mockScope.AssertNotCalled(t, "RecordTimer")
+			}
+
+			emitWorkflowExecutionStats(mockMetricsClient, tc.domainName, tc.stats, tc.historySize)
+
+			mockMetricsClient.AssertExpectations(t)
+			mockScope.AssertExpectations(t)
+		})
+	}
+}