Add max scan options

lib/origin/optional.rb

@@ -2,6 +2,7 @@
 require "origin/optional/batch_size"
 require "origin/optional/hint"
 require "origin/optional/limit"
+require "origin/optional/max_scan"
 require "origin/optional/no_timeout"
 require "origin/optional/only"
 require "origin/optional/read"
@@ -11,14 +12,14 @@
 require "origin/optional/snapshot"
 require "origin/optional/without"
 
-# require "origin/optional/max_scan"
 # require "origin/optional/transformer"
 
 module Origin
   module Optional
     include BatchSize
     include Hint
     include Limit
+    include MaxScan
     include NoTimeout
     include Only
     include Read

lib/origin/optional/max_scan.rb

@@ -0,0 +1,11 @@
+# encoding: utf-8
+module Origin
+  module Optional
+    module MaxScan
+
+      def max_scan(value = nil)
+        option(value) { |options| options.store(:max_scan, value) }
+      end
+    end
+  end
+end

spec/origin/optional/max_scan_spec.rb

@@ -0,0 +1,64 @@
+require "spec_helper"
+
+describe Origin::Optional::MaxScan do
+
+  let(:query) do
+    Origin::Query.new
+  end
+
+  describe "#max_scan" do
+
+    context "when provided no options" do
+
+      let(:selection) do
+        query.max_scan
+      end
+
+      it "does not add any options" do
+        selection.options.should eq({})
+      end
+
+      it "returns the query" do
+        selection.should eq(query)
+      end
+
+      it "returns a cloned query" do
+        selection.should_not equal(query)
+      end
+    end
+
+    context "when provided nil" do
+
+      let(:selection) do
+        query.max_scan(nil)
+      end
+
+      it "does not add any options" do
+        selection.options.should eq({})
+      end
+
+      it "returns the query" do
+        selection.should eq(query)
+      end
+
+      it "returns a cloned query" do
+        selection.should_not equal(query)
+      end
+    end
+
+    context "when provided arguments" do
+
+      let(:selection) do
+        query.max_scan(500)
+      end
+
+      it "adds the field options" do
+        selection.options.should eq({ :max_scan => 500 })
+      end
+
+      it "returns a cloned query" do
+        selection.should_not equal(query)
+      end
+    end
+  end
+end