1/** 2 * Copyright 2018 Google Inc. All rights reserved. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17import expect from 'expect'; 18import { 19 getTestState, 20 setupTestBrowserHooks, 21 setupTestPageAndContextHooks, 22 itFailsFirefox, 23} from './mocha-utils'; // eslint-disable-line import/extensions 24 25describe('Screenshots', function () { 26 setupTestBrowserHooks(); 27 setupTestPageAndContextHooks(); 28 29 describe('Page.screenshot', function () { 30 it('should work', async () => { 31 const { page, server } = getTestState(); 32 33 await page.setViewport({ width: 500, height: 500 }); 34 await page.goto(server.PREFIX + '/grid.html'); 35 const screenshot = await page.screenshot(); 36 expect(screenshot).toBeGolden('screenshot-sanity.png'); 37 }); 38 it('should clip rect', async () => { 39 const { page, server } = getTestState(); 40 41 await page.setViewport({ width: 500, height: 500 }); 42 await page.goto(server.PREFIX + '/grid.html'); 43 const screenshot = await page.screenshot({ 44 clip: { 45 x: 50, 46 y: 100, 47 width: 150, 48 height: 100, 49 }, 50 }); 51 expect(screenshot).toBeGolden('screenshot-clip-rect.png'); 52 }); 53 it( 54 'should get screenshot bigger than the viewport', 55 async () => { 56 const { page, server } = getTestState(); 57 await page.setViewport({ width: 50, height: 50 }); 58 await page.goto(server.PREFIX + '/grid.html'); 59 const screenshot = await page.screenshot({ 60 clip: { 61 x: 25, 62 y: 25, 63 width: 100, 64 height: 100, 65 }, 66 }); 67 expect(screenshot).toBeGolden('screenshot-offscreen-clip.png'); 68 } 69 ); 70 it('should run in parallel', async () => { 71 const { page, server } = getTestState(); 72 73 await page.setViewport({ width: 500, height: 500 }); 74 await page.goto(server.PREFIX + '/grid.html'); 75 const promises = []; 76 for (let i = 0; i < 3; ++i) { 77 promises.push( 78 page.screenshot({ 79 clip: { 80 x: 50 * i, 81 y: 0, 82 width: 50, 83 height: 50, 84 }, 85 }) 86 ); 87 } 88 const screenshots = await Promise.all(promises); 89 expect(screenshots[1]).toBeGolden('grid-cell-1.png'); 90 }); 91 it('should take fullPage screenshots', async () => { 92 const { page, server } = getTestState(); 93 94 await page.setViewport({ width: 500, height: 500 }); 95 await page.goto(server.PREFIX + '/grid.html'); 96 const screenshot = await page.screenshot({ 97 fullPage: true, 98 }); 99 expect(screenshot).toBeGolden('screenshot-grid-fullpage.png'); 100 }); 101 it('should run in parallel in multiple pages', async () => { 102 const { server, context } = getTestState(); 103 104 const N = 2; 105 const pages = await Promise.all( 106 Array(N) 107 .fill(0) 108 .map(async () => { 109 const page = await context.newPage(); 110 await page.goto(server.PREFIX + '/grid.html'); 111 return page; 112 }) 113 ); 114 const promises = []; 115 for (let i = 0; i < N; ++i) 116 promises.push( 117 pages[i].screenshot({ 118 clip: { x: 50 * i, y: 0, width: 50, height: 50 }, 119 }) 120 ); 121 const screenshots = await Promise.all(promises); 122 for (let i = 0; i < N; ++i) 123 expect(screenshots[i]).toBeGolden(`grid-cell-${i}.png`); 124 await Promise.all(pages.map((page) => page.close())); 125 }); 126 it('should allow transparency', async () => { 127 const { page, server } = getTestState(); 128 129 await page.setViewport({ width: 100, height: 100 }); 130 await page.goto(server.EMPTY_PAGE); 131 const screenshot = await page.screenshot({ omitBackground: true }); 132 expect(screenshot).toBeGolden('transparent.png'); 133 }); 134 it('should render white background on jpeg file', async () => { 135 const { page, server } = getTestState(); 136 137 await page.setViewport({ width: 100, height: 100 }); 138 await page.goto(server.EMPTY_PAGE); 139 const screenshot = await page.screenshot({ 140 omitBackground: true, 141 type: 'jpeg', 142 }); 143 expect(screenshot).toBeGolden('white.jpg'); 144 }); 145 it('should work with odd clip size on Retina displays', async () => { 146 const { page } = getTestState(); 147 148 const screenshot = await page.screenshot({ 149 clip: { 150 x: 0, 151 y: 0, 152 width: 11, 153 height: 11, 154 }, 155 }); 156 expect(screenshot).toBeGolden('screenshot-clip-odd-size.png'); 157 }); 158 it('should return base64', async () => { 159 const { page, server } = getTestState(); 160 161 await page.setViewport({ width: 500, height: 500 }); 162 await page.goto(server.PREFIX + '/grid.html'); 163 const screenshot = await page.screenshot({ 164 encoding: 'base64', 165 }); 166 // TODO (@jackfranklin): improve the screenshot types. 167 // - if we pass encoding: 'base64', it returns a string 168 // - else it returns a buffer. 169 // If we can fix that we can avoid this "as string" here. 170 expect(Buffer.from(screenshot as string, 'base64')).toBeGolden( 171 'screenshot-sanity.png' 172 ); 173 }); 174 }); 175 176 describe('ElementHandle.screenshot', function () { 177 it('should work', async () => { 178 const { page, server } = getTestState(); 179 180 await page.setViewport({ width: 500, height: 500 }); 181 await page.goto(server.PREFIX + '/grid.html'); 182 await page.evaluate(() => window.scrollBy(50, 100)); 183 const elementHandle = await page.$('.box:nth-of-type(3)'); 184 const screenshot = await elementHandle.screenshot(); 185 expect(screenshot).toBeGolden('screenshot-element-bounding-box.png'); 186 }); 187 it('should take into account padding and border', async () => { 188 const { page } = getTestState(); 189 190 await page.setViewport({ width: 500, height: 500 }); 191 await page.setContent(` 192 something above 193 <style>div { 194 border: 2px solid blue; 195 background: green; 196 width: 50px; 197 height: 50px; 198 } 199 </style> 200 <div></div> 201 `); 202 const elementHandle = await page.$('div'); 203 const screenshot = await elementHandle.screenshot(); 204 expect(screenshot).toBeGolden('screenshot-element-padding-border.png'); 205 }); 206 it( 207 'should capture full element when larger than viewport', 208 async () => { 209 const { page } = getTestState(); 210 211 await page.setViewport({ width: 500, height: 500 }); 212 213 await page.setContent(` 214 something above 215 <style> 216 div.to-screenshot { 217 border: 1px solid blue; 218 width: 600px; 219 height: 600px; 220 margin-left: 50px; 221 } 222 ::-webkit-scrollbar{ 223 display: none; 224 } 225 </style> 226 <div class="to-screenshot"></div> 227 `); 228 const elementHandle = await page.$('div.to-screenshot'); 229 const screenshot = await elementHandle.screenshot(); 230 expect(screenshot).toBeGolden( 231 'screenshot-element-larger-than-viewport.png' 232 ); 233 234 expect( 235 await page.evaluate(() => ({ 236 w: window.innerWidth, 237 h: window.innerHeight, 238 })) 239 ).toEqual({ w: 500, h: 500 }); 240 } 241 ); 242 it('should scroll element into view', async () => { 243 const { page } = getTestState(); 244 245 await page.setViewport({ width: 500, height: 500 }); 246 await page.setContent(` 247 something above 248 <style>div.above { 249 border: 2px solid blue; 250 background: red; 251 height: 1500px; 252 } 253 div.to-screenshot { 254 border: 2px solid blue; 255 background: green; 256 width: 50px; 257 height: 50px; 258 } 259 </style> 260 <div class="above"></div> 261 <div class="to-screenshot"></div> 262 `); 263 const elementHandle = await page.$('div.to-screenshot'); 264 const screenshot = await elementHandle.screenshot(); 265 expect(screenshot).toBeGolden( 266 'screenshot-element-scrolled-into-view.png' 267 ); 268 }); 269 it('should work with a rotated element', async () => { 270 const { page } = getTestState(); 271 272 await page.setViewport({ width: 500, height: 500 }); 273 await page.setContent(`<div style="position:absolute; 274 top: 100px; 275 left: 100px; 276 width: 100px; 277 height: 100px; 278 background: green; 279 transform: rotateZ(200deg);"> </div>`); 280 const elementHandle = await page.$('div'); 281 const screenshot = await elementHandle.screenshot(); 282 expect(screenshot).toBeGolden('screenshot-element-rotate.png'); 283 }); 284 it('should fail to screenshot a detached element', async () => { 285 const { page } = getTestState(); 286 287 await page.setContent('<h1>remove this</h1>'); 288 const elementHandle = await page.$('h1'); 289 await page.evaluate( 290 (element: HTMLElement) => element.remove(), 291 elementHandle 292 ); 293 const screenshotError = await elementHandle 294 .screenshot() 295 .catch((error) => error); 296 expect(screenshotError.message).toBe( 297 'Node is either not visible or not an HTMLElement' 298 ); 299 }); 300 it('should not hang with zero width/height element', async () => { 301 const { page } = getTestState(); 302 303 await page.setContent('<div style="width: 50px; height: 0"></div>'); 304 const div = await page.$('div'); 305 const error = await div.screenshot().catch((error_) => error_); 306 expect(error.message).toBe('Node has 0 height.'); 307 }); 308 it('should work for an element with fractional dimensions', async () => { 309 const { page } = getTestState(); 310 311 await page.setContent( 312 '<div style="width:48.51px;height:19.8px;border:1px solid black;"></div>' 313 ); 314 const elementHandle = await page.$('div'); 315 const screenshot = await elementHandle.screenshot(); 316 expect(screenshot).toBeGolden('screenshot-element-fractional.png'); 317 }); 318 it('should work for an element with an offset', async () => { 319 const { page } = getTestState(); 320 321 await page.setContent( 322 '<div style="position:absolute; top: 10.3px; left: 20.4px;width:50.3px;height:20.2px;border:1px solid black;"></div>' 323 ); 324 const elementHandle = await page.$('div'); 325 const screenshot = await elementHandle.screenshot(); 326 expect(screenshot).toBeGolden('screenshot-element-fractional-offset.png'); 327 }); 328 }); 329}); 330