Refactor some test code that uses TestCompositionLocalProvider

Author: tobyhs

app/src/test/java/io/github/tobyhs/weatherweight/forecast/ForecastScreenTest.kt

@@ -14,7 +14,7 @@ import androidx.test.ext.junit.runners.AndroidJUnit4
 import io.github.tobyhs.weatherweight.R
 import io.github.tobyhs.weatherweight.data.model.ForecastResultSet
 import io.github.tobyhs.weatherweight.test.ForecastResultSetFactory
-import io.github.tobyhs.weatherweight.test.TestCompositionLocalProvider
+import io.github.tobyhs.weatherweight.test.setContentWithTestProvider
 import io.github.tobyhs.weatherweight.ui.LoadState
 
 import io.mockk.every
@@ -116,10 +116,8 @@ class ForecastScreenTest {
     }
 
     private fun setContent() {
-        composeRule.setContent {
-            TestCompositionLocalProvider {
-                ForecastScreen(viewModel)
-            }
+        setContentWithTestProvider(composeRule) {
+            ForecastScreen(viewModel)
         }
     }
 

app/src/test/java/io/github/tobyhs/weatherweight/forecast/HourlyForecastCardTest.kt

@@ -7,7 +7,7 @@ import androidx.compose.ui.test.onAllNodesWithTag
 import androidx.test.ext.junit.runners.AndroidJUnit4
 
 import io.github.tobyhs.weatherweight.test.ForecastResultSetFactory
-import io.github.tobyhs.weatherweight.test.TestCompositionLocalProvider
+import io.github.tobyhs.weatherweight.test.setContentWithTestProvider
 
 import org.junit.Rule
 import org.junit.Test
@@ -19,10 +19,8 @@ class HourlyForecastCardTest {
 
     @Test
     fun compose() {
-        composeRule.setContent {
-            TestCompositionLocalProvider {
-                HourlyForecastCard(ForecastResultSetFactory.createHourlyForecasts()[0])
-            }
+        setContentWithTestProvider(composeRule) {
+            HourlyForecastCard(ForecastResultSetFactory.createHourlyForecasts()[0])
         }
         checkContent(composeRule)
     }

app/src/test/java/io/github/tobyhs/weatherweight/test/TestCompositionLocalProvider.kt

@@ -2,6 +2,7 @@ package io.github.tobyhs.weatherweight.test
 
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.CompositionLocalProvider
+import androidx.compose.ui.test.junit4.ComposeContentTestRule
 
 import io.github.tobyhs.weatherweight.ui.LocalClock
 
@@ -13,3 +14,7 @@ fun TestCompositionLocalProvider(content: @Composable () -> Unit) {
     val clock = Clock.system(ZoneId.of("America/Los_Angeles"))
     CompositionLocalProvider(LocalClock provides clock, content)
 }
+
+fun setContentWithTestProvider(composeRule: ComposeContentTestRule, content: @Composable () -> Unit) {
+    composeRule.setContent { TestCompositionLocalProvider(content) }
+}