merging from ruby

ruby · Nov 7, 2011 · 7f812e9 · 7f812e9
1 parent 24c31c2
commit 7f812e9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/test/psych/test_yamldbm.rb b/test/psych/test_yamldbm.rb
@@ -1,15 +1,17 @@
 # -*- coding: UTF-8 -*-
+
+require 'psych/helper'
+require 'tmpdir'
+
 begin
-  require 'test/unit'
   require 'yaml/dbm'
-  require 'tmpdir'
 rescue LoadError
 end
 
 module Psych
   ::Psych::DBM = ::YAML::DBM unless defined?(::Psych::DBM)
 
-  class YAMLDBMTest < Test::Unit::TestCase
+  class YAMLDBMTest < TestCase
     def setup
       @engine, YAML::ENGINE.yamler = YAML::ENGINE.yamler, 'psych'
       @dir = Dir.mktmpdir("rubytest-file")
@@ -191,4 +193,4 @@ def test_selsct_with_block
       assert_equal([], @yamldbm.select {false})
     end
   end
-end if defined?(YAML::DBM)
+end if defined?(YAML::DBM) && defined?(Psych)
diff --git a/test/psych/test_yamlstore.rb b/test/psych/test_yamlstore.rb
@@ -1,11 +1,11 @@
-require 'test/unit'
+require 'psych/helper'
 require 'yaml/store'
 require 'tmpdir'
 
-Psych::Store = YAML::Store unless defined?(Psych::Store)
-
 module Psych
-  class YAMLStoreTest < Test::Unit::TestCase
+  Psych::Store = YAML::Store unless defined?(Psych::Store)
+
+  class YAMLStoreTest < TestCase
     def setup
       @engine, YAML::ENGINE.yamler = YAML::ENGINE.yamler, 'psych'
       @dir = Dir.mktmpdir("rubytest-file")
@@ -77,11 +77,11 @@ def test_changes_are_not_written_on_abort
     end
 
     def test_writing_inside_readonly_transaction_raises_error
-      assert_raise(PStore::Error) do
+      assert_raises(PStore::Error) do
         @yamlstore.transaction(true) do
           @yamlstore[:foo] = "bar"
         end
       end
     end
   end
-end
+end if defined?(Psych)