forked from mirrors/gecko-dev
		
	
		
			
				
	
	
		
			89 lines
		
	
	
	
		
			2.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			89 lines
		
	
	
	
		
			2.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| /* Any copyright is dedicated to the Public Domain.
 | |
|  * http://creativecommons.org/publicdomain/zero/1.0/ */
 | |
| 
 | |
| "use strict";
 | |
| 
 | |
| const URL = getRootDirectory(gTestPath) + "browser_pageStyle_sample.html";
 | |
| const URL_NESTED = getRootDirectory(gTestPath) + "browser_pageStyle_sample_nested.html";
 | |
| 
 | |
| /**
 | |
|  * This test ensures that page style information is correctly persisted.
 | |
|  */
 | |
| add_task(function page_style() {
 | |
|   let tab = gBrowser.addTab(URL);
 | |
|   let browser = tab.linkedBrowser;
 | |
|   yield promiseBrowserLoaded(browser);
 | |
|   let sheets = yield getStyleSheets(browser);
 | |
| 
 | |
|   // Enable all style sheets one by one.
 | |
|   for (let [title, disabled] of sheets) {
 | |
|     yield enableStyleSheetsForSet(browser, title);
 | |
| 
 | |
|     let tab2 = gBrowser.duplicateTab(tab);
 | |
|     yield promiseTabRestored(tab2);
 | |
| 
 | |
|     let sheets = yield getStyleSheets(tab2.linkedBrowser);
 | |
|     let enabled = sheets.filter(([title, disabled]) => !disabled);
 | |
| 
 | |
|     if (title.startsWith("fail_")) {
 | |
|       ok(!enabled.length, "didn't restore " + title);
 | |
|     } else {
 | |
|       is(enabled.length, 1, "restored one style sheet");
 | |
|       is(enabled[0][0], title, "restored correct sheet");
 | |
|     }
 | |
| 
 | |
|     gBrowser.removeTab(tab2);
 | |
|   }
 | |
| 
 | |
|   // Disable all styles and verify that this is correctly persisted.
 | |
|   yield setAuthorStyleDisabled(browser, true);
 | |
| 
 | |
|   let tab2 = gBrowser.duplicateTab(tab);
 | |
|   yield promiseTabRestored(tab2);
 | |
| 
 | |
|   let authorStyleDisabled = yield getAuthorStyleDisabled(tab2.linkedBrowser);
 | |
|   ok(authorStyleDisabled, "disabled all stylesheets");
 | |
| 
 | |
|   // Clean up.
 | |
|   gBrowser.removeTab(tab);
 | |
|   gBrowser.removeTab(tab2);
 | |
| });
 | |
| 
 | |
| /**
 | |
|  * This test ensures that page style notification from nested documents are
 | |
|  * received and the page style is persisted correctly.
 | |
|  */
 | |
| add_task(function nested_page_style() {
 | |
|   let tab = gBrowser.addTab(URL_NESTED);
 | |
|   let browser = tab.linkedBrowser;
 | |
|   yield promiseBrowserLoaded(browser);
 | |
| 
 | |
|   yield enableSubDocumentStyleSheetsForSet(browser, "alternate");
 | |
|   yield promiseRemoveTab(tab);
 | |
| 
 | |
|   let [{state: {pageStyle}}] = JSON.parse(ss.getClosedTabData(window));
 | |
|   let expected = JSON.stringify({children: [{pageStyle: "alternate"}]});
 | |
|   is(JSON.stringify(pageStyle), expected, "correct pageStyle persisted");
 | |
| });
 | |
| 
 | |
| function getStyleSheets(browser) {
 | |
|   return sendMessage(browser, "ss-test:getStyleSheets");
 | |
| }
 | |
| 
 | |
| function enableStyleSheetsForSet(browser, name) {
 | |
|   return sendMessage(browser, "ss-test:enableStyleSheetsForSet", name);
 | |
| }
 | |
| 
 | |
| function enableSubDocumentStyleSheetsForSet(browser, name) {
 | |
|   return sendMessage(browser, "ss-test:enableSubDocumentStyleSheetsForSet", {
 | |
|     id: "iframe", set: name
 | |
|   });
 | |
| }
 | |
| 
 | |
| function getAuthorStyleDisabled(browser) {
 | |
|   return sendMessage(browser, "ss-test:getAuthorStyleDisabled");
 | |
| }
 | |
| 
 | |
| function setAuthorStyleDisabled(browser, val) {
 | |
|   return sendMessage(browser, "ss-test:setAuthorStyleDisabled", val)
 | |
| }
 | 
