Fix block removal on cleanup

Author: bergie

index.coffee

@@ -111,10 +111,10 @@ module.exports = class Flatten
           blocks.push b
 
       blockIdx = item.content.indexOf block
-      return callback() if blockIdx is -1
+      #return callback() if blockIdx is -1
       unless blocks.length
         # Empty block, remove
-        item.content = item.content.splice blockIdx, 1
+        item.content.splice blockIdx, 1
         return callback()
       if blocks.length is 1
         # Block returned only one block, update values

package.json

@@ -1,6 +1,6 @@
 {
   "name": "html-flatten",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "flattens arbitrary html into a sane structure",
   "main": "index.coffee",
   "scripts": {

spec/Cleanup.coffee

@@ -29,7 +29,7 @@ describe 'Cleanup', ->
         done()
 
   describe 'cleaning a dirty HTML structure', ->
-    it 'produce clean blocks', (done) ->
+    it 'should produce clean blocks', (done) ->
       item =
         content: [
           id: 'foo'
@@ -58,3 +58,18 @@ describe 'Cleanup', ->
           html: '<p>Foobar</p>'
         chai.expect(item.content[2]).to.eql orig.content[1]
         done()
+
+  describe 'cleaning up a full item', ->
+    it 'should produce clean blocks', (done) ->
+      fs = require 'fs'
+      path = require 'path'
+      item = JSON.parse fs.readFileSync __dirname+'/fixtures/put.json', 'utf-8'
+      orig = JSON.parse JSON.stringify item
+      f.flattenItem item, ->
+        types = item.content.map (b) -> b.type
+        chai.expect(types).to.eql [
+          'image'
+          'image'
+          'image'
+        ]
+        done()