bytemark/flexnbd-c
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Rubocop

Browse Source
This commit is contained in:
Patrick J Cherry
2018-02-02 21:34:14 +00:00
parent 1b7b688f7a
commit 9c48da82cc
40 changed files with 666 additions and 858 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
tests/acceptance/test_write_during_migration.rb
View File

@@ -9,102 +9,98 @@ require 'tmpdir'
Thread.abort_on_exception = true
class TestWriteDuringMigration < Test::Unit::TestCase
def setup
@flexnbd = File.expand_path("../../build/flexnbd")
@flexnbd = File.expand_path('../../build/flexnbd')
raise "No binary!" unless File.executable?( @flexnbd )
raise 'No binary!' unless File.executable?(@flexnbd)
@size = 20*1024*1024 # 20MB
@write_data = "foo!" * 2048 # 8K write
@size = 20 * 1024 * 1024 # 20MB
@write_data = 'foo!' * 2048 # 8K write
@source_port = 9990
@dest_port = 9991
@source_sock = "src.sock"
@dest_sock = "dst.sock"
@source_file = "src.file"
@dest_file = "dst.file"
@source_sock = 'src.sock'
@dest_sock = 'dst.sock'
@source_file = 'src.file'
@dest_file = 'dst.file'
end
def teardown
[@dst_proc, @src_proc].each do |pid|
if pid
Process.kill( "KILL", pid ) rescue nil
next unless pid
begin
Process.kill('KILL', pid)
rescue StandardError
nil
end
end
end
def debug_arg
ENV['DEBUG'] ? "--verbose" : ""
ENV['DEBUG'] ? '--verbose' : ''
end
def launch_servers
@dst_proc = fork() {
@dst_proc = fork do
cmd = "#{@flexnbd} listen -l 127.0.0.1 -p #{@dest_port} -f #{@dest_file} -s #{@dest_sock} #{debug_arg}"
exec cmd
}
end
@src_proc = fork() {
@src_proc = fork do
cmd = "#{@flexnbd} serve -l 127.0.0.1 -p #{@source_port} -f #{@source_file} -s #{@source_sock} #{debug_arg}"
exec cmd
}
end
begin
awaiting = nil
Timeout.timeout(10) do
awaiting = :source
sleep 0.1 while !File.exists?( @source_sock )
sleep 0.1 until File.exist?(@source_sock)
awaiting = :dest
sleep 0.1 while !File.exists?( @dest_sock )
sleep 0.1 until File.exist?(@dest_sock)
end
rescue Timeout::Error
case awaiting
when :source
fail "Couldn't get a source socket."
raise "Couldn't get a source socket."
when :dest
fail "Couldn't get a destination socket."
raise "Couldn't get a destination socket."
else
fail "Something went wrong I don't understand."
raise "Something went wrong I don't understand."
end
end
end
def make_files()
def make_files
FileUtils.touch(@source_file)
File.truncate(@source_file, @size)
FileUtils.touch(@dest_file)
File.truncate(@dest_file, @size)
File.open(@source_file, "wb"){|f| f.write "a"*@size }
File.open(@source_file, 'wb') { |f| f.write 'a' * @size }
end
def start_mirror
UNIXSocket.open(@source_sock) {|sock|
sock.write(["mirror", "127.0.0.1", @dest_port.to_s, "exit"].join("\x0A") + "\x0A\x0A")
UNIXSocket.open(@source_sock) do |sock|
sock.write(['mirror', '127.0.0.1', @dest_port.to_s, 'exit'].join("\x0A") + "\x0A\x0A")
sock.flush
rsp = sock.readline
}
end
end
def wait_for_quit()
Timeout.timeout( 10 ) do
start_time = Time.now
dst_result = Process::waitpid2(@dst_proc)
src_result = Process::waitpid2(@src_proc)
def wait_for_quit
Timeout.timeout(10) do
start_time = Time.now
dst_result = Process.waitpid2(@dst_proc)
src_result = Process.waitpid2(@src_proc)
end
end
def source_writer
client = FlexNBD::FakeSource.new( "127.0.0.1", @source_port, "Timed out connecting" )
offsets = Range.new(0, (@size - @write_data.size) / 4096 ).to_a
client = FlexNBD::FakeSource.new('127.0.0.1', @source_port, 'Timed out connecting')
offsets = Range.new(0, (@size - @write_data.size) / 4096).to_a
loop do
begin
client.write(offsets[rand(offsets.size)] * 4096, @write_data)
rescue => err
rescue StandardError => err
# We expect a broken write at some point, so ignore it
break
end
@@ -115,32 +111,32 @@ class TestWriteDuringMigration < Test::Unit::TestCase
# puts `md5sum #{@source_file} #{@dest_file}`
# Ensure each block matches
File.open(@source_file, "r") do |source|
File.open(@dest_file, "r") do |dest|
0.upto( @size / 4096 ) do |block_num|
s_data = source.read( 4096 )
d_data = dest.read( 4096 )
File.open(@source_file, 'r') do |source|
File.open(@dest_file, 'r') do |dest|
0.upto(@size / 4096) do |block_num|
s_data = source.read(4096)
d_data = dest.read(4096)
assert s_data == d_data, "Block #{block_num} mismatch!"
source.seek( 4096, IO::SEEK_CUR )
dest.seek( 4096, IO::SEEK_CUR )
source.seek(4096, IO::SEEK_CUR)
dest.seek(4096, IO::SEEK_CUR)
end
end
end
end
def test_write_during_migration
Dir.mktmpdir() do |tmpdir|
Dir.chdir( tmpdir ) do
make_files()
Dir.mktmpdir do |tmpdir|
Dir.chdir(tmpdir) do
make_files
launch_servers()
launch_servers
src_writer = Thread.new { source_writer }
start_mirror()
wait_for_quit()
start_mirror
wait_for_quit
src_writer.join
assert_both_sides_identical
end
@@ -148,24 +144,21 @@ class TestWriteDuringMigration < Test::Unit::TestCase
end
def test_many_clients_during_migration
Dir.mktmpdir() do |tmpdir|
Dir.chdir( tmpdir ) do
make_files()
Dir.mktmpdir do |tmpdir|
Dir.chdir(tmpdir) do
make_files
launch_servers()
launch_servers
src_writers_1 = (1..5).collect { Thread.new { source_writer } }
start_mirror()
start_mirror
src_writers_2 = (1..5).collect { Thread.new { source_writer } }
wait_for_quit()
( src_writers_1 + src_writers_2 ).each {|t| t.join }
wait_for_quit
(src_writers_1 + src_writers_2).each(&:join)
assert_both_sides_identical
end
end end
end end
end