Isolate runner's logic from entry-point

bin/atom-mocha

@@ -1,30 +1,5 @@
 #!/usr/local/bin/node --es_staging
 "use strict";
 
-/** Bring on a real test-runner */
-const Mocha = require("mocha");
-const Chai  = require("chai");
-Chai.should();
-
-
-/** Check if an element contains one or more CSS classes */
-Chai.Assertion.addMethod("class", function(classNames){
-	const subject   = Chai.util.flag(this, "object");
-	const classList = subject.classList;
-
-	if(!Array.isArray(classNames))
-		classNames = classNames.split(/\s+/g);
-
-	for(let name of classNames){
-		this.assert(
-			classList.contains(name),
-			"expected classList '" + subject.className + "' to include #{exp}",
-			"expected classList '" + subject.className + "' not to include #{exp}",
-			name
-		);
-	}
-});
-
-
 /** Connect Atom to our test runner */
 module.exports = require("../lib/main.js");

lib/main.js

@@ -5,6 +5,31 @@ const fs     = require("fs");
 const path   = require("path");
 
 
+/** Bring on a real test-runner */
+const Mocha = require("mocha");
+const Chai  = require("chai");
+Chai.should();
+
+
+/** Check if an element contains one or more CSS classes */
+Chai.Assertion.addMethod("class", function(classNames){
+	const subject   = Chai.util.flag(this, "object");
+	const classList = subject.classList;
+
+	if(!Array.isArray(classNames))
+		classNames = classNames.split(/\s+/g);
+
+	for(let name of classNames){
+		this.assert(
+			classList.contains(name),
+			"expected classList '" + subject.className + "' to include #{exp}",
+			"expected classList '" + subject.className + "' not to include #{exp}",
+			name
+		);
+	}
+});
+
+
 const patchReporter = require("./reporter-patch");
 
 module.exports = function(args){