mongo 2.11.6 → 2.12.0.rc0

data/spec/mongo/crypt/helpers/mongo_crypt_spec_helper.rb

@@ -0,0 +1,108 @@
+module MongoCryptSpecHelper
+  def bind_crypto_hooks(mongocrypt)
+    Mongo::Crypt::Binding.mongocrypt_setopt_crypto_hooks(
+      mongocrypt,
+      method(:aes_encrypt),
+      method(:aes_decrypt),
+      method(:random),
+      method(:hmac_sha_512),
+      method(:hmac_sha_256),
+      method(:hmac_hash),
+      nil
+    )
+  end
+  module_function :bind_crypto_hooks
+
+  def mongocrypt_binary_t_from(string)
+    bytes = string.unpack('C*')
+
+    p = FFI::MemoryPointer
+      .new(bytes.size)
+      .write_array_of_type(FFI::TYPE_UINT8, :put_uint8, bytes)
+
+    Mongo::Crypt::Binding.mongocrypt_binary_new_from_data(p, bytes.length)
+  end
+  module_function :mongocrypt_binary_t_from
+
+  private
+
+  def string_from_binary(binary_p)
+    str_p = Mongo::Crypt::Binding.mongocrypt_binary_data(binary_p)
+    len = Mongo::Crypt::Binding.mongocrypt_binary_len(binary_p)
+    str_p.read_string(len)
+  end
+  module_function :string_from_binary
+
+  def write_to_binary(binary_p, data)
+    str_p = Mongo::Crypt::Binding.mongocrypt_binary_data(binary_p)
+    str_p.put_bytes(0, data)
+  end
+  module_function :write_to_binary
+
+  def aes_encrypt(_, key_binary_p, iv_binary_p, input_binary_p, output_binary_p,
+    response_length_p, status_p)
+    key = string_from_binary(key_binary_p)
+    iv = string_from_binary(iv_binary_p)
+    input = string_from_binary(input_binary_p)
+
+    output = Mongo::Crypt::Hooks.aes(key, iv, input)
+    write_to_binary(output_binary_p, output)
+    response_length_p.write_int(output.length)
+
+    true
+  end
+  module_function :aes_encrypt
+
+  def aes_decrypt(_, key_binary_p, iv_binary_p, input_binary_p, output_binary_p,
+    response_length_p, status_p)
+    key = string_from_binary(key_binary_p)
+    iv = string_from_binary(iv_binary_p)
+    input = string_from_binary(input_binary_p)
+
+    output = Mongo::Crypt::Hooks.aes(key, iv, input, decrypt: true)
+    write_to_binary(output_binary_p, output)
+    response_length_p.write_int(output.length)
+
+    true
+  end
+  module_function :aes_decrypt
+
+  def random(_, output_binary_p, num_bytes, status_p)
+    output = Mongo::Crypt::Hooks.random(num_bytes)
+    write_to_binary(output_binary_p, output)
+
+    true
+  end
+  module_function :random
+
+  def hmac_sha_512(_, key_binary_p, input_binary_p, output_binary_p, status_p)
+    key = string_from_binary(key_binary_p)
+    input = string_from_binary(input_binary_p)
+
+    output = Mongo::Crypt::Hooks.hmac_sha('SHA512', key, input)
+    write_to_binary(output_binary_p, output)
+
+    true
+  end
+  module_function :hmac_sha_512
+
+  def hmac_sha_256(_, key_binary_p, input_binary_p, output_binary_p, status_p)
+    key = string_from_binary(key_binary_p)
+    input = string_from_binary(input_binary_p)
+
+    output = Mongo::Crypt::Hooks.hmac_sha('SHA256', key, input)
+    write_to_binary(output_binary_p, output)
+
+    true
+  end
+  module_function :hmac_sha_256
+
+  def hmac_hash(_, input_binary_p, output_binary_p, status_p)
+    input = string_from_binary(input_binary_p)
+    output = Mongo::Crypt::Hooks.hash_sha256(input)
+    write_to_binary(output_binary_p, output)
+
+    true
+  end
+  module_function :hmac_hash
+end

data/spec/mongo/crypt/status_spec.rb

