From a23596f70f9ca8a60382ac065a49cfad39c3f38c Mon Sep 17 00:00:00 2001
From: Lulu Wu <luluwu@fb.com>
Date: Mon, 27 Jul 2020 06:34:37 -0700
Subject: [PATCH] Enable registerSegment in Venice

Summary:
Enable registerSegment in Venice and verify bundle splitting works for Pokes with Venice.

Changelog: [Internal]

Reviewed By: ejanzer

Differential Revision: D22666115

fbshipit-source-id: 74ef830b802634b1019d4371873aba599438de37
---
 .../java/com/facebook/react/bridge/ReactContext.java  | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java b/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
index b74f9b4873df1d..1bbd069bce3749 100644
--- a/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
+++ b/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
@@ -462,4 +462,15 @@ public JavaScriptContextHolder getJavaScriptContextHolder() {
   public @Nullable String getSourceURL() {
     return mCatalystInstance.getSourceURL();
   }
+
+  /**
+   * Register a JS segment after loading it from cache or server, make sure mCatalystInstance is
+   * properly initialised and not null before calling.
+   *
+   * @param segmentId
+   * @param path
+   */
+  public void registerSegment(int segmentId, String path) {
+    Assertions.assertNotNull(mCatalystInstance).registerSegment(segmentId, path);
+  }
 }