spec/support/redis_stub.rb in redis_failover-0.3.0 vs spec/support/redis_stub.rb in redis_failover-0.4.0

@@ -4,10 +4,11 @@
   # Test stub for Redis.
   class RedisStub
     class Proxy
       def initialize(queue, opts = {})
         @info = {'role' => 'master'}
+        @config = {'slave-serve-stale-data' => 'yes'}
         @queue = queue
       end
 
       def blpop(*args)
         @queue.pop.tap do |value|
@@ -23,56 +24,69 @@
       end
 
       def slaveof(host, port)
         if host == 'no' && port == 'one'
           @info['role'] = 'master'
+          @info.delete('master_host')
+          @info.delete('master_port')
         else
           @info['role'] = 'slave'
+          @info['master_host'] = host
+          @info['master_port'] = port.to_s
         end
       end
 
       def info
         @info.dup
       end
 
-      def ping
-        'pong'
-      end
-
       def change_role_to(role)
         @info['role'] = role
       end
+
+      def config(action, attribute)
+        [action, @config[attribute]]
+      end
+
+      def force_sync_with_master(serve_stale_reads)
+        @config['slave-serve-stale-data'] = serve_stale_reads ? 'yes' : 'no'
+        @info['master_sync_in_progress'] = '1'
+      end
+
+      def force_sync_done
+        @info['master_sync_in_progress'] = '0'
+      end
     end
 
-    attr_reader :host, :port, :reachable
+    attr_reader :host, :port, :available
     def initialize(opts = {})
       @host = opts[:host]
       @port = Integer(opts[:port])
       @queue = Queue.new
       @proxy = Proxy.new(@queue, opts)
-      @reachable = true
+      @available = true
     end
 
     def method_missing(method, *args, &block)
-      if @reachable
+      if @available
         @proxy.send(method, *args, &block)
       else
         raise Errno::ECONNREFUSED
       end
     end
 
     def change_role_to(role)
       @proxy.change_role_to(role)
     end
 
-    def make_reachable!
-      @reachable = true
+    def make_available!
+      @available = true
     end
 
-    def make_unreachable!
+    def make_unavailable!
       @queue << Errno::ECONNREFUSED
-      @reachable = false
+      @available = false
     end
 
     def to_s
       "#{@host}:#{@port}"
     end
@@ -84,7 +98,8 @@
 
   module RedisStubSupport
     def redis
       @redis ||= RedisStub.new(:host => @host, :port => @port)
     end
+    alias_method :new_client, :redis
   end
 end
\ No newline at end of file