From 2abf0afecd2b772ca55338ea19254e8c967ceeb7 Mon Sep 17 00:00:00 2001
From: Cesar Munoz <56847527+LikeTheSalad@users.noreply.github.com>
Date: Thu, 11 Jan 2024 14:55:51 +0100
Subject: [PATCH] Moving DefaultExportScheduler instance to
 DefaultExportScheduleHandler's constructor

---
 .../diskbuffering/scheduler/DefaultExportScheduleHandler.kt  | 5 +++--
 .../scheduler/DefaultExportScheduleHandlerTest.kt            | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

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