test: add workspace ref in e2e (#3217)

tests/affine-local/e2e/local-first-restore-page.spec.ts

@@ -5,11 +5,11 @@ import {
   newPage,
   waitEditorLoad,
 } from '@affine-test/kit/utils/page-logic';
-import { assertCurrentWorkspaceFlavour } from '@affine-test/kit/utils/workspace';
 import { expect } from '@playwright/test';
 
 test('New a page , then delete it in all pages, restore it', async ({
   page,
+  workspace,
 }) => {
   await openHomePage(page);
   await waitEditorLoad(page);
@@ -52,5 +52,7 @@ test('New a page , then delete it in all pages, restore it', async ({
     name: 'this is a new page to restore',
   });
   expect(restoreCell).not.toBeUndefined();
-  await assertCurrentWorkspaceFlavour('local', page);
+  const currentWorkspace = await workspace.current();
+
+  expect(currentWorkspace.flavour).toContain('local');
 });