Added two tests that ensure that writing comments doesn't corrupt files.

[ruby-vorbistagger.git] / test / test_main.rb

diff --git a/test/test_main.rb b/test/test_main.rb
index be81bb87048ca263bf456854eb0b627553db0f70..bbc5b0023a2987e31af883c4573dde96f4605dd5 100644 (file)
--- a/test/test_main.rb
+++ b/test/test_main.rb
@@ -1,30 +1,13 @@
 require "test/unit"
 require "ogg/vorbis/tagger"
 require "fileutils"
-require "open3"
+require "digest/md5"
 
 class MainTest < Test::Unit::TestCase
        OGG_FILE = "test/test.ogg"
 
        def setup
-               tmp =<<EOF
--c "artist=Bolt Thrower" -c "album=...For Victory" -c "date=1994"
-EOF
-
-               f = File.open(OGG_FILE, "w")
-               cmd = "oggenc -q 0 #{tmp.strip} -r -"
-
-               Open3.popen3(cmd) do |sin, sout, _|
-                       sin.write("\0\0\0\0")
-                       sin.close
-
-                       begin
-                               tmp = sout.read
-                               f << tmp
-                       end while !tmp.length.zero?
-               end
-
-               f.close
+               FileUtils.cp("test/sample.ogg", OGG_FILE)
        end
 
        def teardown
@@ -47,6 +30,32 @@ EOF
                end
        end
 
+       def test_write_is_non_destructive
+               a = Digest::MD5.hexdigest(File.read(OGG_FILE))
+
+               Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
+                       t.write
+               end
+
+               b = Digest::MD5.hexdigest(File.read(OGG_FILE))
+
+               assert_equal(a, b)
+       end
+
+       def test_multiple_writes
+               a = Digest::MD5.hexdigest(File.read(OGG_FILE))
+
+               Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
+                       1.upto(10) do
+                               t.write
+                       end
+               end
+
+               b = Digest::MD5.hexdigest(File.read(OGG_FILE))
+
+               assert_equal(a, b)
+       end
+
        def test_write_stable_order
                Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
                        assert_equal(3, t.write)
@@ -230,4 +239,17 @@ EOF
                                     t.comments.values)
                end
        end
+
+       def test_shift
+               Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
+                       assert_equal(["artist", "Bolt Thrower"], t.comments.shift)
+                       assert_equal(["album", "...For Victory"], t.comments.shift)
+                       assert_equal(["date", "1994"], t.comments.shift)
+                       assert_equal(nil, t.comments.shift)
+               end
+       end
+
+       def test_close
+               Ogg::Vorbis::Tagger.new(OGG_FILE).close
+       end
 end