diff options
Diffstat (limited to '')
-rw-r--r-- | dom/base/test/browser_blocking_image.js | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/dom/base/test/browser_blocking_image.js b/dom/base/test/browser_blocking_image.js new file mode 100644 index 0000000000..5749937f07 --- /dev/null +++ b/dom/base/test/browser_blocking_image.js @@ -0,0 +1,183 @@ +const TEST_URI = + getRootDirectory(gTestPath).replace( + "chrome://mochitests/content", + "https://example.com" + ) + "file_blocking_image.html"; + +/** + * Loading an image from https:// should work. + */ +add_task(async function load_image_from_https_test() { + let tab = BrowserTestUtils.addTab(gBrowser, TEST_URI); + await BrowserTestUtils.browserLoaded(tab.linkedBrowser); + + gBrowser.selectedTab = tab; + + await SpecialPowers.spawn(tab.linkedBrowser, [], async function () { + function imgListener(img) { + return new Promise((resolve, reject) => { + img.addEventListener("load", () => resolve()); + img.addEventListener("error", () => reject()); + }); + } + + let img = content.document.createElement("img"); + img.src = "https://example.com/tests/image/test/mochitest/shaver.png"; + content.document.body.appendChild(img); + + try { + await imgListener(img); + Assert.ok(true); + } catch (e) { + Assert.ok(false); + } + }); + + gBrowser.removeTab(tab); +}); + +/** + * Loading an image from http:// should be rejected. + */ +add_task(async function load_image_from_http_test() { + let tab = BrowserTestUtils.addTab(gBrowser, TEST_URI); + await BrowserTestUtils.browserLoaded(tab.linkedBrowser); + + gBrowser.selectedTab = tab; + + await SpecialPowers.spawn(tab.linkedBrowser, [], async function () { + function imgListener(img) { + return new Promise((resolve, reject) => { + img.addEventListener("load", () => reject()); + img.addEventListener("error", () => resolve()); + }); + } + + let img = content.document.createElement("img"); + img.src = "http://example.com/tests/image/test/mochitest/shaver.png"; + content.document.body.appendChild(img); + + try { + await imgListener(img); + Assert.ok(true); + } catch (e) { + Assert.ok(false); + } + }); + + gBrowser.removeTab(tab); +}); + +/** + * Loading an image from http:// immediately after loading from https:// + * The load from https:// should be replaced. + */ +add_task(async function load_https_and_http_test() { + let tab = BrowserTestUtils.addTab(gBrowser, TEST_URI); + await BrowserTestUtils.browserLoaded(tab.linkedBrowser); + + // Clear image cache, otherwise in non-e10s mode the image might be cached by + // previous test, and make the image from https is loaded immediately. + let imgTools = Cc["@mozilla.org/image/tools;1"].getService(Ci.imgITools); + let imageCache = imgTools.getImgCacheForDocument(window.document); + imageCache.clearCache(false); // false=content + + gBrowser.selectedTab = tab; + + await SpecialPowers.spawn(tab.linkedBrowser, [], async function () { + function imgListener(img) { + return new Promise((resolve, reject) => { + img.addEventListener("load", () => reject()); + img.addEventListener("error", () => resolve()); + }); + } + + let img = content.document.createElement("img"); + img.src = "https://example.com/tests/image/test/mochitest/shaver.png"; + img.src = "http://example.com/tests/image/test/mochitest/shaver.png"; + + content.document.body.appendChild(img); + + try { + await imgListener(img); + Assert.ok(true); + } catch (e) { + Assert.ok(false); + } + }); + + gBrowser.removeTab(tab); +}); + +/** + * Loading an image from https. + * Then after we have size information of the image, we immediately change the + * location to a http:// site (hence should be blocked by CSP). + */ +add_task(async function block_pending_request_test() { + let tab = BrowserTestUtils.addTab(gBrowser, TEST_URI); + await BrowserTestUtils.browserLoaded(tab.linkedBrowser); + + gBrowser.selectedTab = tab; + + await SpecialPowers.spawn(tab.linkedBrowser, [], async function () { + let wrapper = { + _resolve: null, + _sizeAvail: false, + + sizeAvailable(request) { + // In non-e10s mode the image may have already been cached, so sometimes + // sizeAvailable() is called earlier then waitUntilSizeAvailable(). + if (this._resolve) { + this._resolve(); + } else { + this._sizeAvail = true; + } + }, + + waitUntilSizeAvailable() { + return new Promise(resolve => { + this._resolve = resolve; + if (this._sizeAvail) { + resolve(); + } + }); + }, + + QueryInterface(aIID) { + if (aIID.equals(Ci.imgIScriptedNotificationObserver)) { + return this; + } + throw Components.Exception("", Cr.NS_ERROR_NO_INTERFACE); + }, + }; + + let observer = Cc["@mozilla.org/image/tools;1"] + .getService(Ci.imgITools) + .createScriptedObserver(wrapper); + + let img = content.document.createElement("img"); + img.src = "https://example.com/tests/image/test/mochitest/shaver.png"; + + let req = img.getRequest(Ci.nsIImageLoadingContent.CURRENT_REQUEST); + img.addObserver(observer); + + content.document.body.appendChild(img); + + info("Wait until Size Available"); + await wrapper.waitUntilSizeAvailable(); + info("Size Available now!"); + img.removeObserver(observer); + + // Now we change to load from http:// site, which will be blocked. + img.src = "http://example.com/tests/image/test/mochitest/shaver.png"; + + Assert.equal( + img.getRequest(Ci.nsIImageLoadingContent.CURRENT_REQUEST), + req, + "CURRENT_REQUEST shouldn't be replaced." + ); + }); + + gBrowser.removeTab(tab); +}); |