check filename for .png extension

[mkws-moved-to-github.git] / test / phantom / screenshot.js

diff --git a/test/phantom/screenshot.js b/test/phantom/screenshot.js
index 08e230d..8fea863 100644 (file)
--- a/test/phantom/screenshot.js
+++ b/test/phantom/screenshot.js
@@ -3,12 +3,25 @@ var page = require('webpage').create(),
 
 var url = system.args[1] || 'http://www.indexdata.com/';
 var file_png = system.args[2] || 'indexdata.png';
+var timeout = system.args[5] ? system.args[5] : 0.2;
 
-if (system.args.length === 1) {
+function usage (message) {
+    if (message) {
+        console.log(message + "\n");
+    }
     console.log('Usage: screenshot.js <some URL> <file.png>');
     phantom.exit();
 }
 
+if (system.args.length === 1) {
+    usage();
+}
+
+if (!file_png.match(/\.png$/)) {
+    usage("File name has no .png extension: '" + file_png + "'");
+}
+
+
 // page.zoomFactor = 1.0;
 page.viewportSize = {
     width: system.args[3] ? system.args[3] : 1200,
@@ -23,7 +36,9 @@ page.clipRect = {
 page.open(url, function () {
     // small delay
     setTimeout(function () {
-        page.render(file_png);
+        var ret = page.render(file_png);
         phantom.exit();
-    }, 200);
+    }, timeout * 1000);
 });
+
+// EOF