improve

DataDog · Dec 18, 2023 · 1b5649b · 1b5649b
1 parent 099a6c1
commit 1b5649b
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 6 deletions.
diff --git a/comp/logs/agent/component.go b/comp/logs/agent/component.go
@@ -50,7 +50,6 @@ type Mock interface {
 	Component
 
 	SetSources(sources *sources.LogSources)
-	AsComponent() Component
 }
 
 // Module defines the fx options for this component.

diff --git a/comp/logs/agent/mock.go b/comp/logs/agent/mock.go
@@ -91,7 +91,3 @@ func (a *mockLogsAgent) Flush(ctx context.Context) {
 func (a *mockLogsAgent) GetPipelineProvider() pipeline.Provider {
 	return nil
 }
-
-func (a *mockLogsAgent) AsComponent() Component {
-	return a
-}
diff --git a/comp/metadata/inventorychecks/inventorychecksimpl/inventorychecks_test.go b/comp/metadata/inventorychecks/inventorychecksimpl/inventorychecks_test.go
@@ -145,7 +145,7 @@ func TestGetPayload(t *testing.T) {
 
 	ic := getTestInventoryChecks(t,
 		optional.NewOption[collector.Collector](mockColl),
-		optional.NewOption[logagent.Component](logsAgent.AsComponent()),
+		optional.NewOption[logagent.Component](logsAgent),
 		overrides,
 	)