Revision control
Copy as Markdown
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
import Foundation
import WebKit
@testable import WebEngine
class MockWKContentScriptManager: NSObject, WKContentScriptManager {
var scripts = [String: WKContentScript]()
var addContentScriptCalled = 0
var addContentScriptToPageCalled = 0
var uninstallCalled = 0
var userContentControllerCalled = 0
var savedContentScriptNames = [String]()
var savedContentScriptPageNames = [String]()
func addContentScript(_ script: WKContentScript,
name: String,
forSession session: WKEngineSession) {
scripts[name] = script
savedContentScriptNames.append(name)
addContentScriptCalled += 1
}
func addContentScriptToPage(_ script: WKContentScript,
name: String,
forSession session: WKEngineSession) {
scripts[name] = script
savedContentScriptPageNames.append(name)
addContentScriptToPageCalled += 1
}
func uninstall(session: WKEngineSession) {
uninstallCalled += 1
}
func userContentController(_ userContentController: WKUserContentController,
didReceive message: WKScriptMessage) {
userContentControllerCalled += 1
}
}