@@ -0,0 +1,152 @@
+require 'mongo'
+require 'support/lite_constraints'
+
+RSpec.configure do |config|
+  config.extend(LiteConstraints)
+end
+
+describe Mongo::Crypt::Status do
+  require_libmongocrypt
+
+  let(:status) { described_class.new }
+
+  let(:label) { :error_client }
+  let(:code) { 401 }
+  let(:message) { 'Unauthorized' }
+
+  let(:status_with_info) do
+    status.update(label, code, message)
+  end
+
+  describe '#initialize' do
+    it 'doesn\'t throw an error' do
+      expect { status }.not_to raise_error
+    end
+  end
+
+  describe '#self.from_pointer' do
+    let(:pointer) { Mongo::Crypt::Binding.mongocrypt_status_new }
+    let(:status) { described_class.from_pointer(pointer) }
+
+    after do
+      Mongo::Crypt::Binding.mongocrypt_status_destroy(pointer)
+    end
+
+    it 'creates a status from the pointer passed in' do
+      expect do
+        status
+      end.not_to raise_error
+
+      expect(status.ref).to eq(pointer)
+    end
+  end
+
+  describe '#update' do
+    context 'with invalid label' do
+      it 'raises an exception' do
+        expect do
+          status.update(:random_label, 0, '')
+        end.to raise_error(ArgumentError, /random_label is an invalid value for a Mongo::Crypt::Status label/)
+      end
+
+      it 'works with an empty message' do
+        status.update(:ok, 0, '')
+        expect(status.message).to eq('')
+      end
+    end
+  end
+
+  describe '#label' do
+    context 'new status' do
+      it 'returns :ok' do
+        expect(status.label).to eq(:ok)
+      end
+    end
+
+    context 'status with info' do
+      it 'returns label' do
+        expect(status_with_info.label).to eq(label)
+      end
+    end
+  end
+
+  describe '#code' do
+    context 'new status' do
+      it 'returns 0' do
+        expect(status.code).to eq(0)
+      end
+    end
+
+    context 'status with info' do
+      it 'returns code' do
+        expect(status_with_info.code).to eq(code)
+      end
+    end
+  end
+
+  describe '#message' do
+    context 'new status' do
+      it 'returns an empty string' do
+        expect(status.message).to eq('')
+      end
+    end
+
+    context 'status with info' do
+      it 'returns a message' do
+        expect(status_with_info.message).to eq(message)
+      end
+    end
+  end
+
+  describe '#ok?' do
+    context 'new status' do
+      it 'returns true' do
+        expect(status.ok?).to be true
+      end
+    end
+
+    context 'status with info' do
+      it 'returns false' do
+        expect(status_with_info.ok?).to be false
+      end
+    end
+  end
+
+  describe '#crypt_error' do
+    context 'when status is ok' do
+      before do
+        status.update(:ok, 0, '')
+      end
+
+      it 'does not raise exception' do
+        expect do
+          status.raise_crypt_error
+        end.not_to raise_error
+      end
+    end
+
+    context 'when status is :error_kms' do
+      before do
+        status.update(:error_kms, 100, 'KMS error message')
+      end
+
+      it 'raises exception' do
+        expect do
+          status.raise_crypt_error
+        end.to raise_error(Mongo::Error::KmsError, 'KMS error message (libmongocrypt error code 100)')
+      end
+    end
+
+    context 'when status is error client' do
+      before do
+        status.update(:error_client, 2, 'Client Error')
+      end
+
+      it 'raises exception' do
+        expect do
+          status.raise_crypt_error
+        end.to raise_error(Mongo::Error::CryptError, 'Client Error (libmongocrypt error code 2)')
+      end
+    end
+  end
+end

data/spec/mongo/cursor_spec.rb

@@ -75,6 +75,23 @@ describe Mongo::Cursor do
         end
       end
     end
+
+    context 'server is unknown' do
+      let(:server) do
+        view.send(:server_selector).select_server(authorized_client.cluster).tap do |server|
+          authorized_client.cluster.disconnect!
+          server.unknown!
+        end
+      end
+
+      let(:view) do
+        Mongo::Collection::View.new(authorized_collection)
+      end
+
+      it 'works' do
+        cursor
+      end
+    end
   end
 
   describe '#each' do
@@ -499,11 +516,14 @@ describe Mongo::Cursor do
     end
 
     let(:query_spec) do
-      { :selector => {}, :options => {}, :db_name => SpecConfig.instance.test_db, :coll_name => TEST_COLL }
+      { selector: {}, options: {},
+        db_name: SpecConfig.instance.test_db, coll_name: TEST_COLL }
     end
 
     let(:reply) do
-      Mongo::Operation::Find.new(query_spec)
+      Mongo::Operation::Find.new(query_spec).tap do |reply|
+        allow(reply).to receive(:cursor_id).and_return(42)
+      end
     end
 
     let(:cursor) do
@@ -584,10 +604,10 @@ describe Mongo::Cursor do
           expect(cursor.to_a.map { |doc| doc['a'] }).to eq((0..9).to_a + (20..99).to_a)
         end
 =end
-        it 'raises NotImplementedError' do
+        it 'raises InvalidCursorOperation' do
           expect do
             cursor.to_a
-          end.to raise_error(NotImplementedError, 'Cannot restart iteration of a cursor which issued a getMore')
+          end.to raise_error(Mongo::Error::InvalidCursorOperation, 'Cannot restart iteration of a cursor which issued a getMore')
         end
       end
     end

data/spec/mongo/database_spec.rb

@@ -180,6 +180,26 @@ describe Mongo::Database do
       it 'returns a list of the collections info' do
         expect(result).to include('acol')
       end
+
+      context 'with more than one collection' do
+        before do
+          database[:anothercol].drop
+          database[:anothercol].create
+
+          expect(database.collections.length).to be > 1
+        end
+
+        let(:result) do
+          database.list_collections(filter: { name: 'anothercol' }).map do |info|
+            info['name']
+          end
+        end
+
+        it 'can filter by collection name' do
+          expect(result.length).to eq(1)
+          expect(result.first).to eq('anothercol')
+        end
+      end
     end
 
     context 'server 2.6' do

