[chore] Move test only MockHost in a hosttest package

Signed-off-by: Bogdan Drutu <[email protected]>

Author: bogdandrutu

exporter/exporterhelper/internal/hosttest/hosttest.go

@@ -0,0 +1,19 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package hosttest // import "go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
+
+import (
+	"go.opentelemetry.io/collector/component"
+)
+
+var _ component.Host = (*MockHost)(nil)
+
+type MockHost struct {
+	component.Host
+	Ext map[component.ID]component.Component
+}
+
+func (nh *MockHost) GetExtensions() map[component.ID]component.Component {
+	return nh.Ext
+}

exporter/exporterhelper/internal/hosttest/hosttest_test.go

@@ -0,0 +1,28 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package hosttest
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/component/componenttest"
+)
+
+// nopExtension acts as an extension for testing purposes.
+type nopExtension struct {
+	component.StartFunc
+	component.ShutdownFunc
+}
+
+func TestMockHost(t *testing.T) {
+	componenttest.NewNopHost()
+	ext := map[component.ID]component.Component{
+		component.MustNewID("test"): &nopExtension{},
+	}
+	host := &MockHost{Ext: ext}
+	assert.Same(t, ext, host.Ext)
+}

exporter/exporterhelper/internal/queue_sender.go

@@ -86,12 +86,3 @@ func (qs *QueueSender) Shutdown(ctx context.Context) error {
 func (qs *QueueSender) Send(ctx context.Context, req request.Request) error {
 	return qs.queue.Offer(ctx, req)
 }
-
-type MockHost struct {
-	component.Host
-	Ext map[component.ID]component.Component
-}
-
-func (nh *MockHost) GetExtensions() map[component.ID]component.Component {
-	return nh.Ext
-}

exporter/exporterhelper/internal/queue_sender_test.go

@@ -20,6 +20,7 @@ import (
 	"go.opentelemetry.io/collector/component/componenttest"
 	"go.opentelemetry.io/collector/config/configretry"
 	"go.opentelemetry.io/collector/exporter/exporterbatcher"
+	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/queuebatch"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/request"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/requesttest"
@@ -141,7 +142,7 @@ func TestQueueBatcherPersistenceEnabled(t *testing.T) {
 	extensions := map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}
-	host := &MockHost{Ext: extensions}
+	host := &hosttest.MockHost{Ext: extensions}
 
 	// we start correctly with a file storage extension
 	require.NoError(t, be.Start(context.Background(), host))
@@ -165,7 +166,7 @@ func TestQueueBatcherPersistenceEnabledStorageError(t *testing.T) {
 	extensions := map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(storageError),
 	}
-	host := &MockHost{Ext: extensions}
+	host := &hosttest.MockHost{Ext: extensions}
 
 	// we fail to start if we get an error creating the storage client
 	require.Error(t, be.Start(context.Background(), host), "could not get storage client")
@@ -195,7 +196,7 @@ func TestQueueBatcherPersistentEnabled_NoDataLossOnShutdown(t *testing.T) {
 	extensions := map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}
-	host := &MockHost{Ext: extensions}
+	host := &hosttest.MockHost{Ext: extensions}
 
 	require.NoError(t, be.Start(context.Background(), host))
 

exporter/exporterhelper/logs_test.go

@@ -28,6 +28,7 @@ import (
 	"go.opentelemetry.io/collector/consumer/consumertest"
 	"go.opentelemetry.io/collector/exporter"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal"
+	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/metadatatest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/oteltest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/requesttest"
@@ -177,7 +178,7 @@ func TestLogs_WithPersistentQueue(t *testing.T) {
 	te, err := NewLogs(context.Background(), set, &fakeLogsConfig, ts.ConsumeLogs, WithRetry(rCfg), WithQueue(qCfg))
 	require.NoError(t, err)
 
-	host := &internal.MockHost{Ext: map[component.ID]component.Component{
+	host := &hosttest.MockHost{Ext: map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}}
 	require.NoError(t, te.Start(context.Background(), host))

exporter/exporterhelper/metrics_test.go

@@ -28,6 +28,7 @@ import (
 	"go.opentelemetry.io/collector/consumer/consumertest"
 	"go.opentelemetry.io/collector/exporter"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal"
+	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/metadatatest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/oteltest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/requesttest"
@@ -177,7 +178,7 @@ func TestMetrics_WithPersistentQueue(t *testing.T) {
 	te, err := NewMetrics(context.Background(), set, &fakeMetricsConfig, ms.ConsumeMetrics, WithRetry(rCfg), WithQueue(qCfg))
 	require.NoError(t, err)
 
-	host := &internal.MockHost{Ext: map[component.ID]component.Component{
+	host := &hosttest.MockHost{Ext: map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}}
 	require.NoError(t, te.Start(context.Background(), host))

exporter/exporterhelper/traces_test.go

@@ -28,6 +28,7 @@ import (
 	"go.opentelemetry.io/collector/consumer/consumertest"
 	"go.opentelemetry.io/collector/exporter"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal"
+	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/metadatatest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/oteltest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/requesttest"
@@ -175,7 +176,7 @@ func TestTraces_WithPersistentQueue(t *testing.T) {
 	te, err := NewTraces(context.Background(), set, &fakeTracesConfig, ts.ConsumeTraces, WithRetry(rCfg), WithQueue(qCfg))
 	require.NoError(t, err)
 
-	host := &internal.MockHost{Ext: map[component.ID]component.Component{
+	host := &hosttest.MockHost{Ext: map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}}
 	require.NoError(t, te.Start(context.Background(), host))

exporter/exporterhelper/xexporterhelper/profiles_test.go

@@ -29,6 +29,7 @@ import (
 	"go.opentelemetry.io/collector/exporter"
 	"go.opentelemetry.io/collector/exporter/exporterhelper"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal"
+	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/hosttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/oteltest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/requesttest"
 	"go.opentelemetry.io/collector/exporter/exporterhelper/internal/storagetest"
@@ -175,7 +176,7 @@ func TestProfilesExporter_WithPersistentQueue(t *testing.T) {
 	te, err := NewProfilesExporter(context.Background(), set, &fakeProfilesExporterConfig, ts.ConsumeProfiles, exporterhelper.WithRetry(rCfg), exporterhelper.WithQueue(qCfg))
 	require.NoError(t, err)
 
-	host := &internal.MockHost{Ext: map[component.ID]component.Component{
+	host := &hosttest.MockHost{Ext: map[component.ID]component.Component{
 		storageID: storagetest.NewMockStorageExtension(nil),
 	}}
 	require.NoError(t, te.Start(context.Background(), host))