Fix tests

Tunous · Aug 18, 2020 · 9d13eca · 9d13eca
1 parent 8ed87a9
commit 9d13eca
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/app/src/main/java/me/saket/dank/ui/submission/SubmissionCommentTreeUiConstructor.java b/app/src/main/java/me/saket/dank/ui/submission/SubmissionCommentTreeUiConstructor.java
@@ -172,13 +172,13 @@ public SubmissionCommentTreeUiConstructor(
       Lazy<VotingManager> votingManager,
       Lazy<Markdown> markdown,
       Lazy<UserSessionRepository> userSessionRepository,
-      @Named("show_color_replication_icons") Preference<Boolean> showColorReplicationIcons)
+      @Named("show_color_replication_icons") Lazy<Preference<Boolean>> showColorReplicationIcons)
   {
     this.replyRepository = replyRepository;
     this.votingManager = votingManager;
     this.markdown = markdown;
     this.userSessionRepository = userSessionRepository;
-    this.showColorReplicationIcons = showColorReplicationIcons;
+    this.showColorReplicationIcons = showColorReplicationIcons.get();
   }
 
   @CheckResult

diff --git a/app/src/test/java/me/saket/dank/ui/comments/SubmissionCommentTreeBaseTest.kt b/app/src/test/java/me/saket/dank/ui/comments/SubmissionCommentTreeBaseTest.kt
@@ -1,6 +1,7 @@
 package me.saket.dank.ui.comments
 
 import androidx.test.core.app.ApplicationProvider
+import com.f2prateek.rx.preferences2.Preference
 import com.nhaarman.mockitokotlin2.any
 import com.nhaarman.mockitokotlin2.doAnswer
 import com.nhaarman.mockitokotlin2.doReturn
@@ -51,11 +52,16 @@ abstract class SubmissionCommentTreeBaseTest {
         it.getArgument<Comment>(0).body
       }
     }
+    val prefIcons = mock<Preference<Boolean>> {
+      on { asObservable() } doReturn Observable.empty()
+      on { get() } doReturn false
+    }
     uiConstructor = SubmissionCommentTreeUiConstructor(
       { replyRepository },
       { votingManager },
       { markdown },
-      { mock() }
+      { mock() },
+      { prefIcons }
     )
     SubmissionCommentTreeUiConstructor.COLLAPSED_COMMENT_IDS.clear()
   }