chore: minor test clean up

amplitude · Mar 20, 2024 · e174ca6 · e174ca6
1 parent b0954dd
commit e174ca6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/android/src/test/java/com/amplitude/android/AmplitudeSessionTest.kt b/android/src/test/java/com/amplitude/android/AmplitudeSessionTest.kt
@@ -4,7 +4,6 @@ import android.app.Application
 import android.content.Context
 import android.net.ConnectivityManager
 import com.amplitude.android.plugins.AndroidLifecyclePlugin
-import com.amplitude.android.utilities.SystemTime
 import com.amplitude.android.utils.mockSystemTime
 import com.amplitude.common.android.AndroidContextProvider
 import com.amplitude.core.Storage
@@ -638,7 +637,7 @@ class AmplitudeSessionTest {
         val startTime: Long = 1000
         var time: Long = startTime
 
-        every { SystemTime.getCurrentTimeMillis() } returns time
+        mockSystemTime(time)
 
         val storageProvider = InstanceStorageProvider(InMemoryStorage())
         val config = createConfiguration(storageProvider)
@@ -682,7 +681,7 @@ class AmplitudeSessionTest {
         time += config.minTimeBetweenSessionsMillis + 100
 
         // Mock starting in foreground
-        every { SystemTime.getCurrentTimeMillis() } returns time
+        mockSystemTime(time)
 
         // Create a new instance to simulate recreation at startup in foreground
         val amplitude2 = Amplitude(createConfiguration(storageProvider))

diff --git a/android/src/test/java/com/amplitude/android/AmplitudeTest.kt b/android/src/test/java/com/amplitude/android/AmplitudeTest.kt
@@ -6,7 +6,7 @@ import android.net.ConnectivityManager
 import com.amplitude.analytics.connector.AnalyticsConnector
 import com.amplitude.analytics.connector.Identity
 import com.amplitude.android.plugins.AndroidLifecyclePlugin
-import com.amplitude.android.utilities.SystemTime
+import com.amplitude.android.utils.mockSystemTime
 import com.amplitude.common.android.AndroidContextProvider
 import com.amplitude.core.StorageProvider
 import com.amplitude.core.events.BaseEvent
@@ -18,7 +18,6 @@ import com.amplitude.id.IMIdentityStorageProvider
 import io.mockk.every
 import io.mockk.mockk
 import io.mockk.mockkConstructor
-import io.mockk.mockkObject
 import io.mockk.mockkStatic
 import io.mockk.slot
 import io.mockk.spyk
@@ -221,8 +220,7 @@ class AmplitudeTest {
     @Test
     fun amplitude_getSessionId_should_be_the_current_time_after_isBuilt() = runTest {
         val time = 1000L
-        mockkObject(SystemTime)
-        every { SystemTime.getCurrentTimeMillis() } returns time
+        mockSystemTime(time)
 
         amplitude = Amplitude(createConfiguration())