Added test

embrace-android-sdk/src/integrationTest/kotlin/io/embrace/android/embracesdk/features/ThermalStateTest.kt

@@ -0,0 +1,114 @@
+package io.embrace.android.embracesdk.features
+
+import android.os.Build
+import android.os.PowerManager
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import io.embrace.android.embracesdk.IntegrationTestRule
+import io.embrace.android.embracesdk.arch.schema.EmbType
+import io.embrace.android.embracesdk.config.remote.DataRemoteConfig
+import io.embrace.android.embracesdk.config.remote.RemoteConfig
+import io.embrace.android.embracesdk.fakes.fakeAutoDataCaptureBehavior
+import io.embrace.android.embracesdk.findSpanAttribute
+import io.embrace.android.embracesdk.findSpanSnapshotsOfType
+import io.embrace.android.embracesdk.findSpansOfType
+import io.embrace.android.embracesdk.internal.clock.nanosToMillis
+import io.embrace.android.embracesdk.recordSession
+import org.junit.Assert.assertEquals
+import org.junit.Before
+import org.junit.Rule
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.annotation.Config
+import org.robolectric.shadows.ShadowActivityManager
+
+@Config(sdk = [Build.VERSION_CODES.Q], shadows = [ShadowActivityManager::class])
+@RunWith(AndroidJUnit4::class)
+internal class ThermalStateFeatureTest {
+
+ @Rule
+ @JvmField
+ val testRule: IntegrationTestRule = IntegrationTestRule()
+
+ @Before
+ fun setUp() {
+ setUpThermalCapture()
+ }
+
+ @Test
+ fun `single thermal state change generates a snapshot`() {
+ with(testRule) {
+ var startTimeMs = 0L
+ val message = checkNotNull(harness.recordSession {
+ startTimeMs = harness.overriddenClock.now()
+
+ val dataSource =
+ checkNotNull(bootstrapper.dataSourceModule.thermalStateDataSource?.dataSource)
+ dataSource.handleThermalStateChange(PowerManager.THERMAL_STATUS_NONE)
+ })
+
+ val snapshots = message.findSpanSnapshotsOfType(EmbType.Performance.ThermalState)
+ assertEquals(1, snapshots.size)
+ val snapshot = snapshots.single()
+
+ assertEquals("emb-thermal-state", snapshot.name)
+ assertEquals("perf.thermal_state", snapshot.findSpanAttribute("emb.type"))
+ assertEquals(PowerManager.THERMAL_STATUS_NONE.toString(), snapshot.findSpanAttribute("status"))
+ assertEquals(startTimeMs, snapshot.startTimeNanos.nanosToMillis())
+ }
+ }
+
+ @Test
+ fun `multiple thermal state changes generate spans`() {
+ val tickTimeMs = 3000L
+ with(testRule) {
+ var startTimeMs = 0L
+ val message = checkNotNull(harness.recordSession {
+ startTimeMs = harness.overriddenClock.now()
+
+ val dataSource =
+ checkNotNull(bootstrapper.dataSourceModule.thermalStateDataSource?.dataSource)
+ dataSource.handleThermalStateChange(PowerManager.THERMAL_STATUS_CRITICAL)
+ harness.overriddenClock.tick(tickTimeMs)
+ dataSource.handleThermalStateChange(PowerManager.THERMAL_STATUS_MODERATE)
+ harness.overriddenClock.tick(tickTimeMs)
+ dataSource.handleThermalStateChange(PowerManager.THERMAL_STATUS_NONE)
+ })
+
+ val spans = message.findSpansOfType(EmbType.Performance.ThermalState)
+ assertEquals(2, spans.size)
+
+ spans.forEach {
+ assertEquals("emb-thermal-state", it.name)
+ assertEquals("perf.thermal_state", it.findSpanAttribute("emb.type"))
+ }
+ val firstSpan = spans.first()
+ assertEquals(PowerManager.THERMAL_STATUS_CRITICAL.toString(), firstSpan.findSpanAttribute("status"))
+ assertEquals(startTimeMs, firstSpan.startTimeNanos.nanosToMillis())
+ assertEquals(startTimeMs + tickTimeMs, firstSpan.endTimeNanos.nanosToMillis())
+ val secondSpan = spans.last()
+ assertEquals(PowerManager.THERMAL_STATUS_MODERATE.toString(), secondSpan.findSpanAttribute("status"))
+ assertEquals(startTimeMs + tickTimeMs, secondSpan.startTimeNanos.nanosToMillis())
+ assertEquals(startTimeMs + tickTimeMs * 2, secondSpan.endTimeNanos.nanosToMillis())
+
+ val snapshots = message.findSpanSnapshotsOfType(EmbType.Performance.ThermalState)
+ assertEquals(1, snapshots.size)
+
+ val snapshot = snapshots.single()
+ assertEquals("emb-thermal-state", snapshot.name)
+ assertEquals("perf.thermal_state", snapshot.findSpanAttribute("emb.type"))
+ assertEquals(PowerManager.THERMAL_STATUS_NONE.toString(), snapshot.findSpanAttribute("status"))
+ assertEquals(startTimeMs + tickTimeMs * 2, snapshot.startTimeNanos.nanosToMillis())
+ }
+ }
+
+ private fun setUpThermalCapture() {
+ testRule.harness.overriddenConfigService.autoDataCaptureBehavior =
+ fakeAutoDataCaptureBehavior(
+ remoteCfg = {
+ RemoteConfig(
+ dataConfig = DataRemoteConfig(pctThermalStatusEnabled = 100.0f)
+ )
+ }
+ )
+ }
+}

embrace-android-sdk/src/main/java/io/embrace/android/embracesdk/capture/thermalstate/ThermalStateDataSource.kt

@@ -78,7 +78,7 @@ internal class ThermalStateDataSource(
  }
  )
  }
- // start a new span with the new network status
+ // start a new span with the new thermal state
  captureSpanData(
  countsTowardsLimits = true,
  inputValidation = NoInputValidation