View[] will support specifying store [#22 state:resolved]

yrashk · Apr 30, 2008 · 318f468 · 318f468
1 parent 599c3ba
commit 318f468
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lib/strokedb/views/view.rb b/lib/strokedb/views/view.rb
@@ -235,9 +235,12 @@ def self.[](view_name)
   end
 
   class << View
-    def [](name, nsurl = name.modulize.empty? ? Module.nsurl : name.modulize.constantize.nsurl) # FIXME: it is not nice
+    def [](*args) # FIXME: it is not nice
+      store = args.first.is_a?(Store) ? args.shift : StrokeDB.default_store
+      name = args[0]
+      nsurl = args[1] || (name.modulize.empty? ? Module.nsurl : name.modulize.constantize.nsurl)
       uuid = ::Util.sha1_uuid("view:#{nsurl}##{name}") 
-      StrokeDB.default_store.find(uuid)
+      store.find(uuid)
     end
 
     alias :original_new :new

diff --git a/spec/lib/strokedb/views/view_spec.rb b/spec/lib/strokedb/views/view_spec.rb
@@ -161,6 +161,7 @@ def view.map(uuid, doc)
 
   it "should be findable with #[] syntax" do
     View["SomeView"].should == @view
+    View[StrokeDB.default_store, "SomeView"].should == @view
   end
 
 end
@@ -178,7 +179,9 @@ def view.map(uuid, doc)
   end
   it "should be findable with #[] syntax" do
     View["SomeView", "http://strokedb.com/"].should == @view
+    View[StrokeDB.default_store, "SomeView", "http://strokedb.com/"].should == @view
   end
 
 end
 
+