mongo 2.11.2 → 2.13.0.beta1

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/builder/get_more_command_spec.rb

@@ -10,8 +10,13 @@ describe Mongo::Cursor::Builder::GetMoreCommand do
       end
     end
 
+    let(:description) do
+      Mongo::Server::Description.new(double('description address'),
+        'minWireVersion' => 0, 'maxWireVersion' => 2)
+    end
+
     let(:result) do
-      Mongo::Operation::Result.new(reply)
+      Mongo::Operation::Result.new(reply, description)
     end
 
     let(:cursor) do

data/spec/mongo/cursor/builder/op_get_more_spec.rb

@@ -10,8 +10,13 @@ describe Mongo::Cursor::Builder::OpGetMore do
       end
     end
 
+    let(:description) do
+      Mongo::Server::Description.new(double('description address'),
+        'minWireVersion' => 0, 'maxWireVersion' => 2)
+    end
+
     let(:result) do
-      Mongo::Operation::Result.new(reply)
+      Mongo::Operation::Result.new(reply, description)
     end
 
     let(:view) do

data/spec/mongo/cursor/builder/op_kill_cursors_spec.rb

@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe Mongo::Cursor::Builder::OpKillCursors do
+
+  let(:reply) do
+    Mongo::Protocol::Reply.allocate.tap do |reply|
+      allow(reply).to receive(:cursor_id).and_return(8000)
+    end
+  end
+
+  let(:description) do
+    Mongo::Server::Description.new(double('description address'),
+      'minWireVersion' => 0, 'maxWireVersion' => 2)
+  end
+
+  let(:result) do
+    Mongo::Operation::Result.new(reply, description)
+  end
+
+  let(:view) do
+    Mongo::Collection::View.new(
+      authorized_collection,
+      {},
+      tailable: true,
+      max_time_ms: 100
+    )
+  end
+
+  let(:cursor) do
+    Mongo::Cursor.new(view, result, authorized_primary)
+  end
+
+  let(:builder) do
+    described_class.new(cursor)
+  end
+
+  describe '#specification' do
+
+    let(:specification) do
+      builder.specification
+    end
+
+    it 'includes the cursor ids' do
+      expect(specification[:cursor_ids]).to eq([BSON::Int64.new(8000)])
+    end
+
+    it 'includes the database name' do
+      expect(specification[:db_name]).to eq(SpecConfig.instance.test_db)
+    end
+
+    it 'includes the collection name' do
+      expect(specification[:coll_name]).to eq(TEST_COLL)
+    end
+  end
+
+  describe '.get_cursors_list' do
+    it 'returns integer cursor ids' do
+      expect(described_class.get_cursors_list(builder.specification)).to eq([8000])
+    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
@@ -417,6 +434,14 @@ describe Mongo::Cursor do
   context 'when an implicit session is used' do
     min_server_fcv '3.6'
 
+    let(:subscriber) { EventSubscriber.new }
+
+    let(:subscribed_client) do
+      authorized_client.tap do |client|
+        client.subscribe(Mongo::Monitoring::COMMAND, subscriber)
+      end
+    end
+
     let(:collection) do
       subscribed_client[TEST_COLL]
     end
@@ -439,7 +464,7 @@ describe Mongo::Cursor do
     end
 
     let(:find_events) do
-      EventSubscriber.started_events.select { |e| e.command_name == "find" }
+      subscriber.started_events.select { |e| e.command_name == "find" }
     end
 
     context 'when all results are retrieved in the first response' do
@@ -499,11 +524,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 +612,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
@@ -315,8 +335,12 @@ describe Mongo::Database do
         client.start_session
       end
 
+      let(:subscriber) { EventSubscriber.new }
+
       let(:client) do
-        subscribed_client
+        authorized_client.tap do |client|
+          client.subscribe(Mongo::Monitoring::COMMAND, subscriber)
+        end
       end
 
       it_behaves_like 'an operation using a session'
@@ -324,7 +348,7 @@ describe Mongo::Database do
 
 
       let(:full_command) do
-        EventSubscriber.started_events.find { |cmd| cmd.command_name == 'ismaster' }.command
+        subscriber.started_events.find { |cmd| cmd.command_name == 'ismaster' }.command
       end
 
       it 'does not add a afterClusterTime field' do