mongo 2.11.2 → 2.13.0.beta1

data/spec/support/crypt/schema_maps/schema_map_aws.json

@@ -0,0 +1,17 @@
+{
+  "properties": {
+    "ssn": {
+      "encrypt": {
+        "keyId": [{
+          "$binary": {
+            "base64": "AWSAAAAAAAAAAAAAAAAAAA==",
+            "subType": "04"
+          }
+        }],
+        "bsonType": "string",
+        "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
+      }
+    }
+  },
+  "bsonType": "object"
+}

data/spec/support/crypt/schema_maps/schema_map_aws_key_alt_names.json

@@ -0,0 +1,12 @@
+{
+  "properties": {
+    "ssn": {
+      "encrypt": {
+        "keyId": "/altname",
+        "bsonType": "string",
+        "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Random"
+      }
+    }
+  },
+  "bsonType": "object"
+}

data/spec/support/crypt/schema_maps/schema_map_local.json

@@ -0,0 +1,18 @@
+{
+  "properties": {
+    "ssn": {
+      "encrypt": {
+        "keyId": [{
+          "$binary": {
+              "base64": "LOCALAAAAAAAAAAAAAAAAA==",
+              "subType": "04"
+
+              }
+        }],
+        "bsonType": "string",
+        "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
+      }
+    }
+  },
+  "bsonType": "object"
+}

data/spec/support/crypt/schema_maps/schema_map_local_key_alt_names.json

@@ -0,0 +1,12 @@
+{
+  "properties": {
+    "ssn": {
+      "encrypt": {
+        "keyId": "/altname",
+        "bsonType": "string",
+        "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Random"
+      }
+    }
+  },
+  "bsonType": "object"
+}

data/spec/support/event_subscriber.rb

@@ -3,110 +3,120 @@
 # @since 2.5.0
 class EventSubscriber
 
-  module Impl
-
-    # The started events.
-    #
-    # @since 2.5.0
-    attr_reader :started_events
-
-    # The succeeded events.
-    #
-    # @since 2.5.0
-    attr_reader :succeeded_events
-
-    # The failed events.
-    #
-    # @since 2.5.0
-    attr_reader :failed_events
-
-    attr_reader :published_events
-
-    # Cache the succeeded event.
-    #
-    # @param [ Event ] event The event.
-    #
-    # @since 2.5.0
-    def succeeded(event)
-      @mutex.synchronize do
-        succeeded_events.push(event)
-      end
+  # The started events.
+  #
+  # @since 2.5.0
+  attr_reader :started_events
+
+  # The succeeded events.
+  #
+  # @since 2.5.0
+  attr_reader :succeeded_events
+
+  # The failed events.
+  #
+  # @since 2.5.0
+  attr_reader :failed_events
+
+  attr_reader :published_events
+
+  # Cache the succeeded event.
+  #
+  # @param [ Event ] event The event.
+  #
+  # @since 2.5.0
+  def succeeded(event)
+    @mutex.synchronize do
+      succeeded_events.push(event)
     end
+  end
 
-    # Cache the started event.
-    #
-    # @param [ Event ] event The event.
-    #
-    # @since 2.5.0
-    def started(event)
-      @mutex.synchronize do
-        started_events.push(event)
-      end
+  # Cache the started event.
+  #
+  # @param [ Event ] event The event.
+  #
+  # @since 2.5.0
+  def started(event)
+    @mutex.synchronize do
+      started_events.push(event)
     end
+  end
 
-    # Filters command started events for the specified command name.
-    def command_started_events(command_name)
-      started_events.select do |event|
-        event.command[command_name]
-      end
+  # Filters command started events for the specified command name.
+  def command_started_events(command_name)
+    started_events.select do |event|
+      event.command[command_name]
     end
+  end
 
-    # Locates command stated events for the specified command name,
-    # asserts that there is exactly one such event, and returns it.
-    def single_command_started_event(command_name)
-      events = started_events.select do |event|
-        event.command[command_name]
-      end
-      if events.length != 1
-        raise "Expected a single #{command_name} event but we have #{events.length}"
+  def non_auth_command_started_events
+    started_events.reject do |event|
+      %w(authenticate getnonce saslSstart saslContinue).any? do |cmd|
+        event.command[cmd]
       end
-      events.first
     end
+  end
 
