1 files changed, 29 insertions, 0 deletions
diff --git a/addon-sdk/source/examples/ui-button-apis/tests/test-main.js b/addon-sdk/source/examples/ui-button-apis/tests/test-main.js
new file mode 100644
index 000000000..49bdc863a
--- /dev/null
+++ b/addon-sdk/source/examples/ui-button-apis/tests/test-main.js
@@ -0,0 +1,29 @@
+/* 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";
+
+try {
+  // CFX use case..
+  var { actionButton, toggleButton, icon } = require("main");
+}
+catch (e) {
+  // JPM use case..
+  let mainURI = "../lib/main";
+  var { actionButton, toggleButton, icon } = require(mainURI);
+}
+var self = require("sdk/self");
+
+exports.testActionButton = function(assert) {
+  assert.equal(actionButton.id, "test-action-button", "action button id is correct");
+  assert.equal(actionButton.label, "Action Button", "action button label is correct");
+  assert.equal(actionButton.icon, icon, "action button icon is correct");
+}
+
+exports.testToggleButton = function(assert) {
+  assert.equal(toggleButton.id, "test-toggle-button", "toggle button id is correct");
+  assert.equal(toggleButton.label, "Toggle Button", "toggle button label is correct");
+  assert.equal(toggleButton.icon, icon, "toggle button icon is correct");
+}
+
+require("sdk/test").run(exports);