electron/spec/api-native-image-spec.js

615 строки
23 KiB
JavaScript

'use strict'
const chai = require('chai')
const dirtyChai = require('dirty-chai')
const { nativeImage } = require('electron')
const path = require('path')
const { expect } = chai
chai.use(dirtyChai)
describe('nativeImage module', () => {
const ImageFormat = {
PNG: 'png',
JPEG: 'jpeg'
}
const images = [
{
filename: 'logo.png',
format: ImageFormat.PNG,
hasAlphaChannel: true,
hasDataUrl: false,
width: 538,
height: 190
},
{
dataUrl: 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAAC0lEQVQYlWNgAAIAAAUAAdafFs0AAAAASUVORK5CYII=',
filename: '1x1.png',
format: ImageFormat.PNG,
hasAlphaChannel: true,
hasDataUrl: true,
height: 1,
width: 1
},
{
dataUrl: 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAIAAAD91JpzAAAAFklEQVQYlWP8//8/AwMDEwMDAwMDAwAkBgMBBMzldwAAAABJRU5ErkJggg==',
filename: '2x2.jpg',
format: ImageFormat.JPEG,
hasAlphaChannel: false,
hasDataUrl: true,
height: 2,
width: 2
},
{
dataUrl: 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAYAAABWKLW/AAAADElEQVQYlWNgIAoAAAAnAAGZWEMnAAAAAElFTkSuQmCC',
filename: '3x3.png',
format: ImageFormat.PNG,
hasAlphaChannel: true,
hasDataUrl: true,
height: 3,
width: 3
}
]
/**
* @param {?string} filename
* @returns {?string} Full path.
*/
const getImagePathFromFilename = (filename) => {
return (filename === null) ? null
: path.join(__dirname, 'fixtures', 'assets', filename)
}
/**
* @param {!Object} image
* @param {Object} filters
* @returns {boolean}
*/
const imageMatchesTheFilters = (image, filters = null) => {
if (filters === null) {
return true
}
return Object.entries(filters)
.every(([key, value]) => image[key] === value)
}
/**
* @param {!Object} filters
* @returns {!Array} A matching images list.
*/
const getImages = (filters) => {
const matchingImages = images
.filter(i => imageMatchesTheFilters(i, filters))
// Add `.path` property to every image.
matchingImages
.forEach(i => { i.path = getImagePathFromFilename(i.filename) })
return matchingImages
}
/**
* @param {!Object} filters
* @returns {Object} A matching image if any.
*/
const getImage = (filters) => {
const matchingImages = getImages(filters)
let matchingImage = null
if (matchingImages.length > 0) {
matchingImage = matchingImages[0]
}
return matchingImage
}
describe('isMacTemplateImage property', () => {
before(function () {
if (process.platform !== 'darwin') this.skip()
})
it('returns whether the image is a template image', () => {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
expect(image.isMacTemplateImage).to.be.a('boolean')
expect(image.isTemplateImage).to.be.a('function')
expect(image.setTemplateImage).to.be.a('function')
})
it('correctly recognizes a template image', () => {
const templateImage = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo_Template.png'))
expect(templateImage.isMacTemplateImage).to.be.true()
})
it('sets a template image', function () {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
expect(image.isMacTemplateImage).to.be.false()
image.isMacTemplateImage = true
expect(image.isMacTemplateImage).to.be.true()
})
})
describe('createEmpty()', () => {
it('returns an empty image', () => {
const empty = nativeImage.createEmpty()
expect(empty.isEmpty()).to.be.true()
expect(empty.getAspectRatio()).to.equal(1)
expect(empty.toDataURL()).to.equal('data:image/png;base64,')
expect(empty.toDataURL({ scaleFactor: 2.0 })).to.equal('data:image/png;base64,')
expect(empty.getSize()).to.deep.equal({ width: 0, height: 0 })
expect(empty.getBitmap()).to.be.empty()
expect(empty.getBitmap({ scaleFactor: 2.0 })).to.be.empty()
expect(empty.toBitmap()).to.be.empty()
expect(empty.toBitmap({ scaleFactor: 2.0 })).to.be.empty()
expect(empty.toJPEG(100)).to.be.empty()
expect(empty.toPNG()).to.be.empty()
expect(empty.toPNG({ scaleFactor: 2.0 })).to.be.empty()
if (process.platform === 'darwin') {
expect(empty.getNativeHandle()).to.be.empty()
}
})
})
describe('createFromBitmap(buffer, options)', () => {
it('returns an empty image when the buffer is empty', () => {
expect(nativeImage.createFromBitmap(Buffer.from([]), { width: 0, height: 0 }).isEmpty()).to.be.true()
})
it('returns an image created from the given buffer', () => {
const imageA = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
const imageB = nativeImage.createFromBitmap(imageA.toBitmap(), imageA.getSize())
expect(imageB.getSize()).to.deep.equal({ width: 538, height: 190 })
const imageC = nativeImage.createFromBuffer(imageA.toBitmap(), { ...imageA.getSize(), scaleFactor: 2.0 })
expect(imageC.getSize()).to.deep.equal({ width: 269, height: 95 })
})
it('throws on invalid arguments', () => {
expect(() => nativeImage.createFromBitmap(null, {})).to.throw('buffer must be a node Buffer')
expect(() => nativeImage.createFromBitmap([12, 14, 124, 12], {})).to.throw('buffer must be a node Buffer')
expect(() => nativeImage.createFromBitmap(Buffer.from([]), {})).to.throw('width is required')
expect(() => nativeImage.createFromBitmap(Buffer.from([]), { width: 1 })).to.throw('height is required')
expect(() => nativeImage.createFromBitmap(Buffer.from([]), { width: 1, height: 1 })).to.throw('invalid buffer size')
})
})
describe('createFromBuffer(buffer, options)', () => {
it('returns an empty image when the buffer is empty', () => {
expect(nativeImage.createFromBuffer(Buffer.from([])).isEmpty()).to.be.true()
})
it('returns an empty image when the buffer is too small', () => {
const image = nativeImage.createFromBuffer(Buffer.from([1, 2, 3, 4]), { width: 100, height: 100 })
expect(image.isEmpty()).to.be.true()
})
it('returns an image created from the given buffer', () => {
const imageA = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
const imageB = nativeImage.createFromBuffer(imageA.toPNG())
expect(imageB.getSize()).to.deep.equal({ width: 538, height: 190 })
expect(imageA.toBitmap().equals(imageB.toBitmap())).to.be.true()
const imageC = nativeImage.createFromBuffer(imageA.toJPEG(100))
expect(imageC.getSize()).to.deep.equal({ width: 538, height: 190 })
const imageD = nativeImage.createFromBuffer(imageA.toBitmap(),
{ width: 538, height: 190 })
expect(imageD.getSize()).to.deep.equal({ width: 538, height: 190 })
const imageE = nativeImage.createFromBuffer(imageA.toBitmap(),
{ width: 100, height: 200 })
expect(imageE.getSize()).to.deep.equal({ width: 100, height: 200 })
const imageF = nativeImage.createFromBuffer(imageA.toBitmap())
expect(imageF.isEmpty()).to.be.true()
const imageG = nativeImage.createFromBuffer(imageA.toPNG(),
{ width: 100, height: 200 })
expect(imageG.getSize()).to.deep.equal({ width: 538, height: 190 })
const imageH = nativeImage.createFromBuffer(imageA.toJPEG(100),
{ width: 100, height: 200 })
expect(imageH.getSize()).to.deep.equal({ width: 538, height: 190 })
const imageI = nativeImage.createFromBuffer(imageA.toBitmap(),
{ width: 538, height: 190, scaleFactor: 2.0 })
expect(imageI.getSize()).to.deep.equal({ width: 269, height: 95 })
})
it('throws on invalid arguments', () => {
expect(() => nativeImage.createFromBuffer(null)).to.throw('buffer must be a node Buffer')
expect(() => nativeImage.createFromBuffer([12, 14, 124, 12])).to.throw('buffer must be a node Buffer')
})
})
describe('createFromDataURL(dataURL)', () => {
it('returns an empty image from the empty string', () => {
expect(nativeImage.createFromDataURL('').isEmpty()).to.be.true()
})
it('returns an image created from the given string', () => {
const imagesData = getImages({ hasDataUrl: true })
for (const imageData of imagesData) {
const imageFromPath = nativeImage.createFromPath(imageData.path)
const imageFromDataUrl = nativeImage.createFromDataURL(imageData.dataUrl)
expect(imageFromDataUrl.isEmpty()).to.be.false()
expect(imageFromDataUrl.getSize()).to.deep.equal(imageFromPath.getSize())
expect(imageFromDataUrl.toBitmap()).to.satisfy(
bitmap => imageFromPath.toBitmap().equals(bitmap))
expect(imageFromDataUrl.toDataURL()).to.equal(imageFromPath.toDataURL())
}
})
})
describe('toDataURL()', () => {
it('returns a PNG data URL', () => {
const imagesData = getImages({ hasDataUrl: true })
for (const imageData of imagesData) {
const imageFromPath = nativeImage.createFromPath(imageData.path)
const scaleFactors = [1.0, 2.0]
for (const scaleFactor of scaleFactors) {
expect(imageFromPath.toDataURL({ scaleFactor })).to.equal(imageData.dataUrl)
}
}
})
it('returns a data URL at 1x scale factor by default', () => {
const imageData = getImage({ filename: 'logo.png' })
const image = nativeImage.createFromPath(imageData.path)
const imageOne = nativeImage.createFromBuffer(image.toPNG(), {
width: image.getSize().width,
height: image.getSize().height,
scaleFactor: 2.0
})
expect(imageOne.getSize()).to.deep.equal(
{ width: imageData.width / 2, height: imageData.height / 2 })
const imageTwo = nativeImage.createFromDataURL(imageOne.toDataURL())
expect(imageTwo.getSize()).to.deep.equal(
{ width: imageData.width, height: imageData.height })
expect(imageOne.toBitmap().equals(imageTwo.toBitmap())).to.be.true()
})
it('supports a scale factor', () => {
const imageData = getImage({ filename: 'logo.png' })
const image = nativeImage.createFromPath(imageData.path)
const expectedSize = { width: imageData.width, height: imageData.height }
const imageFromDataUrlOne = nativeImage.createFromDataURL(
image.toDataURL({ scaleFactor: 1.0 }))
expect(imageFromDataUrlOne.getSize()).to.deep.equal(expectedSize)
const imageFromDataUrlTwo = nativeImage.createFromDataURL(
image.toDataURL({ scaleFactor: 2.0 }))
expect(imageFromDataUrlTwo.getSize()).to.deep.equal(expectedSize)
})
})
describe('toPNG()', () => {
it('returns a buffer at 1x scale factor by default', () => {
const imageData = getImage({ filename: 'logo.png' })
const imageA = nativeImage.createFromPath(imageData.path)
const imageB = nativeImage.createFromBuffer(imageA.toPNG(), {
width: imageA.getSize().width,
height: imageA.getSize().height,
scaleFactor: 2.0
})
expect(imageB.getSize()).to.deep.equal(
{ width: imageData.width / 2, height: imageData.height / 2 })
const imageC = nativeImage.createFromBuffer(imageB.toPNG())
expect(imageC.getSize()).to.deep.equal(
{ width: imageData.width, height: imageData.height })
expect(imageB.toBitmap().equals(imageC.toBitmap())).to.be.true()
})
it('supports a scale factor', () => {
const imageData = getImage({ filename: 'logo.png' })
const image = nativeImage.createFromPath(imageData.path)
const imageFromBufferOne = nativeImage.createFromBuffer(
image.toPNG({ scaleFactor: 1.0 }))
expect(imageFromBufferOne.getSize()).to.deep.equal(
{ width: imageData.width, height: imageData.height })
const imageFromBufferTwo = nativeImage.createFromBuffer(
image.toPNG({ scaleFactor: 2.0 }), { scaleFactor: 2.0 })
expect(imageFromBufferTwo.getSize()).to.deep.equal(
{ width: imageData.width / 2, height: imageData.height / 2 })
})
})
describe('createFromPath(path)', () => {
it('returns an empty image for invalid paths', () => {
expect(nativeImage.createFromPath('').isEmpty()).to.be.true()
expect(nativeImage.createFromPath('does-not-exist.png').isEmpty()).to.be.true()
expect(nativeImage.createFromPath('does-not-exist.ico').isEmpty()).to.be.true()
expect(nativeImage.createFromPath(__dirname).isEmpty()).to.be.true()
expect(nativeImage.createFromPath(__filename).isEmpty()).to.be.true()
})
it('loads images from paths relative to the current working directory', () => {
const imagePath = path.relative('.', path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
const image = nativeImage.createFromPath(imagePath)
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 538, height: 190 })
})
it('loads images from paths with `.` segments', () => {
const imagePath = `${path.join(__dirname, 'fixtures')}${path.sep}.${path.sep}${path.join('assets', 'logo.png')}`
const image = nativeImage.createFromPath(imagePath)
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 538, height: 190 })
})
it('loads images from paths with `..` segments', () => {
const imagePath = `${path.join(__dirname, 'fixtures', 'api')}${path.sep}..${path.sep}${path.join('assets', 'logo.png')}`
const image = nativeImage.createFromPath(imagePath)
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 538, height: 190 })
})
it('Gets an NSImage pointer on macOS', function () {
if (process.platform !== 'darwin') {
// FIXME(alexeykuzmin): Skip the test.
// this.skip()
return
}
const imagePath = `${path.join(__dirname, 'fixtures', 'api')}${path.sep}..${path.sep}${path.join('assets', 'logo.png')}`
const image = nativeImage.createFromPath(imagePath)
const nsimage = image.getNativeHandle()
expect(nsimage).to.have.lengthOf(8)
// If all bytes are null, that's Bad
const allBytesAreNotNull = nsimage.reduce((acc, x) => acc || (x !== 0), false)
expect(allBytesAreNotNull)
})
it('loads images from .ico files on Windows', function () {
if (process.platform !== 'win32') {
// FIXME(alexeykuzmin): Skip the test.
// this.skip()
return
}
const imagePath = path.join(__dirname, 'fixtures', 'assets', 'icon.ico')
const image = nativeImage.createFromPath(imagePath)
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 256, height: 256 })
})
})
describe('createFromNamedImage(name)', () => {
it('returns empty for invalid options', () => {
const image = nativeImage.createFromNamedImage('totally_not_real')
expect(image.isEmpty()).to.be.true()
})
it('returns empty on non-darwin platforms', function () {
if (process.platform === 'darwin') {
// FIXME(alexeykuzmin): Skip the test.
// this.skip()
return
}
const image = nativeImage.createFromNamedImage('NSActionTemplate')
expect(image.isEmpty()).to.be.true()
})
it('returns a valid image on darwin', function () {
if (process.platform !== 'darwin') {
// FIXME(alexeykuzmin): Skip the test.
// this.skip()
return
}
const image = nativeImage.createFromNamedImage('NSActionTemplate')
expect(image.isEmpty()).to.be.false()
})
it('returns allows an HSL shift for a valid image on darwin', function () {
if (process.platform !== 'darwin') {
// FIXME(alexeykuzmin): Skip the test.
// this.skip()
return
}
const image = nativeImage.createFromNamedImage('NSActionTemplate', [0.5, 0.2, 0.8])
expect(image.isEmpty()).to.be.false()
})
})
describe('resize(options)', () => {
it('returns a resized image', () => {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
for (const [resizeTo, expectedSize] of new Map([
[{}, { width: 538, height: 190 }],
[{ width: 269 }, { width: 269, height: 95 }],
[{ width: 600 }, { width: 600, height: 212 }],
[{ height: 95 }, { width: 269, height: 95 }],
[{ height: 200 }, { width: 566, height: 200 }],
[{ width: 80, height: 65 }, { width: 80, height: 65 }],
[{ width: 600, height: 200 }, { width: 600, height: 200 }],
[{ width: 0, height: 0 }, { width: 0, height: 0 }],
[{ width: -1, height: -1 }, { width: 0, height: 0 }]
])) {
const actualSize = image.resize(resizeTo).getSize()
expect(actualSize).to.deep.equal(expectedSize)
}
})
it('returns an empty image when called on an empty image', () => {
expect(nativeImage.createEmpty().resize({ width: 1, height: 1 }).isEmpty()).to.be.true()
expect(nativeImage.createEmpty().resize({ width: 0, height: 0 }).isEmpty()).to.be.true()
})
it('supports a quality option', () => {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
const good = image.resize({ width: 100, height: 100, quality: 'good' })
const better = image.resize({ width: 100, height: 100, quality: 'better' })
const best = image.resize({ width: 100, height: 100, quality: 'best' })
expect(good.toPNG()).to.have.lengthOf.at.most(better.toPNG().length)
expect(better.toPNG()).to.have.lengthOf.below(best.toPNG().length)
})
})
describe('crop(bounds)', () => {
it('returns an empty image when called on an empty image', () => {
expect(nativeImage.createEmpty().crop({ width: 1, height: 2, x: 0, y: 0 }).isEmpty()).to.be.true()
expect(nativeImage.createEmpty().crop({ width: 0, height: 0, x: 0, y: 0 }).isEmpty()).to.be.true()
})
it('returns an empty image when the bounds are invalid', () => {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
expect(image.crop({ width: 0, height: 0, x: 0, y: 0 }).isEmpty()).to.be.true()
expect(image.crop({ width: -1, height: 10, x: 0, y: 0 }).isEmpty()).to.be.true()
expect(image.crop({ width: 10, height: -35, x: 0, y: 0 }).isEmpty()).to.be.true()
expect(image.crop({ width: 100, height: 100, x: 1000, y: 1000 }).isEmpty()).to.be.true()
})
it('returns a cropped image', () => {
const image = nativeImage.createFromPath(path.join(__dirname, 'fixtures', 'assets', 'logo.png'))
const cropA = image.crop({ width: 25, height: 64, x: 0, y: 0 })
const cropB = image.crop({ width: 25, height: 64, x: 30, y: 40 })
expect(cropA.getSize()).to.deep.equal({ width: 25, height: 64 })
expect(cropB.getSize()).to.deep.equal({ width: 25, height: 64 })
expect(cropA.toPNG().equals(cropB.toPNG())).to.be.false()
})
})
describe('getAspectRatio()', () => {
it('returns an aspect ratio of an empty image', () => {
expect(nativeImage.createEmpty().getAspectRatio()).to.equal(1.0)
})
it('returns an aspect ratio of an image', () => {
const imageData = getImage({ filename: 'logo.png' })
// imageData.width / imageData.height = 2.831578947368421
const expectedAspectRatio = 2.8315789699554443
const image = nativeImage.createFromPath(imageData.path)
expect(image.getAspectRatio()).to.equal(expectedAspectRatio)
})
})
describe('addRepresentation()', () => {
it('does not add representation when the buffer is too small', () => {
const image = nativeImage.createEmpty()
image.addRepresentation({
buffer: Buffer.from([1, 2, 3, 4]),
width: 100,
height: 100
})
expect(image.isEmpty()).to.be.true()
})
it('supports adding a buffer representation for a scale factor', () => {
const image = nativeImage.createEmpty()
const imageDataOne = getImage({ width: 1, height: 1 })
image.addRepresentation({
scaleFactor: 1.0,
buffer: nativeImage.createFromPath(imageDataOne.path).toPNG()
})
const imageDataTwo = getImage({ width: 2, height: 2 })
image.addRepresentation({
scaleFactor: 2.0,
buffer: nativeImage.createFromPath(imageDataTwo.path).toPNG()
})
const imageDataThree = getImage({ width: 3, height: 3 })
image.addRepresentation({
scaleFactor: 3.0,
buffer: nativeImage.createFromPath(imageDataThree.path).toPNG()
})
image.addRepresentation({
scaleFactor: 4.0,
buffer: 'invalid'
})
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 1, height: 1 })
expect(image.toDataURL({ scaleFactor: 1.0 })).to.equal(imageDataOne.dataUrl)
expect(image.toDataURL({ scaleFactor: 2.0 })).to.equal(imageDataTwo.dataUrl)
expect(image.toDataURL({ scaleFactor: 3.0 })).to.equal(imageDataThree.dataUrl)
expect(image.toDataURL({ scaleFactor: 4.0 })).to.equal(imageDataThree.dataUrl)
})
it('supports adding a data URL representation for a scale factor', () => {
const image = nativeImage.createEmpty()
const imageDataOne = getImage({ width: 1, height: 1 })
image.addRepresentation({
scaleFactor: 1.0,
dataURL: imageDataOne.dataUrl
})
const imageDataTwo = getImage({ width: 2, height: 2 })
image.addRepresentation({
scaleFactor: 2.0,
dataURL: imageDataTwo.dataUrl
})
const imageDataThree = getImage({ width: 3, height: 3 })
image.addRepresentation({
scaleFactor: 3.0,
dataURL: imageDataThree.dataUrl
})
image.addRepresentation({
scaleFactor: 4.0,
dataURL: 'invalid'
})
expect(image.isEmpty()).to.be.false()
expect(image.getSize()).to.deep.equal({ width: 1, height: 1 })
expect(image.toDataURL({ scaleFactor: 1.0 })).to.equal(imageDataOne.dataUrl)
expect(image.toDataURL({ scaleFactor: 2.0 })).to.equal(imageDataTwo.dataUrl)
expect(image.toDataURL({ scaleFactor: 3.0 })).to.equal(imageDataThree.dataUrl)
expect(image.toDataURL({ scaleFactor: 4.0 })).to.equal(imageDataThree.dataUrl)
})
it('supports adding a representation to an existing image', () => {
const imageDataOne = getImage({ width: 1, height: 1 })
const image = nativeImage.createFromPath(imageDataOne.path)
const imageDataTwo = getImage({ width: 2, height: 2 })
image.addRepresentation({
scaleFactor: 2.0,
dataURL: imageDataTwo.dataUrl
})
const imageDataThree = getImage({ width: 3, height: 3 })
image.addRepresentation({
scaleFactor: 2.0,
dataURL: imageDataThree.dataUrl
})
expect(image.toDataURL({ scaleFactor: 1.0 })).to.equal(imageDataOne.dataUrl)
expect(image.toDataURL({ scaleFactor: 2.0 })).to.equal(imageDataTwo.dataUrl)
})
})
})