lib/fakeredis.rb in fakeredis-0.6.0 vs lib/fakeredis.rb in fakeredis-0.7.0

@@ -1,6 +1,18 @@
 require 'redis'
 require 'redis/connection/memory'
 
 module FakeRedis
   Redis = ::Redis
+
+  def self.enable
+    Redis::Connection.drivers << Redis::Connection::Memory unless enabled?
+  end
+
+  def self.enabled?
+    Redis::Connection.drivers.last == Redis::Connection::Memory
+  end
+
+  def self.disable
+    Redis::Connection.drivers.delete_if {|driver| Redis::Connection::Memory == driver }
+  end
 end