chrome-devtools-frontend
Version:
Chrome DevTools UI
75 lines (69 loc) • 2.32 kB
text/typescript
// Copyright 2023 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
import * as Models from './models.js';
describe('ScreenshotUtils', () => {
async function generateImage(
width: number,
height: number,
): Promise<Models.ScreenshotStorage.Screenshot> {
const img = new Image(width, height);
const promise = new Promise(resolve => {
img.onload = resolve;
});
img.src = `data:image/svg+xml,%3Csvg viewBox='0 0 ${width} ${
height}' xmlns='http://www.w3.org/2000/svg'%3E%3Ccircle cx='50' cy='50' r='50'/%3E%3C/svg%3E`;
await promise;
const canvas = document.createElement('canvas');
canvas.width = width;
canvas.height = height;
const context = canvas.getContext('2d', {willReadFrequently: true});
if (!context) {
throw new Error('Could not create context.');
}
const bitmap = await createImageBitmap(img, {
resizeHeight: height,
resizeWidth: width,
});
context.drawImage(bitmap, 0, 0);
return canvas.toDataURL('image/png') as Models.ScreenshotStorage.Screenshot;
}
async function getScreenshotDimensions(
screenshot: Models.ScreenshotStorage.Screenshot,
): Promise<number[]> {
const tmp = new Image();
const promise = new Promise(resolve => {
tmp.onload = resolve;
});
tmp.src = screenshot;
await promise;
return [tmp.width, tmp.height];
}
it('can resize screenshots to be 160px wide and <= 240px high', async () => {
const {resizeScreenshot} = Models.ScreenshotUtils;
assert.deepEqual(
await getScreenshotDimensions(
await resizeScreenshot(await generateImage(400, 800)),
),
[160, 240],
);
assert.deepEqual(
await getScreenshotDimensions(
await resizeScreenshot(await generateImage(800, 400)),
),
[160, 80],
);
assert.deepEqual(
await getScreenshotDimensions(
await resizeScreenshot(await generateImage(80, 80)),
),
[160, 160],
);
assert.deepEqual(
await getScreenshotDimensions(
await resizeScreenshot(await generateImage(80, 320)),
),
[160, 240],
);
});
});