chore: add snapshotState to local expect

vitest-dev · Jan 1, 2023 · fbebc41 · fbebc41
1 parent 2216429
commit fbebc41
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/packages/vitest/src/integrations/chai/index.ts b/packages/vitest/src/integrations/chai/index.ts
@@ -21,15 +21,20 @@ export function createExpect(test?: Test) {
   expect.getState = () => getState<MatcherState>(expect)
   expect.setState = state => setState(state as Partial<MatcherState>, expect)
 
+  // @ts-expect-error global is not typed
+  const globalState = getState(globalThis[GLOBAL_EXPECT]) || {}
+
   setState<MatcherState>({
+    // this should also add "snapshotState" that is added conditionally
+    ...globalState,
     assertionCalls: 0,
     isExpectingAssertions: false,
     isExpectingAssertionsError: null,
     expectedAssertionsNumber: null,
     expectedAssertionsNumberErrorGen: null,
     environment: getCurrentEnvironment(),
-    testPath: test?.suite.file?.filepath,
-    currentTestName: test ? getFullName(test) : undefined,
+    testPath: test ? test.suite.file?.filepath : globalState.testPath,
+    currentTestName: test ? getFullName(test) : globalState.currentTestName,
   }, expect)
 
   // @ts-expect-error untyped

diff --git a/test/core/test/local-context.test.ts b/test/core/test/local-context.test.ts
@@ -25,7 +25,14 @@ describe('local test context works with implicit type', () => {
   it('works with implicit type', (context: LocalTestContext) => {
     expect(context.bar).toBe('bar')
   })
-  it.only('is chainable with implicit type', (context: LocalTestContext) => {
+  it('is chainable with implicit type', (context: LocalTestContext) => {
     expect(context.bar).toBe('bar')
   })
 })
+
+describe('context expect', () => {
+  it('has snapshotState', ({ expect: localExpect }) => {
+    expect(expect.getState().snapshotState).toBeDefined()
+    expect(localExpect.getState().snapshotState).toBeDefined()
+  })
+})