mongo 1.12.5 → 2.0.0

data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/SecondaryPreferred.yml

@@ -0,0 +1,26 @@
+---
+topology_description:
+  type: ReplicaSetNoPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/SecondaryPreferred_non_matching.yml

@@ -0,0 +1,21 @@
+---
+topology_description:
+  type: ReplicaSetNoPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: sf
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Secondary_non_matching.yml

@@ -0,0 +1,21 @@
+---
+topology_description:
+  type: ReplicaSetNoPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Secondary
+  tag_sets:
+  - data_center: sf
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest.yml

@@ -0,0 +1,33 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &3
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &2
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Nearest
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+- *2
+- *3
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest_non_matching.yml

@@ -0,0 +1,26 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Nearest
+  tag_sets:
+  - data_center: sf
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Primary.yml

@@ -0,0 +1,29 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &1
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Primary
+  tag_sets:
+  - {}
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred.yml

@@ -0,0 +1,29 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &1
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: PrimaryPreferred
+  tag_sets:
+  - {}
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred_non_matching.yml

@@ -0,0 +1,29 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &1
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: PrimaryPreferred
+  tag_sets:
+  - data_center: sf
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Secondary.yml

@@ -0,0 +1,31 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Secondary
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/SecondaryPreferred.yml

@@ -0,0 +1,31 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/SecondaryPreferred_non_matching.yml

@@ -0,0 +1,29 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - &1
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: sf
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Secondary_non_matching.yml

@@ -0,0 +1,26 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tags:
+      data_center: nyc
+  - address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tags:
+      data_center: nyc
+operation: read
+read_preference:
+  mode: Secondary
+  tag_sets:
+  - data_center: sf
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection/selection/Sharded/read/SecondaryPreferred.yml

@@ -0,0 +1,26 @@
+---
+topology_description:
+  type: Sharded
+  servers:
+  - &1
+    address: g:27017
+    avg_rtt_ms: 5
+    type: Mongos
+    tags:
+      data_center: nyc
+  - &2
+    address: h:27017
+    avg_rtt_ms: 35
+    type: Mongos
+    tags:
+      data_center: dc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/Single/read/SecondaryPreferred.yml

@@ -0,0 +1,19 @@
+---
+topology_description:
+  type: Single
+  servers:
+  - &1
+    address: a:27017
+    avg_rtt_ms: 5
+    type: Standalone
+    tags:
+      data_center: dc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/Unknown/read/SecondaryPreferred.yml

@@ -0,0 +1,11 @@
+---
+topology_description:
+  type: Unknown
+  servers: []
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection.rb

