From f8a1f38c0af034028e48bd364d6722b1cdf8b668 Mon Sep 17 00:00:00 2001
From: Andrey Goncharov <aigoncharov@meta.com>
Date: Tue, 4 Apr 2023 05:26:31 -0700
Subject: [PATCH] Set flipperLib implementation automatically in unit tests

Summary: This way plugins get something meaningful when they call `getFlipperLib()` in their tests

Reviewed By: LukeDefeo

Differential Revision: D44662821

fbshipit-source-id: c13800f9d8e905ae4a89955f2186adb816730b77
---
 desktop/flipper-plugin/src/test-utils/test-utils.tsx | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/desktop/flipper-plugin/src/test-utils/test-utils.tsx b/desktop/flipper-plugin/src/test-utils/test-utils.tsx
index 43aa162ad..3c589b53b 100644
--- a/desktop/flipper-plugin/src/test-utils/test-utils.tsx
+++ b/desktop/flipper-plugin/src/test-utils/test-utils.tsx
@@ -31,6 +31,7 @@ import {
   createState,
   TestUtils,
   _StartPluginOptions,
+  _setFlipperLibImplementation,
 } from 'flipper-plugin-core';
 import {SandyPluginRenderer} from '../plugin/PluginRenderer';
 import {DeviceLogEntry} from 'flipper-common';
@@ -227,6 +228,8 @@ export function startPlugin<Module extends _FlipperPluginModule<any>>(
 
   const serverAddOnControls = createServerAddOnControlsMock();
 
+  _setFlipperLibImplementation(flipperUtils);
+
   const pluginInstance = new _SandyPluginInstance(
     serverAddOnControls,
     flipperUtils,
@@ -313,6 +316,9 @@ export function startDevicePlugin<Module extends _FlipperDevicePluginModule>(
   const flipperLib = TestUtils.createMockFlipperLib(options);
   const testDevice = createMockDevice(options);
   const serverAddOnControls = createServerAddOnControlsMock();
+
+  _setFlipperLibImplementation(flipperLib);
+
   const pluginInstance = new _SandyDevicePluginInstance(
     serverAddOnControls,
     flipperLib,