Rubocop; add test to bombard a migration source with status commands
This commit is contained in:
46
tests/acceptance/test_write_during_migration.rb
Normal file → Executable file
46
tests/acceptance/test_write_during_migration.rb
Normal file → Executable file
@@ -82,15 +82,14 @@ class TestWriteDuringMigration < Test::Unit::TestCase
|
|||||||
UNIXSocket.open(@source_sock) do |sock|
|
UNIXSocket.open(@source_sock) do |sock|
|
||||||
sock.write(['mirror', '127.0.0.1', @dest_port.to_s, 'exit'].join("\x0A") + "\x0A\x0A")
|
sock.write(['mirror', '127.0.0.1', @dest_port.to_s, 'exit'].join("\x0A") + "\x0A\x0A")
|
||||||
sock.flush
|
sock.flush
|
||||||
rsp = sock.readline
|
sock.readline
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def wait_for_quit
|
def wait_for_quit
|
||||||
Timeout.timeout(10) do
|
Timeout.timeout(10) do
|
||||||
start_time = Time.now
|
Process.waitpid2(@dst_proc)
|
||||||
dst_result = Process.waitpid2(@dst_proc)
|
Process.waitpid2(@src_proc)
|
||||||
src_result = Process.waitpid2(@src_proc)
|
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
@@ -100,13 +99,28 @@ class TestWriteDuringMigration < Test::Unit::TestCase
|
|||||||
loop do
|
loop do
|
||||||
begin
|
begin
|
||||||
client.write(offsets[rand(offsets.size)] * 4096, @write_data)
|
client.write(offsets[rand(offsets.size)] * 4096, @write_data)
|
||||||
rescue StandardError => err
|
rescue StandardError
|
||||||
# We expect a broken write at some point, so ignore it
|
# We expect a broken write at some point, so ignore it
|
||||||
break
|
break
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def bombard_with_status
|
||||||
|
loop do
|
||||||
|
begin
|
||||||
|
UNIXSocket.open(@source_sock) do |sock|
|
||||||
|
sock.write("status\x0A\x0A")
|
||||||
|
sock.flush
|
||||||
|
sock.readline
|
||||||
|
end
|
||||||
|
rescue Errno::ENOENT
|
||||||
|
# If the socket disappears, that's OK.
|
||||||
|
break
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
def assert_both_sides_identical
|
def assert_both_sides_identical
|
||||||
# puts `md5sum #{@source_file} #{@dest_file}`
|
# puts `md5sum #{@source_file} #{@dest_file}`
|
||||||
|
|
||||||
@@ -160,5 +174,25 @@ class TestWriteDuringMigration < Test::Unit::TestCase
|
|||||||
(src_writers_1 + src_writers_2).each(&:join)
|
(src_writers_1 + src_writers_2).each(&:join)
|
||||||
assert_both_sides_identical
|
assert_both_sides_identical
|
||||||
end
|
end
|
||||||
end end
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
def test_status_call_after_cleanup
|
||||||
|
Dir.mktmpdir do |tmpdir|
|
||||||
|
Dir.chdir(tmpdir) do
|
||||||
|
make_files
|
||||||
|
|
||||||
|
launch_servers
|
||||||
|
|
||||||
|
status_poker = Thread.new { bombard_with_status }
|
||||||
|
|
||||||
|
start_mirror
|
||||||
|
|
||||||
|
wait_for_quit
|
||||||
|
status_poker.join
|
||||||
|
assert_both_sides_identical
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
Reference in New Issue
Block a user