test/test_patman.rb in patman-0.0.3 vs test/test_patman.rb in patman-0.0.4

@@ -1,10 +1,10 @@
 require 'test/unit'
 require 'patman'
 require 'fileutils'
 
-# require 'byebug'
+require 'byebug'
 
 
 class PatmanTest < Test::Unit::TestCase
 
     # Test all editing features.
@@ -20,13 +20,13 @@
         assert_equal( r.edited?, false )
 
         # ------------------------------------------------------------
         # Test diff jumping commands and their return values:
         r.step;        assert_equal( r.line, 2 )
-        r.step -1;     assert_equal( r.line, 1 )
-        true;          assert_equal( r.step.line, 2 )
-        true;          assert_equal( r.step(-1).line, 1 )
+        r.step( -1);   assert_equal( r.line, 1 )
+                       assert_equal( r.step.line, 2 )
+                       assert_equal( r.step(-1).line, 1 )
         r.line( 5 );   assert_equal( r.line, 5 )
         r.firstline;   assert_equal( r.line, 1 )
         r.lastline;    assert_equal( r.line, 10 )
 
 
@@ -120,24 +120,24 @@
 
         len = r.length
         assert_equal( r.firstline.insertfile( ifile ).length, 2*len )
         assert_equal( r.lastline.step.insertfile( ifile ).length, 3*len )
 
-        r.do_all do |r|
-            r.sub( /Line/, 'foobar' )
+        r.do_all do |p|
+            p.sub( /Line/, 'foobar' )
         end
         assert_equal( r.get_range(5,6), [ 'foobar 5', 'foobar 6' ] )
         r.blockline
         assert_equal( r.line, 30 )
 
-        r.do_range( 1, r.length-1 ) do |r|
-            r.sub( /foobar/, 'Line' )
+        r.do_range( 1, r.length-1 ) do |p|
+            p.sub( /foobar/, 'Line' )
         end
         assert_equal( r.get_range(5,6), [ 'Line 5', 'Line 6' ] )
 
-        r.do_for( 1, r.length-1 ) do |r|
-            r.sub( /Line/, 'foobar' )
+        r.do_for( 1, r.length-1 ) do |p|
+            p.sub( /Line/, 'foobar' )
         end
         assert_equal( r.get_range(5,6), [ 'foobar 5', 'foobar 6' ] )
 
 
         # ------------------------------------------------------------
@@ -165,9 +165,23 @@
         r3 = Patman.read( ifile )
         r2.lines r3.lines
         r2.write
 
         assert( system( "diff #{ifile} #{ofile} " ) )
+
+        FileUtils.rm_f ofile
+
+
+        # ------------------------------------------------------------
+        # Test Patman.edit:
+
+        ofile = 'test/output/test_file1.txt'
+        Patman.edit( ofile ) do
+            insert 'foobar'
+        end
+
+        p = Patman.edit( ofile )
+        assert_equal( p.get, 'foobar' )
 
         FileUtils.rm_f ofile
 
     end