codeceptjs 4.0.0-rc.9 → 4.0.0

package/lib/helper/AI.js

@@ -1,214 +0,0 @@
-import HelperModule from '@codeceptjs/helper'
-import ora from 'ora-classic'
-import fs from 'fs'
-import path from 'path'
-import ai from '../ai.js'
-import Container from '../container.js'
-import { splitByChunks, minifyHtml } from '../html.js'
-import { beautify } from '../utils.js'
-import output from '../output.js'
-import { registerVariable } from '../pause.js'
-
-const standardActingHelpers = Container.STANDARD_ACTING_HELPERS
-
-const gtpRole = {
-  user: 'user',
-}
-
-/**
- * AI Helper for CodeceptJS.
- *
- * This helper class provides integration with the AI GPT-3.5 or 4 language model for generating responses to questions or prompts within the context of web pages. It allows you to interact with the GPT-3.5 model to obtain intelligent responses based on HTML fragments or general prompts.
- * This helper should be enabled with any web helpers like Playwright or Puppeteer or WebDriver to ensure the HTML context is available.
- *
- * Use it only in development mode. It is recommended to run it only inside pause() mode.
- *
- * ## Configuration
- *
- * This helper should be configured in codecept.conf.{js|ts}
- *
- * * `chunkSize`: (optional, default: 80000) - The maximum number of characters to send to the AI API at once. We split HTML fragments by 8000 chars to not exceed token limit. Increase this value if you use GPT-4.
- */
-class AI extends Helper {
-  constructor(config) {
-    super(config)
-    this.aiAssistant = ai
-
-    this.options = {
-      chunkSize: 80000,
-    }
-    this.options = { ...this.options, ...config }
-    this.aiAssistant.enable(this.config)
-  }
-
-  _beforeSuite() {
-    const helpers = Container.helpers()
-
-    for (const helperName of standardActingHelpers) {
-      if (Object.keys(helpers).indexOf(helperName) > -1) {
-        this.helper = helpers[helperName]
-        break
-      }
-    }
-  }
-
-  /**
-   * Asks the AI GPT language model a question based on the provided prompt within the context of the current page's HTML.
-   *
-   * ```js
-   * I.askGptOnPage('what does this page do?');
-   * ```
-   *
-   * @async
-   * @param {string} prompt - The question or prompt to ask the GPT model.
-   * @returns {Promise<string>} - A Promise that resolves to the generated responses from the GPT model, joined by newlines.
-   */
-  async askGptOnPage(prompt) {
-    const html = await this.helper.grabSource()
-
-    const htmlChunks = splitByChunks(html, this.options.chunkSize)
-
-    if (htmlChunks.length > 1) this.debug(`Splitting HTML into ${htmlChunks.length} chunks`)
-
-    const responses = []
-
-    for (const chunk of htmlChunks) {
-      const messages = [
-        { role: gtpRole.user, content: prompt },
-        { role: gtpRole.user, content: `Within this HTML: ${await minifyHtml(chunk)}` },
-      ]
-
-      if (htmlChunks.length > 1)
-        messages.push({
-          role: 'user',
-          content: 'If action is not possible on this page, do not propose anything, I will send another HTML fragment',
-        })
-
-      const response = await this._processAIRequest(messages)
-
-      output.print(response)
-
-      responses.push(response)
-    }
-
-    return responses.join('\n\n')
-  }
-
-  /**
-   * Asks the AI a question based on the provided prompt within the context of a specific HTML fragment on the current page.
-   *
-   * ```js
-   * I.askGptOnPageFragment('describe features of this screen', '.screen');
-   * ```
-   *
-   * @async
-   * @param {string} prompt - The question or prompt to ask the GPT-3.5 model.
-   * @param {string} locator - The locator or selector used to identify the HTML fragment on the page.
-   * @returns {Promise<string>} - A Promise that resolves to the generated response from the GPT model.
-   */
-  async askGptOnPageFragment(prompt, locator) {
-    const html = await this.helper.grabHTMLFrom(locator)
-
-    const messages = [
-      { role: gtpRole.user, content: prompt },
-      { role: gtpRole.user, content: `Within this HTML: ${await minifyHtml(html)}` },
-    ]
-
-    const response = await this._processAIRequest(messages)
-
-    output.print(response)
-
-    return response
-  }
-
-  /**
-   * Send a general request to AI and return response.
-   * @param {string} prompt
-   * @returns {Promise<string>} - A Promise that resolves to the generated response from the GPT model.
-   */
-  async askGptGeneralPrompt(prompt) {
-    const messages = [{ role: gtpRole.user, content: prompt }]
-
-    const response = await this._processAIRequest(messages)
-
-    output.print(response)
-
-    return response
-  }
-
-  /**
-   * Generates PageObject for current page using AI.
-   *
-   * It saves the PageObject to the output directory. You can review the page object and adjust it as needed and move to pages directory.
-   * Prompt can be customized in a global config file.
-   *
-   * ```js
-   * // create page object for whole page
-   * I.askForPageObject('home');
-   *
-   * // create page object with extra prompt
-   * I.askForPageObject('home', 'implement signIn(username, password) method');
-   *
-   * // create page object for a specific element
-   * I.askForPageObject('home', null, '.detail');
-   * ```
-   *
-   * Asks for a page object based on the provided page name, locator, and extra prompt.
-   *
-   * @async
-   * @param {string} pageName - The name of the page to retrieve the object for.
-   * @param {string|null} [extraPrompt=null] - An optional extra prompt for additional context or information.
-   * @param {string|null} [locator=null] - An optional locator to find a specific element on the page.
-   * @returns {Promise<Object>} A promise that resolves to the requested page object.
-   */
-  async askForPageObject(pageName, extraPrompt = null, locator = null) {
-    const spinner = ora(' Processing AI request...').start()
-
-    try {
-      const html = locator ? await this.helper.grabHTMLFrom(locator) : await this.helper.grabSource()
-      await this.aiAssistant.setHtmlContext(html)
-      const response = await this.aiAssistant.generatePageObject(extraPrompt, locator)
-      spinner.stop()
-
-      if (!response[0]) {
-        output.error('No response from AI')
-        return ''
-      }
-
-      const code = beautify(response[0])
-
-      output.print('----- Generated PageObject ----')
-      output.print(code)
-      output.print('-------------------------------')
-
-      const fileName = path.join(output_dir, `${pageName}Page-${Date.now()}.js`)
-
-      output.print(output.styles.bold(`Page object for ${pageName} is saved to ${output.styles.bold(fileName)}`))
-      fs.writeFileSync(fileName, code)
-
-      try {
-        registerVariable('page', require(fileName))
-        output.success('Page object registered for this session as `page` variable')
-        output.print('Use `=>page.methodName()` in shell to run methods of page object')
-        output.print('Use `click(page.locatorName)` to check locators of page object')
-      } catch (err) {
-        output.error('Error while registering page object')
-        output.error(err.message)
-      }
-
-      return code
-    } catch (e) {
-      spinner.stop()
-      throw Error(`Something went wrong! ${e.message}`)
-    }
-  }
-
-  async _processAIRequest(messages) {
-    const spinner = ora(' Processing AI request...').start()
-    const response = await this.aiAssistant.createCompletion(messages)
-    spinner.stop()
-    return response
-  }
-}
-
-export default AI

