implement summary counts

Author: davidjgoss

javascript/src/SummaryPrinter.ts

@@ -1,8 +1,8 @@
-import { Envelope, TestRunFinished, TestRunStarted, TimeConversion } from '@cucumber/messages'
+import { Envelope, TestRunFinished, TestRunStarted, TestStepResultStatus } from '@cucumber/messages'
 import { Query } from '@cucumber/query'
 
+import { formatCounts, formatDuration } from './helpers.js'
 import type { Options } from './types.js'
-import { formatDuration } from './helpers.js'
 
 export class SummaryPrinter {
   private readonly println: (content?: string) => void
@@ -25,6 +25,48 @@ export class SummaryPrinter {
   }
 
   private printSummary() {
+    this.println()
+    this.printScenarioCounts()
+    this.printStepCounts()
+    this.printDuration()
+  }
+
+  private printScenarioCounts() {
+    const scenarioCountsByStatus = this.query
+      .findAllTestCaseFinished()
+      .map((testCaseFinished) => this.query.findMostSevereTestStepResultBy(testCaseFinished))
+      .map((testStepResult) => testStepResult?.status)
+      .filter((status) => !!status)
+      .reduce(
+        (prev, status) => {
+          return {
+            ...prev,
+            [status]: (prev[status] ?? 0) + 1,
+          }
+        },
+        {} as Partial<Record<TestStepResultStatus, number>>
+      )
+    this.println(formatCounts('scenarios', scenarioCountsByStatus, this.options.theme, this.stream))
+  }
+
+  private printStepCounts() {
+    const stepCountsByStatus = this.query
+      .findAllTestCaseFinished()
+      .flatMap((testCaseFinished) => this.query.findTestStepsFinishedBy(testCaseFinished))
+      .map((testStepFinished) => testStepFinished.testStepResult.status)
+      .reduce(
+        (prev, status) => {
+          return {
+            ...prev,
+            [status]: (prev[status] ?? 0) + 1,
+          }
+        },
+        {} as Partial<Record<TestStepResultStatus, number>>
+      )
+    this.println(formatCounts('steps', stepCountsByStatus, this.options.theme, this.stream))
+  }
+
+  private printDuration() {
     const testRunStarted = this.query.findTestRunStarted() as TestRunStarted
     const testRunFinished = this.query.findTestRunFinished() as TestRunFinished
 

javascript/src/helpers.ts

@@ -20,16 +20,25 @@ import {
   TimeConversion,
   Timestamp,
 } from '@cucumber/messages'
+import { Interval } from 'luxon'
 
 import { TextBuilder } from './TextBuilder.js'
 import { Theme } from './types.js'
-import { Interval } from 'luxon'
 
 export const GHERKIN_INDENT_LENGTH = 2
 export const STEP_ARGUMENT_INDENT_LENGTH = 2
 export const ATTACHMENT_INDENT_LENGTH = 4
 export const ERROR_INDENT_LENGTH = 4
 const DURATION_FORMAT = "m'm' s.S's'"
+const STATUS_ORDER: TestStepResultStatus[] = [
+  TestStepResultStatus.UNKNOWN,
+  TestStepResultStatus.PASSED,
+  TestStepResultStatus.SKIPPED,
+  TestStepResultStatus.PENDING,
+  TestStepResultStatus.UNDEFINED,
+  TestStepResultStatus.AMBIGUOUS,
+  TestStepResultStatus.FAILED,
+]
 const STATUS_CHARACTERS: Record<TestStepResultStatus, string> = {
   [TestStepResultStatus.AMBIGUOUS]: 'A',
   [TestStepResultStatus.FAILED]: 'F',
@@ -340,3 +349,30 @@ export function formatDuration(start: Timestamp, finish: Timestamp) {
   ])
   return duration.toFormat(DURATION_FORMAT)
 }
+
+export function formatCounts(
+  suffix: string,
+  counts: Partial<Record<TestStepResultStatus, number>>,
+  theme: Theme,
+  stream: NodeJS.WritableStream
+) {
+  const builder = new TextBuilder(stream)
+  const total = Object.values(counts).reduce((prev, curr) => prev + curr, 0)
+  builder.append(`${total} ${suffix}`)
+  if (total > 0) {
+    let first = true
+    builder.append(' (')
+    for (const status of STATUS_ORDER) {
+      const count = counts[status]
+      if (count) {
+        builder.append(`${count} ${status.toLowerCase()}`, theme.status?.all?.[status])
+        if (!first) {
+          builder.append(', ')
+        }
+        first = false
+      }
+    }
+    builder.append(')')
+  }
+  return builder.build()
+}