Refactor BaselineProfileGenerator

skydoves · Aug 26, 2023 · 40b30c6 · 40b30c6
1 parent 092340c
commit 40b30c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/benchmark/src/main/kotlin/com/skydoves/balloon/benchmark/BaselineProfileGenerator.kt b/benchmark/src/main/kotlin/com/skydoves/balloon/benchmark/BaselineProfileGenerator.kt
@@ -16,18 +16,22 @@
 
 package com.skydoves.balloon.benchmark
 
+import android.os.Build
+import androidx.annotation.RequiresApi
 import androidx.benchmark.macro.junit4.BaselineProfileRule
 import org.junit.Rule
 import org.junit.Test
 
+@RequiresApi(Build.VERSION_CODES.P)
 class BaselineProfileGenerator {
  @get:Rule
  val baselineProfileRule = BaselineProfileRule()
 
  @Test
  fun startup() =
- baselineProfileRule.collectBaselineProfile(
+ baselineProfileRule.collect(
  packageName = "com.skydoves.balloon.benchmark.app",
+ includeInStartupProfile = true
  ) {
  pressHome()
  // This block defines the app's critical user journey. Here we are interested in