ES2015ify the tests

.editorconfig

@@ -7,7 +7,7 @@ charset = utf-8
 trim_trailing_whitespace = true
 insert_final_newline = true
 
-[package.json]
+[{package.json,*.yml}]
 indent_style = space
 indent_size = 2
 

.travis.yml

@@ -1,5 +1,5 @@
 sudo: false
 language: node_js
 node_js:
-  - 'iojs'
+  - 'stable'
   - '0.12'

package.json

@@ -13,7 +13,7 @@
     "node": ">=0.12.0"
   },
   "scripts": {
-    "test": "xo && ava test.js test-buffer.js && echo unicorns | ava test-real.js"
+    "test": "xo && ava test.js && echo unicorns | ava test-real.js"
   },
   "files": [
     "index.js"
@@ -32,5 +32,10 @@
     "ava": "*",
     "buffer-equals": "^1.0.3",
     "xo": "*"
+  },
+  "xo": {
+    "ignores": [
+      "test.js"
+    ]
   }
 }

test-buffer.js

@@ -1,26 +0,0 @@
-'use strict';
-var test = require('ava');
-var bufferEquals = require('buffer-equals');
-var stdin = require('./');
-
-test('get stdin as a buffer', function (t) {
-	process.stdin.isTTY = false;
-
-	var promise = stdin.buffer(function (data) {
-		t.true(bufferEquals(data, new Buffer('unicorns')));
-		t.is(data.toString().trim(), 'unicorns');
-	});
-
-	process.stdin.push(new Buffer('unicorns'));
-	process.stdin.emit('end');
-
-	return promise;
-});
-
-test('get empty buffer when no stdin', function (t) {
-	process.stdin.isTTY = true;
-
-	return stdin.buffer(function (data) {
-		t.true(bufferEquals(data, new Buffer('')));
-	});
-});

test-real.js

@@ -1,6 +1,3 @@
-'use strict';
+import fn from './';
-var stdin = require('./');
 
-stdin().then(function (data) {
+fn().then(data => process.exit(data ? 0 : 1));
-	process.exit(data ? 0 : 1);
-});

test.js

@@ -1,24 +1,39 @@
-'use strict';
+import test from 'ava';
-var test = require('ava');
+import bufferEquals from 'buffer-equals';
-var stdin = require('./');
+import fn from './';
 
-test('get stdin', function (t) {
+test.serial('get stdin', async t => {
 	process.stdin.isTTY = false;
 
-	var promise = stdin(function (data) {
+	setImmediate(() => {
-		t.is(data.trim(), 'unicorns');
+		process.stdin.push('unicorns');
+		process.stdin.emit('end');
 	});
 
-	process.stdin.push('unicorns');
+	t.is((await fn()).trim(), 'unicorns');
+});
+
+test.serial('get empty string when no stdin', async t => {
+	process.stdin.isTTY = true;
+	t.is(await fn(), '');
+});
+
+test.serial('get stdin as a buffer', t => {
+	process.stdin.isTTY = false;
+
+	const promise = fn.buffer(data => {
+		t.true(bufferEquals(data, new Buffer('unicorns')));
+		t.is(data.toString().trim(), 'unicorns');
+	});
+
+	process.stdin.push(new Buffer('unicorns'));
 	process.stdin.emit('end');
 
 	return promise;
 });
 
-test('get empty string when no stdin', function (t) {
+test.serial('get empty buffer when no stdin', async t => {
 	process.stdin.isTTY = true;
 
-	return stdin(function (data) {
+	t.true(bufferEquals(await fn.buffer(), new Buffer('')));
-		t.is(data, '');
-	});
 });