Have our mock Redis recover from client errors, just like Redis.

This allows us to test behaviors for when clients suddenly disconnect.

Author: djanowski

test/support/redis_mock.rb

@@ -24,20 +24,23 @@ def shutdown
     end
 
     def run
-      loop do
-        session = @server.accept
-
-        begin
-          return if yield(session) == :exit
-        ensure
-          session.close
+      begin
+        loop do
+          session = @server.accept
+
+          begin
+            return if yield(session) == :exit
+          ensure
+            session.close
+          end
         end
+      rescue => ex
+        $stderr.puts "Error running mock server: #{ex.message}" if VERBOSE
+        $stderr.puts ex.backtrace if VERBOSE
+        retry
+      ensure
+        @server.close
       end
-    rescue => ex
-      $stderr.puts "Error running mock server: #{ex.message}" if VERBOSE
-      $stderr.puts ex.backtrace if VERBOSE
-    ensure
-      @server.close
     end
   end