require 'psych/helper' require 'yaml/store' require 'tmpdir' module Psych 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") File.chown(-1, Process.gid, @dir) @yamlstore_file = make_tmp_filename("yamlstore") @yamlstore = YAML::Store.new(@yamlstore_file) end def teardown YAML::ENGINE.yamler = @engine FileUtils.remove_entry_secure @dir end def make_tmp_filename(prefix) @dir + "/" + prefix + File.basename(__FILE__) + ".#{$$}.test" end def test_opening_new_file_in_readonly_mode_should_result_in_empty_values @yamlstore.transaction(true) do assert_nil @yamlstore[:foo] assert_nil @yamlstore[:bar] end end def test_opening_new_file_in_readwrite_mode_should_result_in_empty_values @yamlstore.transaction do assert_nil @yamlstore[:foo] assert_nil @yamlstore[:bar] end end def test_data_should_be_loaded_correctly_when_in_readonly_mode @yamlstore.transaction do @yamlstore[:foo] = "bar" end @yamlstore.transaction(true) do assert_equal "bar", @yamlstore[:foo] end end def test_data_should_be_loaded_correctly_when_in_readwrite_mode @yamlstore.transaction do @yamlstore[:foo] = "bar" end @yamlstore.transaction do assert_equal "bar", @yamlstore[:foo] end end def test_changes_after_commit_are_discarded @yamlstore.transaction do @yamlstore[:foo] = "bar" @yamlstore.commit @yamlstore[:foo] = "baz" end @yamlstore.transaction(true) do assert_equal "bar", @yamlstore[:foo] end end def test_changes_are_not_written_on_abort @yamlstore.transaction do @yamlstore[:foo] = "bar" @yamlstore.abort end @yamlstore.transaction(true) do assert_nil @yamlstore[:foo] end end def test_writing_inside_readonly_transaction_raises_error assert_raises(PStore::Error) do @yamlstore.transaction(true) do @yamlstore[:foo] = "bar" end end end end end if defined?(Psych)