Add ability to set UiContext in FakeModelProvider#onSessionStart()

PiperOrigin-RevId: 243831318
Change-Id: I1205da45bab9e7d17fb13cb6771d1f8a374cbafa
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);
     }
   }