upgrading to work with bit-docs-gen-html 0.1

Author: justinbmeyer

package.json

@@ -33,7 +33,7 @@
     }
   },
   "devDependencies": {
-    "bit-docs-generate-html": "^0.0.3",
+    "bit-docs-generate-html": "^0.1.0",
     "q": "^1.4.1",
     "connect": "^2.14.4",
     "zombie": "^4.2.1"

prettify.js

@@ -1,11 +1,13 @@
 require("./prettify-engine");
 require("./prettify.less")
 
-var codes = document.getElementsByTagName("code");
-for(var i = 0; i < codes.length; i ++){
-    var code = codes[i];
-    if(code.parentNode.nodeName.toUpperCase() === "PRE"){
-        code.className = code.className +" prettyprint";
+module.exports = function(){
+    var codes = document.getElementsByTagName("code");
+    for(var i = 0; i < codes.length; i ++){
+        var code = codes[i];
+        if(code.parentNode.nodeName.toUpperCase() === "PRE"){
+            code.className = code.className +" prettyprint";
+        }
     }
+    prettyPrint();
 }
-prettyPrint();

test.js

@@ -42,10 +42,10 @@ var open = function(url, callback, done){
 		.then(function(){
 			callback(browser, function(){
 				server.close();
-			})
+			});
 		}).catch(function(e){
 			server.close();
-			done(e)
+			done(e);
 		});
 };
 
@@ -75,9 +75,9 @@ describe("bit-docs-prettify", function(){
 
             open("temp/index.html",function(browser, close){
 
-				var prettyprinted = browser.window.document.getElementsByClassName("prettyprint")
+				var prettyprinted = browser.window.document.getElementsByClassName("prettyprint");
 
-				assert.ok(prettyprinted.length, "has a returns object")
+				assert.ok(prettyprinted.length, "has a returns object");
 
 				close();
 				done();
@@ -86,7 +86,7 @@ describe("bit-docs-prettify", function(){
 
         }).catch(function(err){
             console.log("err",err.stack);
-            done(err)
+            done(err);
         });
     });
 });