test: 💍 fix Jest test after refactor

Author: streamich

src/plugins/embeddable/public/index.ts

@@ -68,4 +68,9 @@ export function plugin(initializerContext: PluginInitializerContext) {
   return new EmbeddablePublicPlugin(initializerContext);
 }
 
-export { EmbeddableSetup, EmbeddableStart } from './plugin';
+export {
+  EmbeddableSetup,
+  EmbeddableStart,
+  EmbeddableSetupDependencies,
+  EmbeddableStartDependencies,
+} from './plugin';

src/plugins/embeddable/public/mocks.ts

@@ -16,7 +16,12 @@
  * specific language governing permissions and limitations
  * under the License.
  */
-import { EmbeddableStart, EmbeddableSetup } from '.';
+import {
+  EmbeddableStart,
+  EmbeddableSetup,
+  EmbeddableSetupDependencies,
+  EmbeddableStartDependencies,
+} from '.';
 import { EmbeddablePublicPlugin } from './plugin';
 import { coreMock } from '../../../core/public/mocks';
 
@@ -45,14 +50,14 @@ const createStartContract = (): Start => {
   return startContract;
 };
 
-const createInstance = () => {
+const createInstance = (setupPlugins: Partial<EmbeddableSetupDependencies> = {}) => {
   const plugin = new EmbeddablePublicPlugin({} as any);
   const setup = plugin.setup(coreMock.createSetup(), {
-    uiActions: uiActionsPluginMock.createSetupContract(),
+    uiActions: setupPlugins.uiActions || uiActionsPluginMock.createSetupContract(),
   });
-  const doStart = () =>
+  const doStart = (startPlugins: Partial<EmbeddableStartDependencies> = {}) =>
     plugin.start(coreMock.createStart(), {
-      uiActions: uiActionsPluginMock.createStartContract(),
+      uiActions: startPlugins.uiActions || uiActionsPluginMock.createStartContract(),
       inspector: inspectorPluginMock.createStartContract(),
     });
   return {

x-pack/plugins/advanced_ui_actions/public/mocks.ts

@@ -38,7 +38,9 @@ const createPlugin = (
 ) => {
   const pluginInitializerContext = coreMock.createPluginInitializerContext();
   const uiActions = uiActionsPluginMock.createPlugin();
-  const embeddable = embeddablePluginMock.createInstance();
+  const embeddable = embeddablePluginMock.createInstance({
+    uiActions: uiActions.setup,
+  });
   const plugin = pluginInitializer(pluginInitializerContext);
   const setup = plugin.setup(coreSetup, {
     uiActions: uiActions.setup,
@@ -53,7 +55,9 @@ const createPlugin = (
     setup,
     doStart: (anotherCoreStart: CoreStart = coreStart) => {
       const uiActionsStart = uiActions.doStart();
-      const embeddableStart = embeddable.doStart();
+      const embeddableStart = embeddable.doStart({
+        uiActions: uiActionsStart,
+      });
       return plugin.start(anotherCoreStart, {
         uiActions: uiActionsStart,
         embeddable: embeddableStart,