-    # Cache the failed event.
-    #
-    # @param [ Event ] event The event.
-    #
-    # @since 2.5.0
-    def failed(event)
-      @mutex.synchronize do
-        failed_events.push(event)
-      end
+  # Locates command stated events for the specified command name,
+  # asserts that there is exactly one such event, and returns it.
+  def single_command_started_event(command_name)
+    events = started_events.select do |event|
+      event.command[command_name]
+    end
+    if events.length != 1
+      raise "Expected a single #{command_name} event but we have #{events.length}"
     end
+    events.first
+  end
 
-    def select_published_events(cls)
-      @published_events.select do |event|
-        event.is_a?(cls)
-      end
+  def select_started_events(cls)
+    @started_events.select do |event|
+      event.is_a?(cls)
     end
+  end
 
-    def published(event)
-      @mutex.synchronize do
-        @published_events << event
-      end
+  def select_succeeded_events(cls)
+    @succeeded_events.select do |event|
+      event.is_a?(cls)
+    end
+  end
+
+  # Cache the failed event.
+  #
+  # @param [ Event ] event The event.
+  #
+  # @since 2.5.0
+  def failed(event)
+    @mutex.synchronize do
+      failed_events.push(event)
     end
+  end
 
-    # Clear all cached events.
-    #
-    # @since 2.5.1
-    def clear_events!
-      @started_events = []
-      @succeeded_events = []
-      @failed_events = []
-      @published_events = []
-      self
+  def select_published_events(cls)
+    @published_events.select do |event|
+      event.is_a?(cls)
     end
+  end
 
-    def initialize
-      @mutex = Mutex.new
-      clear_events!
+  def published(event)
+    @mutex.synchronize do
+      @published_events << event
     end
   end
 
-  include Impl
+  # Clear all cached events.
+  #
+  # @since 2.5.1
+  def clear_events!
+    @started_events = []
+    @succeeded_events = []
+    @failed_events = []
+    @published_events = []
+    self
+  end
 
-  class << self
-    include Impl
-    public :initialize
+  def initialize
+    @mutex = Mutex.new
+    clear_events!
   end
 end

data/spec/support/lite_constraints.rb

@@ -8,6 +8,14 @@ module LiteConstraints
     end
   end
 
+  def require_jruby
+    before(:all) do
+      unless BSON::Environment.jruby?
+        skip "JRuby required, we have #{SpecConfig.instance.platform}"
+      end
+    end
+  end
+
   # This is for marking tests that fail on jruby that should
   # in principle work (as opposed to being fundamentally incompatible
   # with jruby).
@@ -39,7 +47,9 @@ module LiteConstraints
 
   def require_mongo_kerberos
     before(:all) do
-      skip 'KERBEROS_REQUIRED env var not specified' unless ENV['KERBEROS_REQUIRED']
+      unless %w(1 yes true).include?(ENV['MONGO_RUBY_DRIVER_KERBEROS']&.downcase)
+        skip 'Set MONGO_RUBY_DRIVER_KERBEROS=1 in environment to run Kerberos unit tests'
+      end
       require 'mongo_kerberos'
     end
   end
@@ -52,6 +62,22 @@ module LiteConstraints
     end
   end
 
+  def require_libmongocrypt
+    before(:all) do
+      unless ENV['LIBMONGOCRYPT_PATH']
+        skip 'Test requires path to libmongocrypt to be specified in LIBMONGOCRYPT_PATH env variable'
+      end
+    end
+  end
+
+  def require_no_libmongocrypt
+    before(:all) do
+      if ENV['LIBMONGOCRYPT_PATH']
+        skip 'Test requires libmongocrypt to not be configured'
+      end
+    end
+  end
+
   # Some tests will fail if linting is enabled:
   # 1. Tests that pass invalid options to client, etc. which the linter
   #    rejects.
@@ -65,4 +91,51 @@ module LiteConstraints
       end
     end
   end
