add tests

package.json

@@ -1,16 +1,25 @@
 {
-  "name" : "url-to-screenshot",
-  "description" : "Capture screenshots using phantomjs",
-  "version" : "0.0.0",
-  "repository" : {
-    "type" : "git",
-    "url" : "git://github.com/re-view/url-to-screenshot.git"
+  "name": "url-to-screenshot",
+  "description": "Capture screenshots using phantomjs",
+  "version": "0.0.0",
+  "repository": {
+    "type": "git",
+    "url": "git://github.com/re-view/url-to-screenshot.git"
+  },
+  "scripts": {
+    "test": "tape test.js"
   },
   "homepage": "https://github.com/re-view/url-to-screenshot",
-  "main" : "index.js",
-  "dependencies" : {},
-  "devDependencies" : {},
-  "keywords": ["phantomjs", "screenshot", "headless"],
+  "main": "index.js",
+  "dependencies": {},
+  "devDependencies": {
+    "tape": "~1.0.4"
+  },
+  "keywords": [
+    "phantomjs",
+    "screenshot",
+    "headless"
+  ],
   "author": {
     "name": "Julian Gruber",
     "email": "[email protected]",

test.js

@@ -0,0 +1,26 @@
+var test = require('tape');
+var Screenshot = require('./');
+
+test('screenshot', function(t) {
+  t.plan(3);
+  Screenshot('http://google.com')
+    .capture(function(err, pic) {
+      t.error(err);
+      t.ok(pic);
+      t.ok(Buffer.isBuffer(pic));
+    });
+});
+
+test('custom args', function(t) {
+  t.plan(3);
+  Screenshot('http://google.com')
+    .width(1024)
+    .height(768)
+    .timeout(100)
+    .format('jpeg')
+    .capture(function(err, pic) {
+      t.error(err);
+      t.ok(pic);
+      t.ok(Buffer.isBuffer(pic));
+    });
+});