Simple file-based store (for testing)

Rakefile

@@ -1,2 +1,7 @@
 require 'bundler'
 Bundler::GemHelper.install_tasks
+
+desc 'Load everything'
+task :console do
+  system "irb -I lib -r rack-oauth2_utils.rb"
+end

lib/rack-oauth2_utils.rb

@@ -1,6 +1,7 @@
 require 'rack'
 require 'rack-oauth2_utils/middleware'
 require 'rack-oauth2_utils/oauth_request'
+require 'rack-oauth2_utils/test_store'
 
 module Rack
   module OAuth2Utils

lib/rack-oauth2_utils/test_store.rb

@@ -0,0 +1,39 @@
+require 'pstore'
+module Rack
+  module OAuth2Utils
+
+    # Test persistent store. Stores to FS using PStore
+    # Not meant for production!
+    #
+    class TestStore
+      def initialize(file_path = '.')
+        @store = PStore.new(file_path)
+      end
+
+      def []=(key, value)
+        @store.transaction do
+          @store[key] = value
+        end
+      end
+
+      def [](key)
+        @store.transaction do
+          @store[key]
+        end
+      end
+
+      def delete(key)
+        @store.transaction do
+          @store.delete(key)
+        end
+      end
+
+      def roots
+        @store.transaction do
+          @store.roots
+        end
+      end
+    end
+
+  end
+end