| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // API test for chrome.extension.pageCapture. | 5 // API test for chrome.extension.pageCapture. |
| 6 // browser_tests.exe --gtest_filter=ExtensionPageCaptureApiTest.* | 6 // browser_tests.exe --gtest_filter=ExtensionApiTest.PageCapture |
| 7 | 7 |
| 8 const assertEq = chrome.test.assertEq; | 8 const assertEq = chrome.test.assertEq; |
| 9 const assertTrue = chrome.test.assertTrue; | 9 const assertTrue = chrome.test.assertTrue; |
| 10 | 10 |
| 11 var testUrl = 'http://www.a.com:PORT' + | 11 var testUrl = 'http://www.a.com:PORT' + |
| 12 '/extensions/api_test/page_capture/google.html'; | 12 '/extensions/api_test/page_capture/google.html'; |
| 13 | 13 |
| 14 function waitForCurrentTabLoaded(callback) { | 14 function waitForCurrentTabLoaded(callback) { |
| 15 chrome.tabs.getSelected(null, function(tab) { | 15 chrome.tabs.getSelected(null, function(tab) { |
| 16 if (tab.status == "complete" && tab.url == testUrl) { | 16 if (tab.status == "complete" && tab.url == testUrl) { |
| 17 callback(); | 17 callback(); |
| 18 return; | 18 return; |
| 19 } | 19 } |
| 20 window.setTimeout(function() { waitForCurrentTabLoaded(callback); }, 100); | 20 window.setTimeout(function() { waitForCurrentTabLoaded(callback); }, 100); |
| 21 }); | 21 }); |
| 22 } | 22 } |
| 23 | 23 |
| 24 chrome.test.getConfig(function(config) { | 24 chrome.test.getConfig(function(config) { |
| 25 testUrl = testUrl.replace(/PORT/, config.testServer.port); | 25 testUrl = testUrl.replace(/PORT/, config.testServer.port); |
| 26 | 26 |
| 27 chrome.test.runTests([ | 27 chrome.test.runTests([ |
| 28 function saveAsMHTML() { | 28 function saveAsMHTML() { |
| 29 chrome.tabs.getSelected(null, function(tab) { | 29 chrome.tabs.getSelected(null, function(tab) { |
| 30 chrome.tabs.update(null, { "url": testUrl }); | 30 chrome.tabs.update(null, { "url": testUrl }); |
| 31 waitForCurrentTabLoaded(function() { | 31 waitForCurrentTabLoaded(function() { |
| 32 chrome.pageCapture.saveAsMHTML({ "tabId": tab.id }, | 32 chrome.pageCapture.saveAsMHTML({ "tabId": tab.id }, |
| 33 function(data) { | 33 function(data) { |
| 34 if (config.customArg == "REQUEST_DENIED") { | |
| 35 chrome.test.assertLastError("User denied request."); | |
| 36 chrome.test.notifyPass(); | |
| 37 return; | |
| 38 } | |
| 39 assertEq(undefined, chrome.runtime.lastError); | 34 assertEq(undefined, chrome.runtime.lastError); |
| 40 assertTrue(data != null); | 35 assertTrue(data != null); |
| 41 // It should contain few KBs of data. | 36 // It should contain few KBs of data. |
| 42 assertTrue(data.size > 100); | 37 assertTrue(data.size > 100); |
| 43 // Let's make sure it contains some well known strings. | 38 // Let's make sure it contains some well known strings. |
| 44 var reader = new FileReader(); | 39 var reader = new FileReader(); |
| 45 reader.onload = function(e) { | 40 reader.onload = function(e) { |
| 46 var text = e.target.result; | 41 var text = e.target.result; |
| 47 assertTrue(text.indexOf(testUrl) != -1); | 42 assertTrue(text.indexOf(testUrl) != -1); |
| 48 assertTrue(text.indexOf("logo.png") != -1); | 43 assertTrue(text.indexOf("logo.png") != -1); |
| 49 // Run the GC so the blob is deleted. | 44 // Run the GC so the blob is deleted. |
| 50 window.setTimeout(function() { window.gc(); }); | 45 window.setTimeout(function() { window.gc(); }); |
| 51 window.setTimeout(function() { chrome.test.notifyPass(); }, 0); | 46 window.setTimeout(function() { chrome.test.notifyPass(); }, 0); |
| 52 }; | 47 }; |
| 53 reader.readAsText(data); | 48 reader.readAsText(data); |
| 54 }); | 49 }); |
| 55 }); | 50 }); |
| 56 }); | 51 }); |
| 57 } | 52 } |
| 58 ]); | 53 ]); |
| 59 }); | 54 }); |
| 60 | 55 |
| OLD | NEW |