From ef8dea8cb29c970c9c355e6946c2a3abdb57e968 Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Mon, 24 Apr 2023 13:18:37 -0500
Subject: [PATCH] test: fix flaky in `customElements` (#2109)

---
 .../__tests__/WorkSpaceSliderBar.spec.tsx     | 121 ------------------
 scripts/setup/lit.ts                          |  12 ++
 vitest.config.ts                              |   1 +
 3 files changed, 13 insertions(+), 121 deletions(-)
 delete mode 100644 apps/web/src/components/__tests__/WorkSpaceSliderBar.spec.tsx
 create mode 100644 scripts/setup/lit.ts

diff --git a/apps/web/src/components/__tests__/WorkSpaceSliderBar.spec.tsx b/apps/web/src/components/__tests__/WorkSpaceSliderBar.spec.tsx
deleted file mode 100644
index 628381729..000000000
--- a/apps/web/src/components/__tests__/WorkSpaceSliderBar.spec.tsx
+++ /dev/null
@@ -1,121 +0,0 @@
-/**
- * @vitest-environment happy-dom
- */
-import 'fake-indexeddb/auto';
-
-import {
-  rootCurrentPageIdAtom,
-  rootCurrentWorkspaceIdAtom,
-} from '@affine/workspace/atom';
-import { WorkspaceFlavour } from '@affine/workspace/type';
-import { assertExists } from '@blocksuite/store';
-import { render, renderHook } from '@testing-library/react';
-import { useBlockSuiteWorkspaceHelper } from '@toeverything/hooks/use-block-suite-workspace-helper';
-import { createStore, getDefaultStore, Provider, useAtomValue } from 'jotai';
-import { useRouter } from 'next/router';
-import type React from 'react';
-import { useCallback } from 'react';
-import { beforeEach, describe, expect, test, vi } from 'vitest';
-
-import { workspacesAtom } from '../../atoms';
-import { rootCurrentWorkspaceAtom } from '../../atoms/root';
-import {
-  currentWorkspaceAtom,
-  useCurrentWorkspace,
-} from '../../hooks/current/use-current-workspace';
-import { useAppHelper } from '../../hooks/use-workspaces';
-import { ThemeProvider } from '../../providers/ThemeProvider';
-import { pathGenerator } from '../../shared';
-import { WorkSpaceSliderBar } from '../pure/workspace-slider-bar';
-
-vi.mock('../blocksuite/header/editor-mode-switch/CustomLottie', () => ({
-  default: (props: React.PropsWithChildren) => <>{props.children}</>,
-}));
-
-// fetchMocker.enableMocks();
-let store = getDefaultStore();
-beforeEach(async () => {
-  store = createStore();
-  await store.get(workspacesAtom);
-});
-
-const ProviderWrapper: React.FC<React.PropsWithChildren> = ({ children }) => {
-  return <Provider store={store}>{children}</Provider>;
-};
-
-describe('WorkSpaceSliderBar', () => {
-  test('basic', async () => {
-    // fetchMocker.mock
-
-    const onOpenWorkspaceListModalFn = vi.fn();
-    const onOpenQuickSearchModalFn = vi.fn();
-    const mutationHook = renderHook(() => useAppHelper(), {
-      wrapper: ProviderWrapper,
-    });
-    const id = await mutationHook.result.current.createLocalWorkspace('test0');
-    await store.get(workspacesAtom);
-    mutationHook.rerender();
-    mutationHook.result.current.createWorkspacePage(id, 'test1');
-    store.set(rootCurrentWorkspaceIdAtom, id);
-    await store.get(rootCurrentWorkspaceAtom);
-    const currentWorkspaceHook = renderHook(() => useCurrentWorkspace(), {
-      wrapper: ProviderWrapper,
-    });
-    let i = 0;
-    const Component = () => {
-      const [currentWorkspace] = useCurrentWorkspace();
-      const currentPageId = useAtomValue(rootCurrentPageIdAtom);
-      assertExists(currentWorkspace);
-      const helper = useBlockSuiteWorkspaceHelper(
-        currentWorkspace.blockSuiteWorkspace
-      );
-      return (
-        <WorkSpaceSliderBar
-          currentWorkspace={currentWorkspace}
-          currentPageId={currentPageId}
-          onOpenQuickSearchModal={onOpenQuickSearchModalFn}
-          onOpenWorkspaceListModal={onOpenWorkspaceListModalFn}
-          openPage={useCallback(() => {}, [])}
-          createPage={() => {
-            i++;
-            return helper.createPage('page-test-' + i);
-          }}
-          currentPath={useRouter().asPath}
-          paths={pathGenerator}
-          isPublicWorkspace={false}
-        />
-      );
-    };
-    const App = () => {
-      return (
-        <ThemeProvider>
-          <ProviderWrapper>
-            <Component />
-          </ProviderWrapper>
-        </ThemeProvider>
-      );
-    };
-    currentWorkspaceHook.result.current[1](id);
-    const currentWorkspace = await store.get(currentWorkspaceAtom);
-    expect(currentWorkspace).toBeDefined();
-    expect(currentWorkspace?.flavour).toBe(WorkspaceFlavour.LOCAL);
-    expect(currentWorkspace?.id).toBe(id);
-    const app = render(<App />);
-    const card = await app.findByTestId('current-workspace');
-    expect(onOpenWorkspaceListModalFn).toBeCalledTimes(0);
-    card.click();
-    expect(onOpenWorkspaceListModalFn).toBeCalledTimes(1);
-    const newPageButton = await app.findByTestId('new-page-button');
-    newPageButton.click();
-    expect(
-      currentWorkspaceHook.result.current[0]?.blockSuiteWorkspace.meta
-        .pageMetas[1].id
-    ).toBe('page-test-1');
-    expect(onOpenQuickSearchModalFn).toBeCalledTimes(0);
-    const quickSearchButton = await app.findByTestId(
-      'slider-bar-quick-search-button'
-    );
-    quickSearchButton.click();
-    expect(onOpenQuickSearchModalFn).toBeCalledTimes(1);
-  });
-});
diff --git a/scripts/setup/lit.ts b/scripts/setup/lit.ts
new file mode 100644
index 000000000..fc23c8e65
--- /dev/null
+++ b/scripts/setup/lit.ts
@@ -0,0 +1,12 @@
+import { vi } from 'vitest';
+
+if (typeof window !== 'undefined') {
+  // Refs: https://github.com/jsdom/jsdom/blob/master/lib/jsdom/living/custom-elements/CustomElementRegistry-impl.js
+  const customElements = {
+    define: vi.fn(),
+    get: vi.fn(),
+    whenDefined: vi.fn(),
+    upgrade: vi.fn(),
+  };
+  vi.stubGlobal('customElements', customElements);
+}
diff --git a/vitest.config.ts b/vitest.config.ts
index 7a3b5478f..076971158 100644
--- a/vitest.config.ts
+++ b/vitest.config.ts
@@ -18,6 +18,7 @@ export default defineConfig({
   },
   test: {
     setupFiles: [
+      resolve(rootDir, './scripts/setup/lit.ts'),
       resolve(rootDir, './scripts/setup/search.ts'),
       resolve(rootDir, './scripts/setup/lottie-web.ts'),
     ],