mongo 1.12.5 → 2.0.0.beta

data/spec/support/sdam/single/direct_connection_external_ip.yml

@@ -0,0 +1,34 @@
+description: "Direct connection to RSPrimary via external IP"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true,
+                    hosts: ["b:27017"],  # Internal IP.
+                    setName: "rs"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "RSPrimary",
+                    setName: "rs"
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_mongos.yml

@@ -0,0 +1,33 @@
+description: "Connect to mongos"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true,
+                    msg: "isdbgrid"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "Mongos",
+                    setName:
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_rsarbiter.yml

@@ -0,0 +1,35 @@
+description: "Connect to RSArbiter"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                   ok: 1,
+                   ismaster: false,
+                   arbiterOnly: true,
+                   hosts: ["a"],
+                   setName: "rs"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "RSArbiter",
+                    setName: "rs"
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_rsprimary.yml

@@ -0,0 +1,34 @@
+description: "Connect to RSPrimary"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                   ok: 1,
+                   ismaster: true,
+                   hosts: ["a"],
+                   setName: "rs"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "RSPrimary",
+                    setName: "rs"
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_rssecondary.yml

@@ -0,0 +1,35 @@
+description: "Connect to RSSecondary"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: false,
+                    secondary: true,
+                    hosts: ["a"],
+                    setName: "rs"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "RSSecondary",
+                    setName: "rs"
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_slave.yml

@@ -0,0 +1,32 @@
+description: "Direct connection to slave"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: false
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "Standalone",
+                    setName:
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/direct_connection_standalone.yml

@@ -0,0 +1,32 @@
+description: "Connect to standalone"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "Standalone",
+                    setName:
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/not_ok_response.yml

@@ -0,0 +1,39 @@
+description: "Handle a not-ok ismaster response"
+
+uri: "mongodb://a"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true
+                }],
+
+                ["a:27017", {
+
+                    ok: 0,
+                    ismaster: true
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    address: "a",
+                    type: "Unknown",
+                    setName:
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/standalone_removed.yml

@@ -0,0 +1,32 @@
+description: "Standalone removed from multi-server topology"
+
+uri: "mongodb://a,b"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "b:27017": {
+
+                    type: "Unknown",
+                    setName:
+                }
+            },
+
+            topologyType: "Unknown",
+            setName:
+        }
+    }
+]

data/spec/support/sdam/single/unavailable_seed.yml

@@ -0,0 +1,28 @@
+description: "Unavailable seed"
+
+uri: "mongodb://a/?connect=direct"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {}]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "Unknown",
+                    setName:
+                }
+            },
+
+            topologyType: "Single",
+            setName:
+        }
+    }
+]

data/spec/support/server_discovery_and_monitoring.rb

@@ -0,0 +1,167 @@
+# Matcher for determining if the server is of the expected type according to
+# the test.
+#
+# @since 2.0.0
+RSpec::Matchers.define :be_server_type do |expected|
+
+  match do |actual|
+    case expected
+      when 'Standalone' then actual.standalone?
+      when 'RSPrimary' then actual.primary?
+      when 'RSSecondary' then actual.secondary?
+      when 'RSArbiter' then actual.arbiter?
+      when 'Mongos' then actual.mongos?
+      when 'Unknown' then actual.unknown?
+      when 'PossiblePrimary' then actual.unknown?
+      when 'RSGhost' then actual.ghost?
+      when 'RSOther' then actual.other?
+    end
+  end
+end
+
+# Matcher for determining if the cluster topology is the expected type.
+#
+# @since 2.0.0
+RSpec::Matchers.define :be_topology do |expected|
+
+  match do |actual|
+    case expected
+      when 'ReplicaSetWithPrimary' then actual.replica_set?
+      when 'ReplicaSetNoPrimary' then actual.replica_set?
+      when 'Sharded' then actual.sharded?
+      when 'Single' then actual.standalone?
+      when 'Unknown' then actual.unknown?
+    end
+  end
+end
+
+module Mongo
+  module SDAM
+
+    # Convenience helper to find a server by it's URI.
+    #
+    # @since 2.0.0
+    def find_server(client, uri)
+      client.cluster.instance_variable_get(:@servers).detect{ |s| s.address.to_s == uri }
+    end
+
+    # Represents a specification.
+    #
+    # @since 2.0.0
+    class Spec
+
+      # @return [ String ] description The spec description.
+      attr_reader :description
+
+      # @return [ Array<Phase> ] phases The spec phases.
+      attr_reader :phases
+
+      # @return [ Mongo::URI ] uri The URI object.
+      attr_reader :uri
+
+      # @return [ String ] uri_string The passed uri string.
+      attr_reader :uri_string
+
+      # 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['description']
+        @uri_string = @test['uri']
+        @uri = URI.new(uri_string)
+        @phases = @test['phases'].map{ |phase| Phase.new(phase, uri) }
+      end
+    end
+
+    # Represents a phase in the spec. Phases are sequential.
+    #
+    # @since 2.0.0
+    class Phase
+
+      # @return [ Outcome ] outcome The phase outcome.
+      attr_reader :outcome
+
+      # @return [ Array<Response> ] responses The responses for each server in
+      #   the phase.
+      attr_reader :responses
+
+      # Create the new phase.
+      #
+      # @example Create the new phase.
+      #   Phase.new(phase, uri)
+      #
+      # @param [ Hash ] phase The phase hash.
+      # @param [ Mongo::URI ] uri The URI.
+      #
+      # @since 2.0.0
+      def initialize(phase, uri)
+        @phase = phase
+        @responses = @phase['responses'].map{ |response| Response.new(response, uri) }
+        @outcome = Outcome.new(@phase['outcome'])
+      end
+    end
+
+    # Represents a server response during a phase.
+    #
+    # @since 2.0.0
+    class Response
+
+      # @return [ String ] address The server address.
+      attr_reader :address
+
+      # @return [ Hash ] ismaster The ismaster response.
+      attr_reader :ismaster
+
+      # Create the new response.
+      #
+      # @example Create the response.
+      #   Response.new(response, uri)
+      #
+      # @param [ Hash ] response The response value.
+      # @param [ Mongo::URI ] uri The URI.
+      #
+      # @since 2.0.0
+      def initialize(response, uri)
+        @uri = uri
+        @address = response[0]
+        @ismaster = response[1]
+      end
+    end
+
+    # Get the outcome or expectations from the phase.
+    #
+    # @since 2.0.0
+    class Outcome
+
+      # @return [ Array<Array<String, Hash>>] servers The expecations for
+      #   server states.
+      attr_reader :servers
+
+      # @return [ String ] set_name The expected RS set name.
+      attr_reader :set_name
+
+      # @return [ String ] topology_type The expected cluster topology type.
+      attr_reader :topology_type
+
+      # Create the new outcome.
+      #
+      # @example Create the new outcome.
+      #   Outcome.new(outcome)
+      #
+      # @param [ Hash ] outcome The outcome object.
+      #
+      # @since 2.0.0
+      def initialize(outcome)
+        @servers = outcome['servers']
+        @set_name = outcome['setName']
+        @topology_type = outcome['topologyType']
+      end
+    end
+  end
+end