update polling builder to use DataDestination/StatusReporter

launchdarkly · Sep 18, 2024 · 9b60c73 · 9b60c73
1 parent b3af68a
commit 9b60c73
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/ldcomponents/polling_data_source_builder_v2_test.go b/ldcomponents/polling_data_source_builder_v2_test.go
@@ -59,9 +59,11 @@ func TestPollingDataSourceV2Builder(t *testing.T) {
 
 		p := PollingDataSourceV2()
 
-		dsu := mocks.NewMockDataSourceUpdates(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		dd := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		statusReporter := mocks.NewMockStatusReporter()
 		clientContext := makeTestContextWithBaseURIs(baseURI)
-		clientContext.BasicClientContext.DataSourceUpdateSink = dsu
+		clientContext.BasicClientContext.DataDestination = dd
+		clientContext.BasicClientContext.DataSourceStatusReporter = statusReporter
 		ds, err := p.Build(clientContext)
 		require.NoError(t, err)
 		require.NotNil(t, ds)
@@ -79,9 +81,11 @@ func TestPollingDataSourceV2Builder(t *testing.T) {
 
 		p := PollingDataSourceV2().PollInterval(interval).PayloadFilter(filter)
 
-		dsu := mocks.NewMockDataSourceUpdates(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		dd := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		statusReporter := mocks.NewMockStatusReporter()
 		clientContext := makeTestContextWithBaseURIs(baseURI)
-		clientContext.BasicClientContext.DataSourceUpdateSink = dsu
+		clientContext.BasicClientContext.DataDestination = dd
+		clientContext.BasicClientContext.DataSourceStatusReporter = statusReporter
 		ds, err := p.Build(clientContext)
 		require.NoError(t, err)
 		require.NotNil(t, ds)

diff --git a/ldcomponents/streaming_data_source_builder_v2_test.go b/ldcomponents/streaming_data_source_builder_v2_test.go
@@ -60,10 +60,10 @@ func TestStreamingDataSourceV2Builder(t *testing.T) {
 
 		s := StreamingDataSourceV2()
 
-		dsu := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		dd := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
 		statusReporter := mocks.NewMockStatusReporter()
 		clientContext := makeTestContextWithBaseURIs(baseURI)
-		clientContext.BasicClientContext.DataDestination = dsu
+		clientContext.BasicClientContext.DataDestination = dd
 		clientContext.BasicClientContext.DataSourceStatusReporter = statusReporter
 		ds, err := s.Build(clientContext)
 		require.NoError(t, err)
@@ -83,10 +83,10 @@ func TestStreamingDataSourceV2Builder(t *testing.T) {
 
 		s := StreamingDataSourceV2().InitialReconnectDelay(delay).PayloadFilter(filter)
 
-		dsu := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
+		dd := mocks.NewMockDataDestination(datastore.NewInMemoryDataStore(sharedtest.NewTestLoggers()))
 		statusReporter := mocks.NewMockStatusReporter()
 		clientContext := makeTestContextWithBaseURIs(baseURI)
-		clientContext.BasicClientContext.DataDestination = dsu
+		clientContext.BasicClientContext.DataDestination = dd
 		clientContext.BasicClientContext.DataSourceStatusReporter = statusReporter
 		ds, err := s.Build(clientContext)
 		require.NoError(t, err)