summaryrefslogtreecommitdiffstats
path: root/spec/integration/util/file_locking_spec.rb
blob: 20c61d3d5cb07f7f2f41324ce08cef1c8299c409 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/usr/bin/env ruby

Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }

require 'puppet/util/file_locking'

describe Puppet::Util::FileLocking do
  it "should be able to keep file corruption from happening when there are multiple writers" do
    file = Tempfile.new("puppetspec")
    filepath = file.path
    file.close!()
    file = filepath
    data = {:a => :b, :c => "A string", :d => "another string", :e => %w{an array of strings}}
    File.open(file, "w") { |f| f.puts YAML.dump(data) }

    threads = []
    sync = Sync.new
    9.times { |a|
      threads << Thread.new {
        9.times { |b|
          sync.synchronize(Sync::SH) {
            Puppet::Util::FileLocking.readlock(file) { |f|
              YAML.load(f.read).should == data
            }
          }
          sleep 0.01
          sync.synchronize(Sync::EX) {
            Puppet::Util::FileLocking.writelock(file) { |f|
              f.puts YAML.dump(data)
            }
          }
        }
      }
    }
    threads.each { |th| th.join }
  end
end