Pass samples loading tests

e2e/playwright/testing-samples-loading.spec.ts

@@ -1,35 +1,25 @@
-import { test, expect } from '@playwright/test'
-import { getUtils, setup, setupElectron, tearDown } from './test-utils'
+import { test, expect } from './zoo-test'
+import { getUtils } from './test-utils'
 import { bracket } from 'lib/exampleKcl'
 import * as fsp from 'fs/promises'
 import { join } from 'path'
 import { FILE_EXT } from 'lib/constants'
 import { UnitLength_type } from '@kittycad/lib/dist/types/src/models'
 
-test.beforeEach(async ({ context, page }, testInfo) => {
-  await setup(context, page, testInfo)
-})
-
-test.afterEach(async ({ page }, testInfo) => {
-  await tearDown(page, testInfo)
-})
-
 test.describe('Testing in-app sample loading', () => {
   /**
    * Note this test implicitly depends on the KCL sample "car-wheel.kcl",
    * its title, and its units settings. https://github.com/KittyCAD/kcl-samples/blob/main/car-wheel/car-wheel.kcl
    */
-  test('Web: should overwrite current code, cannot create new file', async ({
-    page,
-  }) => {
+  test('Web: should overwrite current code, cannot create new file', async ({ editor, context, page,  homePage }) => {
     const u = await getUtils(page)
   
   await test.step(`Test setup`, async () => {
-      await page.addInitScript((code) => {
+    await context.addInitScript((code) => {
       window.localStorage.setItem('persistCode', code)
     }, bracket)
-      await page.setViewportSize({ width: 1200, height: 500 })
-      await u.waitForAuthSkipAppStart()
+    await page.setBodyDimensions({ width: 1200, height: 500 })
+    await homePage.goToModelingScene()
   })
   
   // Locators and constants
@@ -54,13 +44,13 @@ test.describe('Testing in-app sample loading', () => {
     })
   const warningText = page.getByText('Overwrite current file and units?')
   const confirmButton = page.getByRole('button', { name: 'Submit command' })
-    const codeLocator = page.locator('.cm-content')
   const unitsToast = (unit: UnitLength_type) =>
     page.getByText(`Set default unit to "${unit}" for this project`)
   
   await test.step(`Precondition: check the initial code`, async () => {
     await u.openKclCodePanel()
-      await expect(codeLocator).toContainText(bracket.split('\n')[0])
+    await editor.scrollToText(bracket.split('\n')[0])
+    await editor.expectEditor.toContain(bracket.split('\n')[0])
   })
   
   await test.step(`Load a KCL sample with the command palette`, async () => {
@@ -73,10 +63,9 @@ test.describe('Testing in-app sample loading', () => {
     await expect(warningText).toBeVisible()
     await confirmButton.click()
   
-      await expect(codeLocator).toContainText('// ' + newSample.title)
+    await editor.expectEditor.toContain('// ' + newSample.title)
     await expect(unitsToast('in')).toBeVisible()
-    })
-  })
+  }) })
 
   /**
    * Note this test implicitly depends on the KCL samples:
@@ -86,16 +75,13 @@ test.describe('Testing in-app sample loading', () => {
   test(
     'Desktop: should create new file by default, optionally overwrite',
     { tag: '@electron' },
-    async ({ browserName: _ }, testInfo) => {
-      const { electronApp, page, dir } = await setupElectron({
-        testInfo,
-        folderSetupFn: async (dir) => {
+    async ({ editor, context, page, browserName: _ }, testInfo) => {
+      const { dir } = await context.folderSetupFn(async (dir) => {
         const bracketDir = join(dir, 'bracket')
         await fsp.mkdir(bracketDir, { recursive: true })
         await fsp.writeFile(join(bracketDir, 'main.kcl'), bracket, {
           encoding: 'utf-8',
         })
-        },
       })
       const u = await getUtils(page)
 
@@ -134,19 +120,19 @@ test.describe('Testing in-app sample loading', () => {
         page.getByRole('listitem').filter({
           has: page.getByRole('button', { name }),
         })
-      const codeLocator = page.locator('.cm-content')
       const unitsToast = (unit: UnitLength_type) =>
         page.getByText(`Set default unit to "${unit}" for this project`)
 
       await test.step(`Test setup`, async () => {
-        await page.setViewportSize({ width: 1200, height: 500 })
+        await page.setBodyDimensions({ width: 1200, height: 500 })
         await projectCard.click()
         await u.waitForPageLoad()
       })
 
       await test.step(`Precondition: check the initial code`, async () => {
         await u.openKclCodePanel()
-        await expect(codeLocator).toContainText(bracket.split('\n')[0])
+        await editor.scrollToText(bracket.split('\n')[0])
+        await editor.expectEditor.toContain(bracket.split('\n')[0])
         await u.openFilePanel()
 
         await expect(projectMenuButton).toContainText('main.kcl')
@@ -163,7 +149,7 @@ test.describe('Testing in-app sample loading', () => {
       })
 
       await test.step(`Ensure we made and opened a new file`, async () => {
-        await expect(codeLocator).toContainText('// ' + sampleOne.title)
+        await editor.expectEditor.toContain('// ' + sampleOne.title)
         await expect(newlyCreatedFile(sampleOne.file)).toBeVisible()
         await expect(projectMenuButton).toContainText(sampleOne.file)
         await expect(unitsToast('in')).toBeVisible()
@@ -182,7 +168,7 @@ test.describe('Testing in-app sample loading', () => {
       })
 
       await test.step(`Ensure we overwrote the current file without navigating`, async () => {
-        await expect(codeLocator).toContainText('// ' + sampleTwo.title)
+        await editor.expectEditor.toContain('// ' + sampleTwo.title)
         await test.step(`Check actual file contents`, async () => {
           await expect
             .poll(async () => {
@@ -198,8 +184,6 @@ test.describe('Testing in-app sample loading', () => {
         await expect(projectMenuButton).toContainText(sampleOne.file)
         await expect(unitsToast('mm')).toBeVisible()
       })
-
-      await electronApp.close()
     }
   )
 })