diff --git a/instrumentation/src/main/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt b/instrumentation/src/main/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt index f0a531a22..ba5820bda 100644 --- a/instrumentation/src/main/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt +++ b/instrumentation/src/main/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt @@ -9,13 +9,14 @@ import io.opentelemetry.android.internal.services.ServiceManager import io.opentelemetry.android.internal.services.periodicwork.PeriodicWorkService import java.util.concurrent.atomic.AtomicBoolean -class DefaultExportScheduleHandler : ExportScheduleHandler { +class DefaultExportScheduleHandler(private val exportScheduler: DefaultExportScheduler) : + ExportScheduleHandler { private val enabled = AtomicBoolean(false) override fun enable() { if (!enabled.getAndSet(true)) { ServiceManager.get().getService(PeriodicWorkService::class.java) - .enqueue(DefaultExportScheduler()) + .enqueue(exportScheduler) } } } diff --git a/instrumentation/src/test/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt b/instrumentation/src/test/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt index a7246c3be..b0000c54f 100644 --- a/instrumentation/src/test/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt +++ b/instrumentation/src/test/java/io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt @@ -23,7 +23,7 @@ class DefaultExportScheduleHandlerTest { @BeforeEach fun setUp() { - handler = DefaultExportScheduleHandler() + handler = DefaultExportScheduleHandler(DefaultExportScheduler()) } @Test