@@ -0,0 +1,157 @@
+module Mongo
+  module ServerSelection
+    module Read
+
+      # Represents a Server Selection specification test.
+      #
+      # @since 2.0.0
+      class Spec
+
+        # Mapping of topology description strings to topology type classes.
+        #
+        # @since 2.0.0
+        TOPOLOGY_TYPES = {
+          'ReplicaSetNoPrimary' => Mongo::Cluster::Topology::ReplicaSet,
+          'ReplicaSetWithPrimary' => Mongo::Cluster::Topology::ReplicaSet,
+          'Sharded' => Mongo::Cluster::Topology::Sharded,
+          'Single' => Mongo::Cluster::Topology::Standalone,
+          'Unknown' => Mongo::Cluster::Topology::Unknown
+        }
+
+        # Mapping of read preference modes.
+        #
+        # @since 2.0.0
+        READ_PREFERENCES = {
+          'Primary' => :primary,
+          'Secondary' => :secondary,
+          'PrimaryPreferred' => :primary_preferred,
+          'SecondaryPreferred' => :secondary_preferred,
+          'Nearest' => :nearest,
+        }
+
+        # @return [ String ] description The spec description.
+        #
+        # @since 2.0.0
+        attr_reader :description
+
+        # @return [ Hash ] read_preference The read preference to be used for selection.
+        #
+        # @since 2.0.0
+        attr_reader :read_preference
+
+        # @return [ Array<Hash> ] candidate_servers The candidate servers.
+        #
+        # @since 2.0.0
+        attr_reader :candidate_servers
+
+        # @return [ Array<Hash> ] eligible_servers The eligible servers before the latency
+        #   window is taken into account.
+        #
+        # @since 2.0.0
+        attr_reader :eligible_servers
+
+        # @return [ Array<Hash> ] suitable_servers The set of servers matching all server
+        #  selection logic. May be a subset of eligible_servers and/or candidate_servers.
+        #
+        # @since 2.0.0
+        attr_reader :suitable_servers
+
+        # @return [ Mongo::Cluster::Topology ] type The topology type.
+        #
+        # @since 2.0.0
+        attr_reader :type
+
+        # Instantiate the new spec.
+        #
+        # @example Create the spec.
+        #   Spec.new(file)
+        #
+        # @param [ String ] file The name of the file.
+        #
+        # @since 2.0.0
+        def initialize(file)
+          @test = YAML.load(ERB.new(File.new(file).read).result)
+          @description = "#{@test['topology_description']['type']}: #{File.basename(file)}"
+          @read_preference = @test['read_preference']
+          @read_preference['mode'] = READ_PREFERENCES[@read_preference['mode']]
+          @candidate_servers = @test['topology_description']['servers']
+          @suitable_servers = @test['suitable_servers']
+          @in_latency_window = @test['in_latency_window']
+          @type = TOPOLOGY_TYPES[@test['topology_description']['type']]
+        end
+
+        # Whether this spec describes a replica set.
+        #
+        # @example Determine if the spec describes a replica set.
+        #   spec.replica_set?
+        #
+        # @return [true, false] If the spec describes a replica set.
+        #
+        # @since 2.0.0
+        def replica_set?
+          type == Mongo::Cluster::Topology::ReplicaSet
+        end
+
+        # Does this spec raise an exception.
+        #
+        # @example Determine if the spec raises an exception.
+        #   spec.raises_exception?
+        #
+        # @return [true, false] If the spec raises an exception.
+        #
+        # @since 2.0.0
+        def raises_exception?
+          !server_available? || invalid_server_preference?
+        end
+
+        # Does this spec expect a server to be found.
+        #
+        # @example Will a server be found with this spec.
+        #   spec.server_available?
+        #
+        # @return [true, false] If a server will be found with this spec.
+        #
+        # @since 2.0.0
+        def server_available?
+          !in_latency_window.empty?
+        end
+
+        # Is the read preference defined in the spec invalid.
+        #
+        # @example Determine if the spec's read preference is invalid.
+        #   spec.invalid_server_preference?
+        #
+        # @return [true, false] If the spec's read preference is invalid.
+        #
+        # @since 2.0.0
+        def invalid_server_preference?
+          read_preference['mode'] == 'Primary' && read_preference['tag_sets']
+        end
+
+        # The subset of suitable servers that falls within the allowable latency
+        #   window.
+        # We have to correct for our server selection algorithm that adds the primary
+        #  to the end of the list for SecondaryPreferred read preference mode.
+        #
+        # @example Get the list of suitable servers within the latency window.
+        #   spec.in_latency_window
+        #
+        # @return [ Array<Hash> ] The servers within the latency window.
+        #
+        # @since 2.0.0
+        def in_latency_window
+          if read_preference['mode'] == :secondary_preferred && primary
+            return @in_latency_window.push(primary).uniq
+          end
+          @in_latency_window
+        end
+
+        private
+
+        def primary
+          @candidate_servers.find { |s| s['type'] == 'RSPrimary' }
+        end
+      end
+    end
+  end
+end

data/spec/support/server_selection_rtt.rb

@@ -0,0 +1,41 @@
+module Mongo
+  module ServerSelection
+    module RTT
+
+      # Represents a specification.
+      #
+      # @since 2.0.0
+      class Spec
+
+        # @return [ String ] description The spec description.
+        attr_reader :description
+
+        # @return [ Float ] avg_rtt_ms The starting average round trip time.
+        attr_reader :avg_rtt_ms
+
+        # @return [ Float ] new_rtt_ms The new round trip time for ismaster.
+        attr_reader :new_rtt_ms
+
+        # @return [ Float ] new_avg_rtt The newly calculated moving average round trip time.
+        attr_reader :new_avg_rtt
+
+        # Instantiate the new spec.
+        #
+        # @example Create the spec.
+        #   Spec.new(file)
+        #
+        # @param [ String ] file The name of the file.
+        #
+        # @since 2.0.0
+        def initialize(file)
+          @test = YAML.load(ERB.new(File.new(file).read).result)
+          @description = "avg_rtt_ms: #{@test['avg_rtt_ms']}, new_rtt_ms: #{@test['new_rtt_ms']}," +
+                           " new_avg_rtt: #{@test['new_avg_rtt']}"
+          @avg_rtt_ms = @test['avg_rtt_ms'] == 'NULL' ? nil : @test['avg_rtt_ms'].to_f
+          @new_rtt_ms = @test['new_rtt_ms'].to_f
+          @new_avg_rtt = @test['new_avg_rtt'].to_f
+        end
+      end
+    end
+  end
+end