From 72f6d2e4f743f022ec4502808ce61b8e9d2e45ea Mon Sep 17 00:00:00 2001
From: Sad Ellie <openthiswhitedoor@gmail.com>
Date: Tue, 7 Jun 2022 20:43:49 +0300
Subject: [PATCH] Fixed test after adding DI

---
 .../java/com/sadellie/unitto/screen/SwapUnitsTest.kt           | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt b/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
index 6eac2314..52c429e6 100644
--- a/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
+++ b/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
@@ -5,6 +5,7 @@ import android.content.Context
 import androidx.room.Room
 import androidx.test.core.app.ApplicationProvider
 import androidx.test.ext.junit.runners.AndroidJUnit4
+import com.sadellie.unitto.data.preferences.DataStoreModule
 import com.sadellie.unitto.data.preferences.UserPreferencesRepository
 import com.sadellie.unitto.data.units.ALL_UNITS
 import com.sadellie.unitto.data.units.MyUnitIDS
@@ -26,7 +27,7 @@ class SwapUnitsTest {
     fun setUp() {
         val context = ApplicationProvider.getApplicationContext<Context>()
         viewModel = MainViewModel(
-            UserPreferencesRepository(context),
+            UserPreferencesRepository(DataStoreModule().provideUserPreferencesDataStore(context)),
             MyBasedUnitsRepository(
                 Room.inMemoryDatabaseBuilder(
                     context,