package/lib/helper/Mochawesome.js

@@ -1,96 +0,0 @@
-let currentTest
-let currentSuite
-
-import Helper from '@codeceptjs/helper'
-import { createRequire } from 'module'
-import { clearString } from '../utils.js'
-import { testToFileName } from '../mocha/test.js'
-
-class Mochawesome extends Helper {
-  constructor(config) {
-    super(config)
-
-    // set defaults
-    this.options = {
-      uniqueScreenshotNames: false,
-      disableScreenshots: false,
-    }
-
-    // ESM-compatible require for CommonJS module
-    const require = createRequire(import.meta.url)
-    this._addContext = require('mochawesome/addContext')
-
-    this._createConfig(config)
-  }
-
-  _createConfig(config) {
-    // override defaults with config
-    Object.assign(this.options, config)
-  }
-
-  _beforeSuite(suite) {
-    currentSuite = suite
-    currentTest = ''
-  }
-
-  _before() {
-    if (currentSuite && currentSuite.ctx) {
-      currentTest = { test: currentSuite.ctx.currentTest }
-    }
-  }
-
-  _test(test) {
-    // If this is a retried test, we want to add context to the retried test
-    // but also potentially preserve context from the original test
-    const originalTest = test.retriedTest && test.retriedTest()
-    if (originalTest) {
-      // This is a retried test - use the retried test for context
-      currentTest = { test }
-
-      // Optionally copy context from original test if it exists
-      // Note: mochawesome context is stored in test.ctx, but we need to be careful
-      // not to break the mocha context structure
-    } else {
-      // Normal test (not a retry)
-      currentTest = { test }
-    }
-  }
-
-  _failed(test) {
-    if (this.options.disableScreenshots) return
-    let fileName
-    // Get proper name if we are fail on hook
-    if (test.ctx?.test?.type === 'hook') {
-      currentTest = { test: test.ctx.test }
-      // ignore retries if we are in hook
-      test._retries = -1
-      fileName = clearString(`${test.title}_${currentTest.test.title}`)
-    } else {
-      currentTest = { test }
-      fileName = testToFileName(test)
-    }
-    if (this.options.uniqueScreenshotNames) {
-      fileName = testToFileName(test, { unique: true })
-    }
-    if (test._retries < 1 || test._retries === test.retryNum) {
-      fileName = `${fileName}.failed.png`
-      return this._addContext(currentTest, fileName)
-    }
-  }
-
-  addMochawesomeContext(context) {
-    if (currentTest === '') currentTest = { test: currentSuite.ctx.test }
-
-    // For retried tests, make sure we're adding context to the current (retried) test
-    // not the original test
-    let targetTest = currentTest
-    if (currentTest.test && currentTest.test.retriedTest && currentTest.test.retriedTest()) {
-      // This test has been retried, make sure we're using the current test for context
-      targetTest = { test: currentTest.test }
-    }
-
-    return this._addContext(targetTest, context)
-  }
-}
-
-export default Mochawesome

