Fixed buffers decoration

atoum · Nov 19, 2015 · 4940d6f · 4940d6f
1 parent fe1ca14
commit 4940d6f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/lib/decorator.coffee b/lib/decorator.coffee
@@ -1,9 +1,14 @@
+{ CompositeDisposable } = require 'atom'
+
 module.exports =
 class AtoumDecorator
     constructor: ->
         @reset()
 
     reset: ->
+        @subscriptions?.dispose()
+
+        @subscriptions = new CompositeDisposable
         @markers = {}
 
     findLineStart: (line) ->
@@ -21,11 +26,16 @@ class AtoumDecorator
     decorate: (editor, file) ->
         return unless @markers[file]
 
-        gutter = editor.addGutter
-            name: 'atoum'
-            priority: 100
+        gutter = editor.gutterWithName 'atoum'
+
+        unless gutter
+            gutter = editor.addGutter
+                name: 'atoum'
+                priority: 100
 
         @markers[file].forEach (test, index) =>
+            return unless test.line
+
             line = editor.lineTextForBufferRow(test.line - 1)
             range = [[test.line - 1, @findLineStart(line)], [test.line - 1, @findLineEnd(line)]]
             marker = editor.markBufferRange range, invalidate: 'touch'

diff --git a/lib/parser.coffee b/lib/parser.coffee
@@ -60,7 +60,7 @@ class AtoumParser extends Emitter
                     if matches
                         if fs.existsSync(matches[1])
                             @test.file = matches[1]
-                            @test.line = matches[2]
+                            @test.line = parseInt(matches[2], 10) if matches[2]
                         else
                             @test.diag += line.replace(/^#\s+/g, '') + '\n'
                     else