mongo 2.3.1 → 2.4.0.rc0

data/spec/mongo/server_selector/nearest_spec.rb

@@ -13,6 +13,53 @@ describe Mongo::ServerSelector::Nearest do
   it_behaves_like 'a server selector accepting tag sets'
   it_behaves_like 'a server selector with sensitive data in its options'
 
+  describe '#initialize' do
+
+    context 'when max_staleness is provided' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      it 'sets the max_staleness option' do
+        expect(selector.max_staleness).to eq(options[:max_staleness])
+      end
+    end
+  end
+
+  describe '#==' do
+
+    context 'when max staleness is the same' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      let(:other) do
+        described_class.new(options)
+      end
+
+      it 'returns true' do
+        expect(selector).to eq(other)
+      end
+    end
+
+    context 'when max staleness is different' do
+
+      let(:other_options) do
+        { max_staleness: 30 }
+      end
+
+      let(:other) do
+        described_class.new(other_options)
+      end
+
+      it 'returns false' do
+        expect(selector).not_to eq(other)
+      end
+    end
+  end
+
   describe '#to_mongos' do
 
     context 'tag set not provided' do
@@ -38,6 +85,32 @@ describe Mongo::ServerSelector::Nearest do
         )
       end
     end
+
+    context 'max staleness not provided' do
+
+      let(:expected) do
+        { :mode => 'nearest' }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
+
+    context 'max staleness provided' do
+
+      let(:max_staleness) do
+        60
+      end
+
+      let(:expected) do
+        { :mode => 'nearest', maxStalenessMS: 60000 }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
   end
 
   describe '#select' do

data/spec/mongo/server_selector/primary_preferred_spec.rb

@@ -13,6 +13,53 @@ describe Mongo::ServerSelector::PrimaryPreferred do
   it_behaves_like 'a server selector accepting tag sets'
   it_behaves_like 'a server selector with sensitive data in its options'
 
+  describe '#initialize' do
+
+    context 'when max_staleness is provided' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      it 'sets the max_staleness option' do
+        expect(selector.max_staleness).to eq(options[:max_staleness])
+      end
+    end
+  end
+
+  describe '#==' do
+
+    context 'when max staleness is the same' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      let(:other) do
+        described_class.new(options)
+      end
+
+      it 'returns true' do
+        expect(selector).to eq(other)
+      end
+    end
+
+    context 'when max staleness is different' do
+
+      let(:other_options) do
+        { max_staleness: 30 }
+      end
+
+      let(:other) do
+        described_class.new(other_options)
+      end
+
+      it 'returns false' do
+        expect(selector).not_to eq(other)
+      end
+    end
+  end
+
   describe '#to_mongos' do
 
     context 'tag sets not provided' do
@@ -31,6 +78,32 @@ describe Mongo::ServerSelector::PrimaryPreferred do
         )
       end
     end
+
+    context 'max staleness not provided' do
+
+      let(:expected) do
+        { :mode => 'primaryPreferred' }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
+
+    context 'max staleness provided' do
+
+      let(:max_staleness) do
+        60
+      end
+
+      let(:expected) do
+        { :mode => 'primaryPreferred', maxStalenessMS: 60000 }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
   end
 
   describe '#select' do

data/spec/mongo/server_selector/primary_spec.rb

@@ -11,6 +11,22 @@ describe Mongo::ServerSelector::Primary do
   end
   it_behaves_like 'a server selector with sensitive data in its options'
 
+  describe '#initialize' do
+
+    context 'when max_staleness is provided' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      it 'raises an exception' do
+        expect {
+          selector
+        }.to raise_exception(Mongo::Error::InvalidServerPreference)
+      end
+    end
+  end
+
   describe '#tag_sets' do
 
     context 'tags not provided' do
@@ -39,6 +55,26 @@ describe Mongo::ServerSelector::Primary do
     it 'returns nil' do
       expect(selector.to_mongos).to be_nil
     end
+
+    context 'max staleness not provided' do
+
+      it 'returns nil' do
+        expect(selector.to_mongos).to be_nil
+      end
+    end
+
+    context 'max staleness provided' do
+
+      let(:max_staleness) do
+        60
+      end
+
+      it 'raises an error' do
+        expect {
+          selector
+        }.to raise_exception(Mongo::Error::InvalidServerPreference)
+      end
+    end
   end
 
   describe '#select' do

data/spec/mongo/server_selector/secondary_preferred_spec.rb

@@ -13,6 +13,53 @@ describe Mongo::ServerSelector::SecondaryPreferred do
 
   it_behaves_like 'a server selector accepting tag sets'
 
