1 files changed, 45 insertions, 0 deletions
diff --git a/addon-sdk/source/test/test-framescript-util.js b/addon-sdk/source/test/test-framescript-util.js
new file mode 100644
index 000000000..0a55bcbf6
--- /dev/null
+++ b/addon-sdk/source/test/test-framescript-util.js
@@ -0,0 +1,45 @@
+/* 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
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+"use strict";
+
+const {loadModule} = require("framescript/manager");
+const {withTab, receiveMessage} = require("./util");
+const {getBrowserForTab} = require("sdk/tabs/utils");
+
+exports["test windowToMessageManager"] = withTab(function*(assert, tab) {
+  const {messageManager} = getBrowserForTab(tab);
+
+  loadModule(messageManager,
+             require.resolve("./framescript-util/frame-script"),
+             true,
+             "onInit");
+
+  messageManager.sendAsyncMessage("framescript-util/window/request");
+
+  const response = yield receiveMessage(messageManager,
+                                        "framescript-util/window/response");
+
+  assert.deepEqual(response.data, {window: true},
+                   "got response");
+}, "data:text/html,<h1>Window to Message Manager</h1>");
+
+
+exports["test nodeToMessageManager"] = withTab(function*(assert, tab) {
+  const {messageManager} = getBrowserForTab(tab);
+
+  loadModule(messageManager,
+             require.resolve("./framescript-util/frame-script"),
+             true,
+             "onInit");
+
+  messageManager.sendAsyncMessage("framescript-util/node/request", "h1");
+
+  const response = yield receiveMessage(messageManager,
+                                        "framescript-util/node/response");
+
+  assert.deepEqual(response.data, {node: true},
+                   "got response");
+}, "data:text/html,<h1>Node to Message Manager</h1>");
+
+require("sdk/test").run(exports);