Fix tests

This commit is contained in:
Tom Coleman 2023-03-03 15:16:12 +11:00
parent 7bbc0e644f
commit 11ee42ef96
2 changed files with 24 additions and 7 deletions

View File

@ -1,5 +1,5 @@
import { Channel } from '@storybook/channels';
import type { Renderer, DocsIndexEntry } from '@storybook/types';
import type { Renderer, DocsIndexEntry, RenderContextCallbacks } from '@storybook/types';
import type { StoryStore } from '../../store';
import { PREPARE_ABORTED } from './Render';
@ -36,7 +36,8 @@ it('throws PREPARE_ABORTED if torndown during prepare', async () => {
const render = new CsfDocsRender(
new Channel(),
mockStore as unknown as StoryStore<Renderer>,
entry
entry,
{} as RenderContextCallbacks<Renderer>
);
const preparePromise = render.prepare();
@ -61,7 +62,12 @@ it('attached immediately', async () => {
storyFromCSFFile: () => story,
} as unknown as StoryStore<Renderer>;
const render = new CsfDocsRender(new Channel(), store, entry);
const render = new CsfDocsRender(
new Channel(),
store,
entry,
{} as RenderContextCallbacks<Renderer>
);
await render.prepare();
const context = render.docsContext(jest.fn());

View File

@ -1,5 +1,5 @@
import { Channel } from '@storybook/channels';
import type { Renderer, DocsIndexEntry } from '@storybook/types';
import type { Renderer, DocsIndexEntry, RenderContextCallbacks } from '@storybook/types';
import type { StoryStore } from '../../store';
import { PREPARE_ABORTED } from './Render';
@ -35,7 +35,8 @@ it('throws PREPARE_ABORTED if torndown during prepare', async () => {
const render = new MdxDocsRender(
new Channel(),
mockStore as unknown as StoryStore<Renderer>,
entry
entry,
{} as RenderContextCallbacks<Renderer>
);
const preparePromise = render.prepare();
@ -60,7 +61,12 @@ describe('attaching', () => {
} as unknown as StoryStore<Renderer>;
it('is not attached if you do not call setMeta', async () => {
const render = new MdxDocsRender(new Channel(), store, entry);
const render = new MdxDocsRender(
new Channel(),
store,
entry,
{} as RenderContextCallbacks<Renderer>
);
await render.prepare();
const context = render.docsContext(jest.fn());
@ -69,7 +75,12 @@ describe('attaching', () => {
});
it('is attached if you call referenceMeta with attach=true', async () => {
const render = new MdxDocsRender(new Channel(), store, entry);
const render = new MdxDocsRender(
new Channel(),
store,
entry,
{} as RenderContextCallbacks<Renderer>
);
await render.prepare();
const context = render.docsContext(jest.fn());