inject factory into test

wealthfront · Oct 27, 2023 · 4847704 · 4847704
1 parent 9cd3f7e
commit 4847704
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/...ple-migration/src/test/java/com/wealthfront/magellan/sample/migration/TestAppComponent.kt b/...ple-migration/src/test/java/com/wealthfront/magellan/sample/migration/TestAppComponent.kt
@@ -1,7 +1,6 @@
 package com.wealthfront.magellan.sample.migration
 
 import com.wealthfront.magellan.sample.migration.api.DogApi
-import com.wealthfront.magellan.sample.migration.tide.DogDetailsScreenTest
 import com.wealthfront.magellan.sample.migration.tide.DogListStepTest
 import com.wealthfront.magellan.sample.migration.toolbar.ToolbarHelper
 import dagger.Component
@@ -14,6 +13,5 @@ interface TestAppComponent : AppComponent {
   val toolbarHelper: ToolbarHelper
   val api: DogApi
 
-  fun inject(test: DogDetailsScreenTest)
   fun inject(test: DogListStepTest)
 }
diff --git a/...migration/src/test/java/com/wealthfront/magellan/sample/migration/tide/DogListStepTest.kt b/...migration/src/test/java/com/wealthfront/magellan/sample/migration/tide/DogListStepTest.kt
@@ -16,19 +16,23 @@ import org.junit.runner.RunWith
 import org.robolectric.Robolectric
 import org.robolectric.RobolectricTestRunner
 import org.robolectric.Shadows.shadowOf
+import javax.inject.Inject
 
 @RunWith(RobolectricTestRunner::class)
 class DogListStepTest {
 
-  private var chosenBreed: String? = null
   private lateinit var dogListStep: DogListStep
+  @Inject lateinit var dogListStepFactory: DogListStepFactory
   private val activityController = Robolectric.buildActivity(ComponentActivity::class.java)
 
+  private var chosenBreed: String? = null
+
   @Before
   fun setUp() {
     val context = ApplicationProvider.getApplicationContext<Application>()
     val component = ((context as AppComponentContainer).injector() as TestAppComponent)
-    dogListStep = DogListStep(component.toolbarHelper, component.api) { chosenBreed = it }
+    component.inject(this)
+    dogListStep = dogListStepFactory.create { chosenBreed = it }
     coWhen { component.api.getAllBreeds() }
       .thenReturn(DogBreedsResponse(message = mapOf("akita" to emptyList()), status = "success"))
   }