Skip to content

Add writer schema request copy and response import tools #208

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
eviltester merged 2 commits into from
Jun 14, 2026
Merged

Add writer schema request copy and response import tools #208

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
d81b212
Add writer schema request copy and response import
eviltester Jun 14, 2026
1255002
Add copy buttons for Chrome flag URLs
eviltester Jun 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
152 changes: 149 additions & 3 deletions apps/web/src/tests/jest/writer-schema-page.test.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,19 @@ import {

describe('writer schema prototype page', () => {
let dom;
let navigatorObj;

beforeEach(() => {
dom = new JSDOM(
`<!doctype html><html><body>
<div class="header" data-role="theme-toggle-host"><div class="pageheading">AnyWayData</div></div>
<main id="writer-schema-page-root">
<p id="writer-schema-support-status">Checking Writer API availability...</p>
<button
id="writer-schema-copy-flag"
type="button"
data-copy-text="chrome://flags/#writer-api-for-gemini-nano"
></button>
<textarea id="writer-schema-prompt"></textarea>
<button id="writer-schema-example-prompt" type="button">Load example prompt</button>
<button id="writer-schema-generate" type="button">Generate schema from prompt</button>
Expand All @@ -28,11 +34,20 @@ describe('writer schema prototype page', () => {
<pre id="writer-schema-raw-output">No raw Writer response yet.</pre>
<pre id="writer-schema-error-output">No errors yet.</pre>
<ol id="writer-schema-progress-output"><li>No generation activity yet.</li></ol>
<button id="writer-schema-copy-request" type="button"></button>
<button id="writer-schema-copy-prompt" type="button"></button>
<textarea id="writer-schema-process-response"></textarea>
<button id="writer-schema-create-schema" type="button">Create Schema</button>
<div id="writer-schema-editor-root"></div>
</main>
</body></html>`,
{ url: 'https://example.test/writer-schema.html' }
);
navigatorObj = {
clipboard: {
writeText: jest.fn(async () => {}),
},
};
});

afterEach(() => {
Expand Down Expand Up @@ -170,6 +185,57 @@ describe('writer schema prototype page', () => {
expect(writer.destroy).toHaveBeenCalledTimes(1);
});

test('runWriterSchemaGeneration accumulates streaming Writer chunks into structured output', async () => {
const writer = {
destroy: jest.fn(),
writeStreaming: jest.fn(async function* () {
yield '{"schemaFields":[';
yield '{"name":"Book Title","sourceType":"domain","command":"commerce.productName"},';
yield '{"name":"Genre","sourceType":"enum","values":["Fiction","Non-fiction"]}';
yield ']}';
}),
};
const WriterCtor = {
create: jest.fn(async () => writer),
};

const result = await runWriterSchemaGeneration({
WriterCtor,
promptText: DEFAULT_PROMPT,
domainCommands: ['commerce.productName', 'person.fullName'],
sampleSchemaText: 'Name\nperson.fullName',
onStatus: jest.fn(),
});

expect(WriterCtor.create).toHaveBeenCalledTimes(1);
expect(writer.writeStreaming).toHaveBeenCalledWith(
DEFAULT_PROMPT,
expect.objectContaining({
context: expect.any(String),
expectedInputLanguages: ['en'],
expectedContextLanguages: ['en'],
outputLanguage: 'en',
})
);
expect(result.parsedPayload.schemaFields).toHaveLength(2);
expect(result.requestDetails).toMatchObject({
promptText: DEFAULT_PROMPT,
taskContext: expect.any(String),
writeOptions: expect.objectContaining({
outputLanguage: 'en',
}),
createOptions: expect.objectContaining({
sharedContext: expect.any(String),
}),
});
expect(result.schemaRows).toMatchObject([
{ name: 'Book Title', sourceType: 'domain', command: 'commerce.productName' },
{ name: 'Genre', sourceType: 'enum', value: '"Fiction","Non-fiction"' },
]);
expect(result.normalizationErrors).toEqual([]);
expect(writer.destroy).toHaveBeenCalledTimes(1);
});

test('bootstrap warns when Writer API support is unavailable', async () => {
const schemaComponent = {
destroy: jest.fn(),
Expand All @@ -184,6 +250,7 @@ describe('writer schema prototype page', () => {
await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor: null,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});
Expand All @@ -194,6 +261,34 @@ describe('writer schema prototype page', () => {
expect(dom.window.document.getElementById('writer-schema-prompt').value).toBe(DEFAULT_PROMPT);
});

test('bootstrap copies setup flag URLs to the clipboard', async () => {
const schemaComponent = {
destroy: jest.fn(),
replaceRows: jest.fn(),
setTextMode: jest.fn(),
render: jest.fn(),
syncTextFromRows: jest.fn(),
validateRows: jest.fn(() => ({ errors: [] })),
getSchemaText: jest.fn(() => ''),
};

await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor: null,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});

dom.window.document.getElementById('writer-schema-copy-flag').click();
await new Promise((resolve) => setTimeout(resolve, 0));

expect(navigatorObj.clipboard.writeText).toHaveBeenCalledWith('chrome://flags/#writer-api-for-gemini-nano');
expect(dom.window.document.getElementById('writer-schema-generation-status').textContent).toContain(
'Copied the Chrome flags URL to the clipboard.'
);
});

test('bootstrap generates rows and populates the shared schema component', async () => {
const schemaComponent = {
destroy: jest.fn(),
Expand All @@ -219,6 +314,7 @@ describe('writer schema prototype page', () => {
const page = await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});
Expand All @@ -233,7 +329,7 @@ describe('writer schema prototype page', () => {
);
expect(schemaComponent.syncTextFromRows).toHaveBeenCalledTimes(1);
expect(dom.window.document.getElementById('writer-schema-generation-status').textContent).toContain(
'Generated 1 schema fields'
'Processed Writer API output into 1 schema fields.'
);
expect(dom.window.document.getElementById('writer-schema-json-output').textContent).toContain('Book Title');
expect(dom.window.document.getElementById('writer-schema-request-output').textContent).toContain(
Expand All @@ -244,7 +340,19 @@ describe('writer schema prototype page', () => {
);
expect(dom.window.document.getElementById('writer-schema-error-output').textContent).toBe('No errors yet.');
expect(dom.window.document.getElementById('writer-schema-progress-output').textContent).toContain(
'Schema generation completed successfully.'
'Processed Writer API output successfully.'
);

await page.copyLatestRequestJson();
await page.copyLatestRequestAsPrompt();

expect(navigatorObj.clipboard.writeText).toHaveBeenNthCalledWith(
1,
expect.stringContaining('"promptText": "Create 10 fields that represent the inventory of a bookshop"')
);
expect(navigatorObj.clipboard.writeText).toHaveBeenNthCalledWith(
2,
expect.stringContaining('Generate an AnyWayData schema response using the following request details.')
);
});

Expand Down Expand Up @@ -272,6 +380,7 @@ describe('writer schema prototype page', () => {
const page = await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});
Expand Down Expand Up @@ -325,6 +434,7 @@ describe('writer schema prototype page', () => {
const page = await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});
Expand All @@ -341,7 +451,43 @@ describe('writer schema prototype page', () => {
'unsupported command "commerce.publisher"'
);
expect(dom.window.document.getElementById('writer-schema-progress-output').textContent).toContain(
'Completed with partial recovery.'
'Processed Writer API output with partial recovery.'
);
});

test('bootstrap processes a pasted AI response into schema rows', async () => {
const schemaComponent = {
destroy: jest.fn(),
replaceRows: jest.fn(),
setTextMode: jest.fn(),
render: jest.fn(),
syncTextFromRows: jest.fn(),
validateRows: jest.fn(() => ({ errors: [] })),
getSchemaText: jest.fn(() => 'Book Title\ncommerce.productName()'),
};

const page = await bootstrapWriterSchemaPage({
documentObj: dom.window.document,
WriterCtor: null,
navigatorObj,
createThemeToggleComponentFn: () => ({ destroy: jest.fn() }),
createSharedSchemaDefinitionComponentFn: () => schemaComponent,
});

dom.window.document.getElementById('writer-schema-process-response').value = JSON.stringify({
schemaFields: [{ name: 'Book Title', sourceType: 'domain', command: 'commerce.productName' }],
});

await page.createSchemaFromResponse();

expect(schemaComponent.replaceRows).toHaveBeenCalledWith(
expect.arrayContaining([expect.objectContaining({ name: 'Book Title', command: 'commerce.productName' })])
);
expect(dom.window.document.getElementById('writer-schema-generation-status').textContent).toContain(
'Processed pasted AI response into 1 schema fields.'
);
expect(dom.window.document.getElementById('writer-schema-raw-output').textContent).toContain(
'commerce.productName'
);
});
});
Comment thread
eviltester marked this conversation as resolved.
Loading