Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion code/core/src/csf/story.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
import type { BoundFunctions, queries } from '@testing-library/dom';
import type { userEvent } from '@testing-library/user-event';

import type { OmitIndexSignature, Simplify, UnionToIntersection } from 'type-fest';

import type { ToolbarArgType } from '../toolbar/index.ts';
Expand Down Expand Up @@ -264,7 +267,7 @@ export type AfterEach<TRenderer extends Renderer = Renderer, TArgs = Args> = (
context: StoryContext<TRenderer, TArgs>
) => Awaitable<void>;

export interface Canvas {}
export interface Canvas extends BoundFunctions<typeof queries> {}

export interface StoryContext<TRenderer extends Renderer = Renderer, TArgs = Args>
extends StoryContextForEnhancers<TRenderer, TArgs>, Required<StoryContextUpdate<TArgs>> {
Expand All @@ -277,6 +280,7 @@ export interface StoryContext<TRenderer extends Renderer = Renderer, TArgs = Arg
step: StepFunction<TRenderer, TArgs>;
context: this;
canvas: Canvas;
userEvent: ReturnType<typeof userEvent.setup>;
mount: TRenderer['mount'];
reporting: ReportingAPI;
}
Expand Down
11 changes: 0 additions & 11 deletions code/core/src/test/index.ts
Original file line number Diff line number Diff line change
@@ -1,27 +1,16 @@
import type { BoundFunctions } from '@testing-library/dom';
import type { userEvent } from '@testing-library/user-event';

import { instrument } from 'storybook/internal/instrumenter';

import { Assertion } from 'chai';

import { expect as rawExpect } from './expect.ts';
import { type queries } from './testing-library.ts';

export * from './spy.ts';
export type { Assertion, Expect } from './expect.ts';

type Queries = BoundFunctions<typeof queries>;

export type UserEventObject = ReturnType<typeof userEvent.setup>;

declare module 'storybook/internal/csf' {
interface Canvas extends Queries {}
interface StoryContext {
userEvent: UserEventObject;
}
}

export const { expect } = instrument(
{ expect: rawExpect },
{
Expand Down
Loading