1 files changed, 29 insertions, 3 deletions
diff --git a/docs/spec/domSpec.js b/docs/spec/domSpec.js
index 58063789..f9308ed6 100644
--- a/docs/spec/domSpec.js
+++ b/docs/spec/domSpec.js
@@ -1,9 +1,14 @@
 var DOM = require('dom.js').DOM;
 
 describe('dom', function(){
+  var dom;
+
+  beforeEach(function(){
+    dom = new DOM();
+  });
+
   describe('example', function(){
     it('should render code, live, test', function(){
-      var dom = new DOM();
       dom.example('desc', 'src', 'scenario');
       expect(dom.toString()).toEqual(
           '<h1>Example</h1>\n' +
@@ -15,7 +20,6 @@ describe('dom', function(){
     });
 
     it('should render non-live, test with description', function(){
-      var dom = new DOM();
       dom.example('desc', 'src', false);
       expect(dom.toString()).toEqual('<h1>Example</h1>\n' +
           '<div class="example">' +
@@ -26,10 +30,32 @@ describe('dom', function(){
     });
 
     it('should render non-live, test', function(){
-      var dom = new DOM();
       dom.example('desc', 'src', false);
       expect(dom.toString()).toContain('<pre class="brush: js; html-script: true;">src</pre>');
     });
+  });
+
+  describe('h', function(){
+
+    it('should render using function', function(){
+      var cbThis;
+      var cdValue;
+      dom.h('heading', 'content', function(value){
+        cbThis = this;
+        cbValue = value;
+      });
+      expect(cbThis).toEqual(dom);
+      expect(cbValue).toEqual('content');
+    });
+
+    it('should update heading numbers', function(){
+      dom.h('heading', function(){
+        this.html('<h1>sub-heading</h1>');
+      });
+      expect(dom.toString()).toContain('<h1>heading</h1>');
+      expect(dom.toString()).toContain('<h2>sub-heading</h2>');
+    });
 
   });
+
 });