snapshotFormatter.test.ts•3.76 kB
/**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import {formatA11ySnapshot} from '../../src/formatters/snapshotFormatter.js';
import assert from 'node:assert';
import {describe, it} from 'node:test';
import type {TextSnapshotNode} from '../../src/McpContext.js';
import type {ElementHandle} from 'puppeteer-core';
describe('snapshotFormatter', () => {
it('formats a snapshot with value properties', () => {
const snapshot: TextSnapshotNode = {
id: '1_1',
role: 'textbox',
name: 'textbox',
value: 'value',
children: [
{
id: '1_2',
role: 'statictext',
name: 'text',
children: [],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
},
],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
};
const formatted = formatA11ySnapshot(snapshot);
assert.strictEqual(
formatted,
`uid=1_1 textbox "textbox" value="value"
uid=1_2 statictext "text"
`,
);
});
it('formats a snapshot with boolean properties', () => {
const snapshot: TextSnapshotNode = {
id: '1_1',
role: 'button',
name: 'button',
disabled: true,
children: [
{
id: '1_2',
role: 'statictext',
name: 'text',
children: [],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
},
],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
};
const formatted = formatA11ySnapshot(snapshot);
assert.strictEqual(
formatted,
`uid=1_1 button "button" disableable disabled
uid=1_2 statictext "text"
`,
);
});
it('formats a snapshot with checked properties', () => {
const snapshot: TextSnapshotNode = {
id: '1_1',
role: 'checkbox',
name: 'checkbox',
checked: true,
children: [
{
id: '1_2',
role: 'statictext',
name: 'text',
children: [],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
},
],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
};
const formatted = formatA11ySnapshot(snapshot);
assert.strictEqual(
formatted,
`uid=1_1 checkbox "checkbox" checked checked="true"
uid=1_2 statictext "text"
`,
);
});
it('formats a snapshot with multiple different type attributes', () => {
const snapshot: TextSnapshotNode = {
id: '1_1',
role: 'root',
name: 'root',
children: [
{
id: '1_2',
role: 'button',
name: 'button',
focused: true,
disabled: true,
children: [],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
},
{
id: '1_3',
role: 'textbox',
name: 'textbox',
value: 'value',
children: [],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
},
],
elementHandle: async (): Promise<ElementHandle<Element> | null> => {
return null;
},
};
const formatted = formatA11ySnapshot(snapshot);
assert.strictEqual(
formatted,
`uid=1_1 root "root"
uid=1_2 button "button" disableable disabled focusable focused
uid=1_3 textbox "textbox" value="value"
`,
);
});
});