+
+  def require_aws_auth
+    before(:all) do
+      unless (ENV['AUTH'] || '') =~ /^aws/
+        skip 'This test requires AUTH=aws* and an appropriately configured runtime environment'
+      end
+    end
+  end
+
+  def require_ec2_host
+    before(:all) do
+      if $have_aws.nil?
+        $have_aws = begin
+          require 'open-uri'
+          begin
+            Timeout.timeout(3.81) do
+              URI.parse('http://169.254.169.254/latest/meta-data/profile').open.read
+            end
+            true
+          # When trying to use the EC2 metadata endpoint on ECS:
+          # Errno::EINVAL: Failed to open TCP connection to 169.254.169.254:80 (Invalid argument - connect(2) for "169.254.169.254" port 80)
+          rescue Timeout::Error, Errno::ETIMEDOUT, Errno::EINVAL, OpenURI::HTTPError => $aws_error
+            false
+          end
+        end
+      end
+      unless $have_aws
+        skip "EC2 instance metadata is not available - assuming not running on an EC2 instance: #{$aws_error.class}: #{$aws_error}"
+      end
+    end
+  end
+
+  def require_stress
+    before(:all) do
+      if !SpecConfig.instance.stress?
+        skip 'Set STRESS=1 in environment to run stress tests'
+      end
+    end
+  end
+
+  def require_fork
+    before(:all) do
+      if !SpecConfig.instance.fork?
+        skip 'Set FORK=1 in environment to run fork tests'
+      end
+    end
+  end
 end

data/spec/support/matchers.rb

@@ -35,3 +35,22 @@ RSpec::Matchers.define :be_bson_sequence do |array|
     actual == array.map(&:to_bson).join
   end
 end
+
+RSpec::Matchers.define :be_ciphertext do
+  match do |object|
+    object.is_a?(BSON::Binary) && object.type == :ciphertext
+  end
+end
+
+
+RSpec::Matchers.define :match_with_type do |event|
+  match do |actual|
+    Utils.match_with_type?(event, actual)
+  end
+end
+
+RSpec::Matchers.define :be_uuid do
+  match do |object|
+    object.is_a?(BSON::Binary) && object.type == :uuid
+  end
+end

data/spec/support/shared/protocol.rb

@@ -1,4 +1,6 @@
 shared_examples 'message with a header' do
+  let(:collection_name) { 'test' }
+
   describe 'header' do
     describe 'length' do
       let(:field) { bytes.to_s[0..3] }

data/spec/support/shared/scram_conversation.rb

@@ -0,0 +1,100 @@
+shared_context 'scram conversation context' do
+  let(:connection) do
+    double('connection').tap do |connection|
+      features = double('features')
+      allow(features).to receive(:op_msg_enabled?)
+      allow(connection).to receive(:features).and_return(features)
+      allow(connection).to receive(:server)
+    end
+  end
+end
+
+shared_examples 'scram conversation' do
+
+  describe '#parse_payload' do
+    let(:user) { double('user') }
+    let(:mechanism) { :scram }
+
+    shared_examples_for 'parses as expected' do
+      it 'parses as expected' do
+        conversation.send(:parse_payload, payload).should == expected
+      end
+    end
+
+    context 'regular payload' do
+      let(:payload) { 'foo=bar,hello=world' }
+      let(:expected) do
+        {'foo' => 'bar', 'hello' => 'world'}
+      end
+
+      it_behaves_like 'parses as expected'
+    end
+
+    context 'equal signs in value' do
+      let(:payload) { 'foo=bar==,hello=world=is=great' }
+      let(:expected) do
+        {'foo' => 'bar==', 'hello' => 'world=is=great'}
+      end
+
+      it_behaves_like 'parses as expected'
+    end
+
+    context 'missing value' do
+      let(:payload) { 'foo=,hello=' }
+      let(:expected) do
+        {'foo' => '', 'hello' => ''}
+      end
+
+      it_behaves_like 'parses as expected'
+    end
+
+    context 'missing key/value pair' do
+      let(:payload) { 'foo=,,hello=' }
+      let(:expected) do
+        {'foo' => '', 'hello' => ''}
+      end
+
+      it_behaves_like 'parses as expected'
+    end
+
+    context 'missing key' do
+      let(:payload) { '=bar' }
+
+      it 'raises an exception' do
+        lambda do
+          conversation.send(:parse_payload, payload)
+        end.should raise_error(Mongo::Error::InvalidServerAuthResponse, /Payload malformed: missing key/)
+      end
+    end
+
+    context 'all keys missing' do
+      let(:payload) { ',,,' }
+      let(:expected) do
+        {}
+      end
+
+      it_behaves_like 'parses as expected'
+    end
+  end
+end
+
+shared_context 'scram continue and finalize replies' do
+
+  let(:continue_document) do
+    BSON::Document.new(
+      'conversationId' => 1,
+      'done' => false,
+      'payload' => continue_payload,
+      'ok' => 1.0
+    )
+  end
+
+  let(:finalize_document) do
+    BSON::Document.new(
+      'conversationId' => 1,
+      'done' => false,
+      'payload' => finalize_payload,
+      'ok' => 1.0
+    )
+  end
+end