Keep the original file mode when writing files.

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

diff --git a/test/test_main.rb b/test/test_main.rb
index 3003783554337a3bd3b2b05722ea3d29d414038a..a4a14504d463f1bf3f91786fd7b28445b1f4c97b 100644 (file)
--- a/test/test_main.rb
+++ b/test/test_main.rb
@@ -1,6 +1,7 @@
 require "test/unit"
 require "ogg/vorbis/tagger"
 require "fileutils"
+require "digest/md5"
 
 class MainTest < Test::Unit::TestCase
        OGG_FILE = "test/test.ogg"
@@ -29,6 +30,32 @@ class MainTest < Test::Unit::TestCase
                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)
@@ -50,6 +77,16 @@ class MainTest < Test::Unit::TestCase
                end
        end
 
+       def test_write_keeps_file_mode
+               mode = File.stat(OGG_FILE).mode
+
+               Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
+                       t.write
+               end
+
+               assert_equal(mode, File.stat(OGG_FILE).mode)
+       end
+
        def test_append
                Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
                        t.comments["genre"] = "Death Metal"
@@ -225,4 +262,37 @@ EOF
        def test_close
                Ogg::Vorbis::Tagger.new(OGG_FILE).close
        end
+
+       def test_close2
+               t = Ogg::Vorbis::Tagger.new(OGG_FILE)
+               t.close
+
+               assert_raise(Ogg::Vorbis::Tagger::ClosedStreamError) do
+                       t.close
+               end
+
+               assert_raise(Ogg::Vorbis::Tagger::ClosedStreamError) do
+                       Ogg::Vorbis::Tagger.open(OGG_FILE) do |t|
+                               t.close
+                       end
+               end
+       end
+
+       def test_open_non_existing_file
+               assert_raises(Ogg::Vorbis::Tagger::OpenError) do
+                       Ogg::Vorbis::Tagger.new("foo.bar")
+               end
+       end
+
+       def test_open_non_ogg_file
+               File.open("test/foo.bar", "w") do |f|
+                       f << "foobarbazxyzzy"
+               end
+
+               assert_raises(Ogg::Vorbis::Tagger::InvalidDataError) do
+                       Ogg::Vorbis::Tagger.new("test/foo.bar")
+               end
+       ensure
+               FileUtils.rm_f("test/foo.bar")
+       end
 end