Merge pull request #86 from nwoythal/fix-first-token-whitespace

puppetlabs · Feb 24, 2023 · b56672d · b56672d
2 parents d1cdc01 + a9e7c21
commit b56672d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/puppet-lint/plugins/check_whitespace/trailing_whitespace.rb b/lib/puppet-lint/plugins/check_whitespace/trailing_whitespace.rb
@@ -30,7 +30,7 @@ def fix(problem)
 
     prev_token = problem[:token].prev_token
     next_token = problem[:token].next_token
-    prev_token.next_token = next_token
+    prev_token.next_token = next_token unless prev_token.nil?
     next_token.prev_token = prev_token unless next_token.nil?
     tokens.delete(problem[:token])
   end

diff --git a/spec/unit/puppet-lint/plugins/check_whitespace/trailing_whitespace_spec.rb b/spec/unit/puppet-lint/plugins/check_whitespace/trailing_whitespace_spec.rb
@@ -105,5 +105,17 @@
         expect(manifest).to eq(fixed)
       end
     end
+
+    context 'empty lines with nothing but whitespace' do
+      let(:code) { " \n " }
+
+      it 'detects problems with both empty lines' do
+        expect(problems).to have(2).problem
+      end
+
+      it 'fixes the manifest' do
+        expect(manifest).to eq("\n")
+      end
+    end
   end
 end