import { shallowMount } from "@vue/test-utils";
import { defu } from "defu";
import { vi } from "vitest";
import { defineComponent, h } from "vue";

type Injections = {
  shopware: unknown;
  apiClient: {
    invoke: ReturnType<typeof vi.fn>;
  };
  swNotifications?: unknown;
};

export function useSetup<V>(setup: () => V, customMocks?: Partial<Injections>) {
  const defaultInjections: Injections = {
    shopware: {},
    apiClient: { invoke: customMocks?.apiClient?.invoke ?? vi.fn() },
  };

  const component = defineComponent({
    setup,
    render() {
      return h("div", []);
    },
  });

  const injections: Injections = defu(
    defaultInjections,
    customMocks || {},
  ) as Injections;

  const wrapper = shallowMount(component, {
    global: {
      provide: injections,
    },
  });

  return {
    injections,
    wrapper,
    vm: wrapper.vm,
  };
}