data/spec/mongo/error/crypt_error_spec.rb

@@ -0,0 +1,26 @@
+require 'lite_spec_helper'
+
+describe Mongo::Error::CryptError do
+  let(:label) { :error_client }
+  let(:code) { 401 }
+  let(:message) { 'Operation unauthorized' }
+
+
+  describe '#initialize' do
+    context 'with code' do
+      let(:error) { described_class.new(message, code: code) }
+
+      it 'correctly generates the error message' do
+        expect(error.message).to eq("#{message} (libmongocrypt error code #{code})")
+      end
+    end
+
+    context 'with code' do
+      let(:error) { described_class.new(message) }
+
+      it 'correctly generates the error message' do
+        expect(error.message).to eq(message)
+      end
+    end
+  end
+end

data/spec/mongo/error/max_bson_size_spec.rb

@@ -0,0 +1,35 @@
+require 'lite_spec_helper'
+
+describe Mongo::Error::MaxBSONSize do
+  describe 'message' do
+    context 'when constructor is given no arguments' do
+      let(:error) do
+        described_class.new
+      end
+
+      it 'is the predefined message' do
+        error.message.should == 'The document exceeds maximum allowed BSON size'
+      end
+    end
+
+    context 'when constructor is given an integer argument' do
+      let(:error) do
+        described_class.new(42)
+      end
+
+      it 'is the predefined message with the size added' do
+        error.message.should == 'The document exceeds maximum allowed BSON size. The maximum allowed size is 42'
+      end
+    end
+
+    context 'when constructor is given a string argument' do
+      let(:error) do
+        described_class.new('hello world')
+      end
+
+      it 'is the provided message' do
+        error.message.should == 'hello world'
+      end
+    end
+  end
+end

data/spec/mongo/error/no_server_available_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require 'lite_spec_helper'
 
 describe Mongo::Error::NoServerAvailable do
   describe 'message' do
@@ -18,5 +18,15 @@ describe Mongo::Error::NoServerAvailable do
     it 'is correct' do
       expect(error.message).to eq('No primary server is available in cluster: #<Cluster topology=Unknown[127.0.0.1:27017] servers=[#<Server address=127.0.0.1:27017 UNKNOWN>]> with timeout=30, LT=0.015')
     end
+
+    context 'when cluster is nil' do
+      let(:error) do
+        Mongo::Error::NoServerAvailable.new(selector, nil)
+      end
+
+      it 'is correct' do
+        expect(error.message).to eq('No primary server is available with timeout=30, LT=0.015')
+      end
+    end
   end
 end

data/spec/mongo/error/operation_failure_spec.rb

@@ -72,7 +72,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is a network error' do
       context 'getMore' do
         let(:error) { Mongo::Error::OperationFailure.new('problem: socket exception',
-          Mongo::Operation::GetMore::Result.new([])) }
+          Mongo::Operation::GetMore::Result.new(nil)) }
 
         it 'returns true' do
           expect(error.change_stream_resumable?).to be true
@@ -91,7 +91,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is a resumable message' do
       context 'getMore response' do
         let(:error) { Mongo::Error::OperationFailure.new('problem: node is recovering',
-          Mongo::Operation::GetMore::Result.new([])) }
+          Mongo::Operation::GetMore::Result.new(nil)) }
 
         it 'returns true' do
           expect(error.change_stream_resumable?).to eql(true)
@@ -110,7 +110,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is a resumable code' do
       context 'getMore response' do
         let(:error) { Mongo::Error::OperationFailure.new('no message',
-          Mongo::Operation::GetMore::Result.new([]),
+          Mongo::Operation::GetMore::Result.new(nil),
           :code => 91, :code_name => 'ShutdownInProgress') }
 
         it 'returns true' do
@@ -131,7 +131,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is a non-resumable code' do
       context 'getMore response' do
         let(:error) { Mongo::Error::OperationFailure.new('no message',
-          Mongo::Operation::GetMore::Result.new([]),
+          Mongo::Operation::GetMore::Result.new(nil),
           :code => 136, :code_name => 'CappedPositionLost') }
 
         it 'returns false' do
@@ -152,7 +152,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is a non-resumable label' do
       context 'getMore response' do
         let(:error) { Mongo::Error::OperationFailure.new('no message',
-          Mongo::Operation::GetMore::Result.new([]),
+          Mongo::Operation::GetMore::Result.new(nil),
           :code => 91, :code_name => 'ShutdownInProgress',
           :labels => ['NonResumableChangeStreamError']) }
 
@@ -175,7 +175,7 @@ describe Mongo::Error::OperationFailure do
     context 'when there is another label' do
       context 'getMore response' do
         let(:error) { Mongo::Error::OperationFailure.new('no message',
-          Mongo::Operation::GetMore::Result.new([]),
+          Mongo::Operation::GetMore::Result.new(nil),
           :code => 91, :code_name => 'ShutdownInProgress',
           :labels => %w(TransientTransactionError)) }