1 # -*- encoding: binary -*-
2 STDIN.sync = STDOUT.sync = STDERR.sync = true
3 Thread.abort_on_exception = true
10 trap('CHLD') { Process.waitpid(-1, Process::WNOHANG) rescue nil }
18 attr_reader :lasterr, :lasterrstr
21 @responses = Hash.new { |h,k| h[k] = [] }
27 err_camel = err_snake.gsub(/(?:^|_)([a-z])/) { $1.upcase } << 'Error'
28 unless MogileFS::Backend.const_defined?(err_camel)
29 MogileFS::Backend.class_eval("class #{err_camel} < MogileFS::Error; end")
31 MogileFS::Backend.const_get(err_camel)
34 def method_missing(meth, *args)
36 if meth =~ /(.*)=$/ then
37 @responses[$1] << args.first
39 response = @responses[meth].shift
42 @lasterr = response.first
43 @lasterrstr = response.last
44 raise error(@lasterr), @lasterrstr
53 class MogileFS::Client
59 attr_reader :port, :pid
62 ObjectSpace.each_object(TempServer) { |t| t.destroy! }
65 at_exit { TempServer.destroy_all! }
67 def initialize(server_proc, port = nil)
72 @port ||= 1024 + rand(32768 - 1024)
73 @sock = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
74 @sock.bind(Socket.pack_sockaddr_in(@port.to_i, '127.0.0.1'))
76 rescue Errno::EADDRINUSE, Errno::EACCES => err
77 @sock.close rescue nil
79 retry if port.nil? && (retries -= 1) > 0
80 warn "retries failed: #{retries} port=#{port.inspect}"
83 @pid = fork { server_proc.call(@sock, @port) }
84 @sock.close rescue nil
88 @sock.close rescue nil
89 Process.kill('KILL', @pid) rescue nil
94 class TestMogileFS < Test::Unit::TestCase
96 undef_method :default_test if method_defined?(:default_test)
99 @client = @klass.new :hosts => ['kaa:6001'], :domain => 'test'
100 @backend = FakeBackend.new
101 @client.instance_variable_set '@backend', @backend
106 # for our mock results
108 alias_method :fetch_row, :shift
114 # devid, hostip, altip, http_port, http_get_port, dev status, host status
115 [ 1, '10.0.0.1', '192.168.0.1', 7500, 7600, 'readonly', 'alive' ],
116 [ 2, '10.0.0.2', '192.168.0.2', 7500, 7600, 'alive', 'alive' ],
117 [ 3, '10.0.0.3', nil, 7500, nil, 'readonly', 'alive' ],
118 [ 4, '10.0.0.4', nil, 7500, nil, 'alive', 'alive' ],
119 [ 5, '10.0.0.5', nil, 7500, nil, 'dead', 'alive' ],
120 [ 6, '10.0.0.6', nil, 7500, nil, 'alive', 'down' ],
133 str.to_s.gsub(/\\/, '\&\&').gsub(/'/, "''")
138 when MogileFS::Mysql::GET_DEVICES then TBL_DEVICES
139 when MogileFS::Mysql::GET_DOMAINS then TBL_DOMAINS