factor out historyReplicatorImpl for future use

cadence-workflow · May 8, 2024 · 1d0c6a3 · 1d0c6a3
1 parent b4a574a
commit 1d0c6a3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/service/history/ndc/history_replicator_test.go b/service/history/ndc/history_replicator_test.go
@@ -41,7 +41,7 @@ import (
 	"github.com/uber/cadence/service/history/shard"
 )
 
-func createTestHistoryReplicator(t *testing.T) {
+func createTestHistoryReplicator(t *testing.T) historyReplicatorImpl {
 	ctrl := gomock.NewController(t)
 
 	mockShard := shard.NewMockContext(ctrl)
@@ -81,11 +81,16 @@ func createTestHistoryReplicator(t *testing.T) {
 	// going back to NewHistoryReplicator
 	mockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(1)
 
-	assert.NotNil(t, NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop()))
+	replicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())
+	replicatorImpl := replicator.(*historyReplicatorImpl)
+	return *replicatorImpl
+}
+
+type name struct {
 }
 
 func TestNewHistoryReplicator(t *testing.T) {
-	createTestHistoryReplicator(t)
+	assert.NotNil(t, createTestHistoryReplicator(t))
 }
 
 func TestNewHistoryReplicator_newBranchManager(t *testing.T) {