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
|