From 5f8de423f190bbb79a62f804151bc24824fa32d8 Mon Sep 17 00:00:00 2001
From: "Matt A. Tobin" <mattatobin@localhost.localdomain>
Date: Fri, 2 Feb 2018 04:16:08 -0500
Subject: Add m-esr52 at 52.6.0

---
 b2g/components/test/mochitest/screenshot_helper.js | 40 ++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 b2g/components/test/mochitest/screenshot_helper.js

(limited to 'b2g/components/test/mochitest/screenshot_helper.js')

diff --git a/b2g/components/test/mochitest/screenshot_helper.js b/b2g/components/test/mochitest/screenshot_helper.js
new file mode 100644
index 000000000..0320a14c1
--- /dev/null
+++ b/b2g/components/test/mochitest/screenshot_helper.js
@@ -0,0 +1,40 @@
+var Cu = Components.utils;
+var Ci = Components.interfaces;
+
+Cu.importGlobalProperties(['File']);
+
+const { Services } = Cu.import("resource://gre/modules/Services.jsm");
+
+// Load a duplicated copy of the jsm to prevent messing with the currently running one
+var scope = {};
+Services.scriptloader.loadSubScript("resource://gre/modules/Screenshot.jsm", scope);
+const { Screenshot } = scope;
+
+var index = -1;
+function next() {
+  index++;
+  if (index >= steps.length) {
+    assert.ok(false, "Shouldn't get here!");
+    return;
+  }
+  try {
+    steps[index]();
+  } catch(ex) {
+    assert.ok(false, "Caught exception: " + ex);
+  }
+}
+
+var steps = [
+  function getScreenshot() {
+    let screenshot = Screenshot.get();
+    assert.ok(screenshot instanceof File,
+              "Screenshot.get() returns a File");
+    next();
+  },
+
+  function endOfTest() {
+    sendAsyncMessage("finish");
+  }
+];
+
+next();
-- 
cgit v1.2.3