Add devices support

Author: Kikobeats

index.js

@@ -1,5 +1,6 @@
 'use strict'
 
+const devices = require('puppeteer/DeviceDescriptors')
 const createTempFile = require('create-temp-file2')
 const puppeteer = require('puppeteer')
 
@@ -13,12 +14,18 @@ async function getHTML (url, opts = {}) {
 }
 
 async function takeScreenshot (url, opts = {}) {
-  const { type = 'png' } = opts
+  const { type = 'png', device: deviceDescriptor } = opts
   const tempFile = createTempFile({ ext: `.${type}` })
   const { path } = tempFile
 
   const browser = await puppeteer.launch()
   const page = await browser.newPage()
+
+  if (deviceDescriptor) {
+    const device = devices[deviceDescriptor]
+    if (device) await page.emulate(device)
+  }
+
   await page.goto(url)
   await page.screenshot(Object.assign({ path, type }, opts))
   browser.close()

test/example-iphone.png

@@ -18,24 +18,38 @@ describe('frowser', () => {
   })
 
   describe('.takeScreenshot', () => {
-    it('png', () =>
-      frowser.takeScreenshot('http://example.com').then(tmpStream => {
-        should(fs.readFileSync(tmpStream.path)).be.eql(
-          fs.readFileSync('test/example.png')
-        )
-        should(path.extname(tmpStream.path)).be.equal('.png')
-        tmpStream.cleanupSync()
-      }))
-
-    it('jpeg', () =>
-      frowser
-        .takeScreenshot('http://example.com', { type: 'jpeg' })
-        .then(tmpStream => {
+    describe('format', () => {
+      it('png', () =>
+        frowser.takeScreenshot('http://example.com').then(tmpStream => {
           should(fs.readFileSync(tmpStream.path)).be.eql(
-            fs.readFileSync('test/example.jpeg')
+            fs.readFileSync('test/example.png')
           )
-          should(path.extname(tmpStream.path)).be.equal('.jpeg')
+          should(path.extname(tmpStream.path)).be.equal('.png')
           tmpStream.cleanupSync()
         }))
+
+      it('jpeg', () =>
+        frowser
+          .takeScreenshot('http://example.com', { type: 'jpeg' })
+          .then(tmpStream => {
+            should(fs.readFileSync(tmpStream.path)).be.eql(
+              fs.readFileSync('test/example.jpeg')
+            )
+            should(path.extname(tmpStream.path)).be.equal('.jpeg')
+            tmpStream.cleanupSync()
+          }))
+    })
+
+    describe('devices', () => {
+      it('iPhone 6', () =>
+        frowser
+          .takeScreenshot('http://example.com', { device: 'iPhone 6' })
+          .then(tmpStream => {
+            should(fs.readFileSync(tmpStream.path)).be.eql(
+              fs.readFileSync('test/example-iphone.png')
+            )
+            tmpStream.cleanupSync()
+          }))
+    })
   })
 })