More specs for Git::Patch

lib/pronto/git/patch.rb

@@ -21,9 +21,9 @@ def blame(lineno)
 
       def lines
         @lines ||= begin
-          hunks.map do |hunk|
+          hunks.flat_map do |hunk|
             hunk.lines.map { |line| Line.new(line, self, hunk) }
-          end.flatten.compact
+          end
         end
       end
 

spec/pronto/git/patch_spec.rb

@@ -20,6 +20,24 @@ module Git
         let(:rugged_patch) { double(stat: [5, 17]) }
         it { should == 17 }
       end
+
+      describe '#lines' do
+        subject { patch.lines }
+
+        let(:hunks) { [double(lines: [1, 2]), double(lines: [3])] }
+        let(:rugged_patch) { double(hunks: hunks) }
+        its(:count) { should == 3 }
+      end
+
+      describe '#new_file_full_path' do
+        subject { patch.new_file_full_path }
+
+        let(:rugged_patch) do
+          double(delta: double(new_file: { path: 'test.md' }))
+        end
+        let(:repo) { double(path: '/house/of/cards/orig.md') }
+        its(:to_s) { should == '/house/of/cards/test.md' }
+      end
     end
   end
 end