mongo 2.11.4 → 2.12.3

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/bulk_write_error_spec.rb

@@ -0,0 +1,49 @@
+require 'lite_spec_helper'
+
+describe Mongo::Error::BulkWriteError do
+  let(:result) do
+    {
+      'writeErrors' => [
+        { 'code' => 1, 'errmsg' => 'message1' },
+        { 'code' => 2, 'errmsg' => 'message2' },
+      ]
+    }
+  end
+
+  let(:error) { described_class.new(result) }
+
+  before do
+    error.add_note('note1')
+    error.add_note('note2')
+  end
+
+  describe '#result' do
+    it 'returns the result' do
+      expect(error.result).to eq(result)
+    end
+  end
+
+  describe '#labels' do
+    it 'returns an empty array' do
+      expect(error.labels).to eq([])
+    end
+  end
+
+  describe '#message' do
+    it 'is correct' do
+      expect(error.message).to eq("message1 (1), message2 (2) (note1, note2)")
+    end
+  end
+
+  describe '#to_s' do
+    it 'is correct' do
+      expect(error.to_s).to eq("message1 (1), message2 (2) (note1, note2)")
+    end
+  end
+
+  describe '#inspect' do
+    it 'is correct' do
+      expect(error.inspect).to eq("#<Mongo::Error::BulkWriteError: message1 (1), message2 (2) (note1, note2)>")
+    end
+  end
+end

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