commit 023317c2746ee5d19d64799b7a7c7f045ae7e02e
author Ryan Bonick <rbonick@google.com> Tue Apr 16 17:20:07 2019
committer Copybara-Service <copybara-worker@google.com> Tue Apr 16 17:20:42 2019
tree 829c360ba15a0293be5b9a4402e8504103e09777
parent cabf690e1eaa94474c57cf75a3f0186a0c5cc15e

Add ability to set UiContext in FakeModelProvider#onSessionStart()

PiperOrigin-RevId: 243831318
Change-Id: I1205da45bab9e7d17fb13cb6771d1f8a374cbafa

src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProvider.java

diff --git a/src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProvider.java b/src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProvider.java
index 8fac5d3..b410338 100644
--- a/src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProvider.java
+++ b/src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProvider.java
@@ -150,9 +150,13 @@
   }
 
   public void triggerOnSessionStart(ModelFeature rootFeature) {
+    triggerOnSessionStart(rootFeature, UiContext.getDefaultInstance());
+  }
+
+  public void triggerOnSessionStart(ModelFeature rootFeature, UiContext uiContext) {
     this.rootFeature = rootFeature;
     for (ModelProviderObserver observer : observers) {
-      observer.onSessionStart(UiContext.getDefaultInstance());
+      observer.onSessionStart(uiContext);
     }
   }