handle strings in batch

maxdemarzi · Mar 27, 2014 · 482a905 · kwent · Mar 27, 2014 · maxdemarzi
1 parent 6fd53ff
commit 482a905
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/lib/neography/rest/batch.rb b/lib/neography/rest/batch.rb
@@ -22,9 +22,9 @@ def do_batch(*args)
       end
 
       def get_batch(args)
-        if args[0].class == Symbol
-          send(("batch_" + args[0].to_s).to_sym, *args[1..-1])
-        else
+        begin
+          send("batch_#{args[0]}".to_sym, *args[1..-1])
+        rescue
           raise "Unknown option #{args[0]} - #{args}"
         end
       end

diff --git a/spec/unit/rest/batch_spec.rb b/spec/unit/rest/batch_spec.rb
@@ -16,6 +16,16 @@ class Rest
         subject.batch [:get_node, "foo"], [:get_node, "bar"]
       end
 
+      it "gets nodes without symbol" do
+        expected_body = [
+          { "id" => 0, "method" => "GET", "to" => "/node/foo" },
+          { "id" => 1, "method" => "GET", "to" => "/node/bar" }
+        ]
+
+        subject.connection.should_receive(:post).with("/batch", json_match(:body, expected_body))
+        subject.batch ["get_node", "foo"], [:get_node, "bar"]
+      end
+
       it "creates nodes" do
         expected_body = [
           { "id" => 0, "method" => "POST", "to" => "/node", "body" => { "foo" => "bar" } },