test/setup.rb

   1 require 'test/unit'
   2
   3 require 'fileutils'
   4 require 'tmpdir'
   5 require 'stringio'
   6
   7 require 'rubygems'
   8 require 'test/zentest_assertions'
   9
  10 $TESTING = true
  11
  12 require 'mogilefs'
  13
  14 class FakeBackend
  15
  16   attr_reader :lasterr, :lasterrstr
  17
  18   def initialize
  19     @responses = Hash.new { |h,k| h[k] = [] }
  20     @lasterr = nil
  21     @lasterrstr = nil
  22   end
  23
  24   def method_missing(meth, *args)
  25     meth = meth.to_s
  26     if meth =~ /(.*)=$/ then
  27       @responses[$1] << args.first
  28     else
  29       response = @responses[meth].shift
  30       case response
  31       when Array then
  32         @lasterr = response.first
  33         @lasterrstr = response.last
  34         return nil
  35       end
  36       return response
  37     end
  38   end
  39
  40 end
  41
  42 class FakeSocket
  43
  44   attr_reader :read_s
  45   attr_reader :write_s
  46   attr_reader :sync
  47
  48   def initialize(read = '', write = StringIO.new)
  49     @read_s = read.class.method_defined?(:sysread) ? read : StringIO.new(read)
  50     @write_s = write
  51     @closed = false
  52     @sync = false
  53   end
  54
  55   def sync=(do_sync)
  56     @write_s.sync = do_sync
  57     @read_s.sync = do_sync
  58   end
  59
  60   def closed?
  61     @closed
  62   end
  63
  64   def close
  65     @closed = true
  66     return nil
  67   end
  68
  69   def gets
  70     @read_s.gets
  71   end
  72
  73   def peeraddr
  74     ['AF_INET', 6001, 'localhost', '127.0.0.1']
  75   end
  76
  77   def read(bytes)
  78     @read_s.read bytes
  79   end
  80
  81   def sysread(bytes, buf = '')
  82     @read_s.sysread bytes, buf
  83   end
  84
  85   def recv_nonblock(bytes, flags = 0)
  86     ret = @read_s.sysread(bytes)
  87     # Ruby doesn't expose pread(2)
  88     if (flags & Socket::MSG_PEEK) != 0
  89       @read_s.sysseek(-ret.size, IO::SEEK_CUR)
  90     end
  91     ret
  92   end
  93   alias_method :recv, :recv_nonblock
  94
  95   def write(data)
  96     @write_s.write data
  97   end
  98
  99   def syswrite(data)
 100     @write_s.syswrite data
 101   end
 102
 103 end
 104
 105 class MogileFS::Client
 106   attr_writer :readonly
 107 end
 108
 109 class TCPSocket
 110
 111   class << self
 112
 113     attr_accessor :connections
 114     attr_accessor :sockets
 115
 116     alias old_new new
 117
 118     def new(host, port)
 119       raise Errno::ECONNREFUSED if @sockets.empty?
 120       @connections << [host, port]
 121       @sockets.pop
 122     end
 123
 124     alias open new
 125
 126   end
 127
 128 end
 129
 130 class TestMogileFS < Test::Unit::TestCase
 131
 132   undef_method :default_test
 133
 134   def setup
 135     @tempdir = File.join Dir.tmpdir, "test_mogilefs_#{$$}"
 136     @root = File.join @tempdir, 'root'
 137     FileUtils.mkdir_p @root
 138
 139     @client = @klass.new :hosts => ['kaa:6001'], :domain => 'test',
 140                                   :root => @root
 141     @backend = FakeBackend.new
 142     @client.instance_variable_set '@backend', @backend
 143
 144     TCPSocket.sockets = []
 145     TCPSocket.connections = []
 146   end
 147
 148   def teardown
 149     FileUtils.rm_rf @tempdir
 150   end
 151
 152 end
 153