+  describe '#initialize' do
+
+    context 'when max_staleness is provided' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      it 'sets the max_staleness option' do
+        expect(selector.max_staleness).to eq(options[:max_staleness])
+      end
+    end
+  end
+
+  describe '#==' do
+
+    context 'when max staleness is the same' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      let(:other) do
+        described_class.new(options)
+      end
+
+      it 'returns true' do
+        expect(selector).to eq(other)
+      end
+    end
+
+    context 'when max staleness is different' do
+
+      let(:other_options) do
+        { max_staleness: 30 }
+      end
+
+      let(:other) do
+        described_class.new(other_options)
+      end
+
+      it 'returns false' do
+        expect(selector).not_to eq(other)
+      end
+    end
+  end
+
   describe '#to_mongos' do
 
     context 'tag sets provided' do
@@ -34,6 +81,32 @@ describe Mongo::ServerSelector::SecondaryPreferred do
         expect(selector.to_mongos).to be_nil
       end
     end
+
+    context 'max staleness not provided' do
+
+      let(:expected) do
+        { :mode => 'secondaryPreferred' }
+      end
+
+      it 'returns nil' do
+        expect(selector.to_mongos).to be_nil
+      end
+    end
+
+    context 'max staleness provided' do
+
+      let(:max_staleness) do
+        60
+      end
+
+      let(:expected) do
+        { :mode => 'secondaryPreferred', maxStalenessMS: 60000 }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
   end
 
   describe '#select' do

data/spec/mongo/server_selector/secondary_spec.rb

@@ -13,6 +13,53 @@ describe Mongo::ServerSelector::Secondary do
 
   it_behaves_like 'a server selector accepting tag sets'
 
+  describe '#initialize' do
+
+    context 'when max_staleness is provided' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      it 'sets the max_staleness option' do
+        expect(selector.max_staleness).to eq(options[:max_staleness])
+      end
+    end
+  end
+
+  describe '#==' do
+
+    context 'when max staleness is the same' do
+
+      let(:options) do
+        { max_staleness: 60 }
+      end
+
+      let(:other) do
+        described_class.new(options)
+      end
+
+      it 'returns true' do
+        expect(selector).to eq(other)
+      end
+    end
+
+    context 'when max staleness is different' do
+
+      let(:other_options) do
+        { max_staleness: 30 }
+      end
+
+      let(:other) do
+        described_class.new(other_options)
+      end
+
+      it 'returns false' do
+        expect(selector).not_to eq(other)
+      end
+    end
+  end
+
   describe '#to_mongos' do
 
     it 'returns read preference formatted for mongos' do
@@ -30,6 +77,32 @@ describe Mongo::ServerSelector::Secondary do
         )
       end
     end
+
+    context 'max staleness not provided' do
+
+      let(:expected) do
+        { :mode => 'secondary' }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
+
+    context 'max staleness provided' do
+
+      let(:max_staleness) do
+        60
+      end
+
+      let(:expected) do
+        { :mode => 'secondary', maxStalenessMS: 60000 }
+      end
+
+      it 'returns a read preference formatted for mongos' do
+        expect(selector.to_mongos).to eq(expected)
+      end
+    end
   end
 
   describe '#select' do

data/spec/mongo/server_selector_spec.rb

@@ -291,4 +291,57 @@ describe Mongo::ServerSelector do
 
     it_behaves_like 'a ServerSelector'
   end
+
+  describe '#inspect' do
+
+    let(:options) do
+      {}
+    end
+
+    let(:read_pref) do
+      described_class.get({ mode: mode }.merge(options))
+    end
+
+    context 'when the mode is primary' do
+
+      let(:mode) do
+        :primary
+      end
+
+      it 'includes the mode in the inspect string' do
+        expect(read_pref.inspect).to match(/#{mode.to_s}/i)
+      end
+    end
+
+    context 'when there are tag sets' do
+
+      let(:mode) do
+        :secondary
+      end
+
+      let(:options) do
+        { tag_sets: [{ 'data_center' => 'nyc' }] }
+      end
+
+      it 'includes the tag sets in the inspect string' do
+        expect(read_pref.inspect).to include(options[:tag_sets].inspect)
+      end
+    end
+
+    context 'when there is a max staleness set' do
+
+      let(:mode) do
+        :secondary
+      end
+
+      let(:options) do
+        { max_staleness: 123 }
+      end
+
+      it 'includes the tag sets in the inspect string' do
+        expect(read_pref.inspect).to match(/max_staleness/i)
+        expect(read_pref.inspect).to match(/123/)
+      end
+    end
+  end
 end