playwright/tests/folio/src/reporters/base.ts

/**
 * Copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { codeFrameColumns } from '@babel/code-frame';
import colors from 'colors/safe';
import fs from 'fs';
import milliseconds from 'ms';
import path from 'path';
import StackUtils from 'stack-utils';
import { TestStatus, Test, Suite, TestResult, TestError, Reporter } from '../types';
import { FullConfig } from '../types';

const stackUtils = new StackUtils();

export class BaseReporter implements Reporter  {
  duration = 0;
  config: FullConfig;
  suite: Suite;
  timeout: number;
  fileDurations = new Map<string, number>();
  monotonicStartTime: number;

  constructor() {
  }

  onBegin(config: FullConfig, suite: Suite) {
    this.monotonicStartTime = monotonicTime();
    this.config = config;
    this.suite = suite;
  }

  onTestBegin(test: Test) {
  }

  onStdOut(chunk: string | Buffer) {
    if (!this.config.quiet)
      process.stdout.write(chunk);
  }

  onStdErr(chunk: string | Buffer) {
    if (!this.config.quiet)
      process.stderr.write(chunk);
  }

  onTestEnd(test: Test, result: TestResult) {
    const spec = test.spec;
    let duration = this.fileDurations.get(spec.file) || 0;
    duration += result.duration;
    this.fileDurations.set(spec.file, duration);
  }

  onError(error: TestError) {
    console.log(formatError(error));
  }

  onTimeout(timeout: number) {
    this.timeout = timeout;
  }

  onEnd() {
    this.duration = monotonicTime() - this.monotonicStartTime;
  }

  private _printSlowTests() {
    const fileDurations = [...this.fileDurations.entries()];
    fileDurations.sort((a, b) => b[1] - a[1]);
    let insertedGap = false;
    for (let i = 0; i < 10 && i < fileDurations.length; ++i) {
      const baseName = path.basename(fileDurations[i][0]);
      const duration = fileDurations[i][1];
      if (duration < 15000)
        break;
      if (!insertedGap) {
        insertedGap = true;
        console.log();
      }
      console.log(colors.yellow('  Slow test: ') + baseName + colors.yellow(` (${milliseconds(duration)})`));
    }
    console.log();
  }

  epilogue(full: boolean) {
    let skipped = 0;
    let expected = 0;
    const unexpected: Test[] = [];
    const flaky: Test[] = [];

    this.suite.findTest(test => {
      switch (test.status()) {
        case 'skipped': ++skipped; break;
        case 'expected': ++expected; break;
        case 'unexpected': unexpected.push(test); break;
        case 'flaky': flaky.push(test); break;
      }
    });

    if (expected)
      console.log(colors.green(`  ${expected} passed`) + colors.dim(` (${milliseconds(this.duration)})`));
    if (skipped)
      console.log(colors.yellow(`  ${skipped} skipped`));
    if (unexpected.length) {
      console.log(colors.red(`  ${unexpected.length} failed`));
      this._printTestHeaders(unexpected);
    }
    if (flaky.length) {
      console.log(colors.red(`  ${flaky.length} flaky`));
      this._printTestHeaders(flaky);
    }
    if (this.timeout)
      console.log(colors.red(`  Timed out waiting ${this.timeout / 1000}s for the entire test run`));

    if (full && unexpected.length) {
      console.log('');
      this._printFailures(unexpected);
    }
    this._printSlowTests();
  }

  private _printTestHeaders(tests: Test[]) {
    tests.forEach(test => {
      console.log(formatTestHeader(this.config, test, '    '));
    });
  }

  private _printFailures(failures: Test[]) {
    failures.forEach((test, index) => {
      console.log(formatFailure(this.config, test, index + 1));
    });
  }

  hasResultWithStatus(test: Test, status: TestStatus): boolean {
    return !!test.results.find(r => r.status === status);
  }

  willRetry(test: Test, result: TestResult): boolean {
    return result.status !== 'passed' && result.status !== test.expectedStatus && test.results.length <= this.config.retries;
  }
}

export function formatFailure(config: FullConfig, test: Test, index?: number): string {
  const tokens: string[] = [];
  tokens.push(formatTestHeader(config, test, '  ', index));
  for (const result of test.results) {
    if (result.status === 'passed')
      continue;
    tokens.push(formatResult(test, result));
  }
  tokens.push('');
  return tokens.join('\n');
}

function formatTestHeader(config: FullConfig, test: Test, indent: string, index?: number): string {
  const tokens: string[] = [];
  const spec = test.spec;
  let relativePath = path.relative(config.testDir, spec.file) || path.basename(spec.file);
  relativePath += ':' + spec.line + ':' + spec.column;
  const passedUnexpectedlySuffix = test.results[0].status === 'passed' ? ' -- passed unexpectedly' : '';
  const runListName = test.alias ? `[${test.alias}] ` : '';
  const header = `${indent}${index ? index + ') ' : ''}${relativePath} › ${runListName}${spec.fullTitle()}${passedUnexpectedlySuffix}`;
  tokens.push(colors.red(pad(header, '=')));
  return tokens.join('\n');
}

function formatResult(test: Test, result: TestResult): string {
  const tokens: string[] = [];
  if (result.retry)
    tokens.push(colors.gray(pad(`\n    Retry #${result.retry}`, '-')));
  if (result.status === 'timedOut') {
    tokens.push('');
    tokens.push(indent(colors.red(`Timeout of ${test.timeout}ms exceeded.`), '    '));
  } else {
    tokens.push(indent(formatError(result.error, test.spec.file), '    '));
  }
  return tokens.join('\n');
}

function formatError(error: TestError, file?: string) {
  const stack = error.stack;
  const tokens = [];
  if (stack) {
    tokens.push('');
    const messageLocation = error.stack.indexOf(error.message);
    const preamble = error.stack.substring(0, messageLocation + error.message.length);
    tokens.push(preamble);
    const position = file ? positionInFile(stack, file) : null;
    if (position) {
      const source = fs.readFileSync(file, 'utf8');
      tokens.push('');
      tokens.push(codeFrameColumns(source, {
        start: position,
      },
      { highlightCode: true}
      ));
    }
    tokens.push('');
    tokens.push(colors.dim(stack.substring(preamble.length + 1)));
  } else {
    tokens.push('');
    tokens.push(error.value);
  }
  return tokens.join('\n');
}

function pad(line: string, char: string): string {
  return line + ' ' + colors.gray(char.repeat(Math.max(0, 100 - line.length - 1)));
}

function indent(lines: string, tab: string) {
  return lines.replace(/^(?=.+$)/gm, tab);
}

function positionInFile(stack: string, file: string): { column: number; line: number; } {
  // Stack will have /private/var/folders instead of /var/folders on Mac.
  file = fs.realpathSync(file);
  for (const line of stack.split('\n')) {
    const parsed = stackUtils.parseLine(line);
    if (!parsed)
      continue;
    if (path.resolve(process.cwd(), parsed.file) === file)
      return {column: parsed.column, line: parsed.line};
  }
  return null;
}

function monotonicTime(): number {
  const [seconds, nanoseconds] = process.hrtime();
  return seconds * 1000 + (nanoseconds / 1000000 | 0);
}

const asciiRegex = new RegExp('[\\u001B\\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[-a-zA-Z\\d\\/#&.:=?%@~_]*)*)?\\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PR-TZcf-ntqry=><~]))', 'g');
export function stripAscii(str: string): string {
  return str.replace(asciiRegex, '');
}