package/lib/helper/extras/PlaywrightReactVueLocator.js

@@ -1,52 +0,0 @@
-async function findReact(matcher, locator) {
-  // Handle both Locator objects and raw locator objects
-  const reactLocator = locator.locator || locator
-  let _locator = `_react=${reactLocator.react}`;
-  let props = '';
-
-  if (reactLocator.props) {
-    props += propBuilder(reactLocator.props);
-    _locator += props;
-  }
-  return matcher.locator(_locator).all();
-}
-
-async function findVue(matcher, locator) {
-  // Handle both Locator objects and raw locator objects
-  const vueLocator = locator.locator || locator
-  let _locator = `_vue=${vueLocator.vue}`;
-  let props = '';
-
-  if (vueLocator.props) {
-    props += propBuilder(vueLocator.props);
-    _locator += props;
-  }
-  return matcher.locator(_locator).all();
-}
-
-async function findByPlaywrightLocator(matcher, locator) {
-  // Handle both Locator objects and raw locator objects
-  const pwLocator = locator.locator || locator
-  if (pwLocator && pwLocator.toString && pwLocator.toString().includes(process.env.testIdAttribute)) {
-    return matcher.getByTestId(pwLocator.pw.value.split('=')[1]);
-  }
-  const pwValue = typeof pwLocator.pw === 'string' ? pwLocator.pw : pwLocator.pw
-  return matcher.locator(pwValue).all();
-}
-
-function propBuilder(props) {
-  let _props = '';
-
-  for (const [key, value] of Object.entries(props)) {
-    if (typeof value === 'object') {
-      for (const [k, v] of Object.entries(value)) {
-        _props += `[${key}.${k} = "${v}"]`;
-      }
-    } else {
-      _props += `[${key} = "${value}"]`;
-    }
-  }
-  return _props;
-}
-
-export { findReact, findVue, findByPlaywrightLocator };

package/lib/helper/extras/React.js

@@ -1,65 +0,0 @@
-import fs from 'fs'
-
-let resqScript
-
-export default async function findReact(matcher, locator) {
-  if (!resqScript) resqScript = fs.readFileSync(new URL('resq', import.meta.resolve('resq')).pathname)
-  await matcher.evaluate(resqScript.toString())
-  await matcher.evaluate(() => window.resq.waitToLoadReact())
-  const arrayHandle = await matcher.evaluateHandle(
-    obj => {
-      const { selector, props, state } = obj
-
-      let elements = window.resq.resq$$(selector)
-      if (Object.keys(props).length) {
-        elements = elements.byProps(props)
-      }
-      if (Object.keys(state).length) {
-        elements = elements.byState(state)
-      }
-
-      if (!elements.length) {
-        return []
-      }
-
-      // resq returns an array of HTMLElements if the React component is a fragment
-      // this avoids having nested arrays of nodes which the driver does not understand
-      // [[div, div], [div, div]] => [div, div, div, div]
-      let nodes = []
-
-      elements.forEach(element => {
-        let { node, isFragment } = element
-
-        if (!node) {
-          isFragment = true
-          node = element.children
-        }
-
-        if (isFragment) {
-          nodes = nodes.concat(node)
-        } else {
-          nodes.push(node)
-        }
-      })
-
-      return [...nodes]
-    },
-    {
-      selector: locator.react,
-      props: locator.props || {},
-      state: locator.state || {},
-    },
-  )
-
-  const properties = await arrayHandle.getProperties()
-  await arrayHandle.dispose()
-  const result = []
-  for (const property of properties.values()) {
-    const elementHandle = property.asElement()
-    if (elementHandle) {
-      result.push(elementHandle)
-    }
-  }
-
-  return result
-}