mongo 1.12.5 → 2.0.0.beta

data/spec/support/helpers.rb

@@ -0,0 +1,140 @@
+# Copyright (C) 2009-2014 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Helper methods and utilities for testing.
+module Helpers
+
+  # Helper method to allow temporary redirection of $stdout.
+  #
+  # @example
+  # silence do
+  #   # your noisey code here
+  # end
+  #
+  # @param A code block to execute.
+  # @return Original $stdout value.
+  def silence(&block)
+    original_stdout = $stdout
+    original_stderr = $stderr
+    $stdout = $stderr = File.new('/dev/null', 'w')
+    yield block
+  ensure
+    $stdout = original_stdout
+    $stderr = original_stderr
+  end
+
+  TEST_KEY_RSA1024 = OpenSSL::PKey::RSA.new <<-_end_of_pem_
+-----BEGIN RSA PRIVATE KEY-----
+MIICXgIBAAKBgQDLwsSw1ECnPtT+PkOgHhcGA71nwC2/nL85VBGnRqDxOqjVh7Cx
+aKPERYHsk4BPCkE3brtThPWc9kjHEQQ7uf9Y1rbCz0layNqHyywQEVLFmp1cpIt/
+Q3geLv8ZD9pihowKJDyMDiN6ArYUmZczvW4976MU3+l54E6lF/JfFEU5hwIDAQAB
+AoGBAKSl/MQarye1yOysqX6P8fDFQt68VvtXkNmlSiKOGuzyho0M+UVSFcs6k1L0
+maDE25AMZUiGzuWHyaU55d7RXDgeskDMakD1v6ZejYtxJkSXbETOTLDwUWTn618T
+gnb17tU1jktUtU67xK/08i/XodlgnQhs6VoHTuCh3Hu77O6RAkEA7+gxqBuZR572
+74/akiW/SuXm0SXPEviyO1MuSRwtI87B02D0qgV8D1UHRm4AhMnJ8MCs1809kMQE
+JiQUCrp9mQJBANlt2ngBO14us6NnhuAseFDTBzCHXwUUu1YKHpMMmxpnGqaldGgX
+sOZB3lgJsT9VlGf3YGYdkLTNVbogQKlKpB8CQQDiSwkb4vyQfDe8/NpU5Not0fII
+8jsDUCb+opWUTMmfbxWRR3FBNu8wnym/m19N4fFj8LqYzHX4KY0oVPu6qvJxAkEA
+wa5snNekFcqONLIE4G5cosrIrb74sqL8GbGb+KuTAprzj5z1K8Bm0UW9lTjVDjDi
+qRYgZfZSL+x1P/54+xTFSwJAY1FxA/N3QPCXCjPh5YqFxAMQs2VVYTfg+t0MEcJD
+dPMQD5JX6g5HKnHFg2mZtoXQrWmJSn7p8GJK8yNTopEErA==
+-----END RSA PRIVATE KEY-----
+  _end_of_pem_
+
+  def issue_cert(dn, key, serial, not_before, not_after, extensions, issuer, issuer_key, digest)
+    cert = OpenSSL::X509::Certificate.new
+    issuer = cert unless issuer
+    issuer_key = key unless issuer_key
+    cert.version = 2
+    cert.serial = serial
+    cert.subject = dn
+    cert.issuer = issuer.subject
+    cert.public_key = key.public_key
+    cert.not_before = not_before
+    cert.not_after = not_after
+    ef = OpenSSL::X509::ExtensionFactory.new
+    ef.subject_certificate = cert
+    ef.issuer_certificate = issuer
+    extensions.each do |oid, value, critical|
+      cert.add_extension(ef.create_extension(oid, value, critical))
+    end
+    cert.sign(issuer_key, digest)
+    cert
+  end
+
+  def collection(name, db)
+    documents = []
+    double(name.to_s).tap do |coll|
+
+      allow(coll).to receive(:db) { db }
+
+      allow(coll).to receive(:save) do |doc|
+        coll.remove({ :_id => doc[:_id] })
+        coll.insert(doc)
+      end
+
+      allow(coll).to receive(:count) { documents.length }
+
+      allow(coll).to receive(:insert) do |doc|
+        if !coll.find({ :_id => doc[:_id] }).empty?
+          raise GridError, "duplicate key error, _id"
+        end
+        documents.push(doc)
+      end
+
+      allow(coll).to receive(:find_one) do |query|
+        result = nil
+        documents.each do |doc|
+          if matches(doc, query)
+            result = doc
+            break
+          end
+        end
+        result
+      end
+
+      allow(coll).to receive(:remove) do |query|
+        documents.dup.each do |doc|
+          if matches(doc, query)
+            documents.delete(doc)
+          end
+        end
+      end
+
+      allow(coll).to receive(:find) do |query|
+        results = []
+        documents.each do |doc|
+          if matches(doc, query)
+            results.push(doc)
+          end
+        end
+        results
+      end
+    end
+  end
+
+  # does only strict equivalence,
+  # {:n => 4} should work
+  # {:n => {"$gt" => 3}} will not work.
+  def matches(doc, query={})
+    match = true
+    query.each do |field, value|
+      if !doc[field] || doc[field] != value
+        match = false
+        break
+      end
+    end
+    match
+  end
+end

data/spec/support/matchers.rb

@@ -0,0 +1,37 @@
+RSpec::Matchers.define :be_int32 do |num|
+  match do |actual|
+    actual == [num].pack('l<')
+  end
+end
+
+RSpec::Matchers.define :be_int64 do |num|
+  match do |actual|
+    actual == [num].pack('q<')
+  end
+end
+
+RSpec::Matchers.define :be_int64_sequence do |array|
+  match do |actual|
+    actual == array.reduce(String.new) do |buffer, num|
+      buffer << [num].pack('q<')
+    end
+  end
+end
+
+RSpec::Matchers.define :be_cstring do |string|
+  match do |actual|
+    actual == "#{string}\0"
+  end
+end
+
+RSpec::Matchers.define :be_bson do |hash|
+  match do |actual|
+    actual == hash.to_bson
+  end
+end
+
+RSpec::Matchers.define :be_bson_sequence do |array|
+  match do |actual|
+    actual == array.map(&:to_bson).join
+  end
+end

data/spec/support/mongo_orchestration.rb

@@ -0,0 +1,61 @@
+# Copyright (C) 2009-2014 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'support/mongo_orchestration/requestable'
+require 'support/mongo_orchestration/standalone'
+
+module MongoOrchestration
+
+  extend self
+
+  TYPES = {
+      standalone: Standalone,
+      #replica_set: ReplicaSet,
+      #sharded_cluster: ShardedCluster
+    }
+
+  # The default base uri for mongo orchestration.
+  #
+  # @since 2.0.0
+  DEFAULT_BASE_URI = 'http://localhost:8889'.freeze
+
+
+  # Get a Mongo Orchestration resource.
+  #
+  # @example Get the Mongo Orchestration resource.
+  #   MongoOrchestration.get(:standalone)
+  #
+  # @param [ Symbol ] type The type of resource.
+  # @param [ Hash ] options Options for creating the resource.
+  #
+  # @option options [ String ] :path The path to use for the
+  #   Mongo Orchestration service.
+  #
+  # @return [ Standlone, ReplicaSet, ShardedCluster ] The resource.
+  #
+  # @since 2.0.0
+  def get(type, options = {})
+    TYPES[type].new(options)
+  end
+
+  # Raised when the Mongo Orchestration service is not available.
+  #
+  # @since 2.0.0
+  class ServiceNotAvailable < RuntimeError
+
+    def initialize
+      super("The Mongo Orchestration service is not available.")
+    end
+  end
+end

data/spec/support/mongo_orchestration/requestable.rb

@@ -0,0 +1,109 @@
+# Copyright (C) 2009-2014 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'httparty'
+
+module MongoOrchestration
+
+  # Encapsulates behavior of a Mongo Orchestration resource.
+  #
+  # @since 2.0.0
+  module Requestable
+    include HTTParty
+
+    # @return [ String ] base_path The path used to connect to the MO service.
+    attr_reader :base_path
+    alias_method :path, :base_path
+
+    # Initialize a Mongo Orchestration resource.
+    #
+    # @example Initialize a Mongo Orchestration resource.
+    #   MongoOrchestration::Standalone.new
+    #
+    # @param [ Hash ] options Options for creating the resource.
+    #
+    # @option options [ String ] :path The path to use for Mongo Orchestration.
+    #
+    # @since 2.0.0
+    def initialize(options = {})
+      @base_path = options[:path] || MongoOrchestration::DEFAULT_BASE_URI
+      create(options)
+    end
+
+    # Is the Mongo Orchestration resource still available?
+    #
+    # @example Check if the Mongo Orchestration resource is available.
+    #   standalone.alive?
+    #
+    # @return [ true, false ] If the resource is available.
+    #
+    # @since 2.0.0
+    def alive?
+      begin
+        get("servers/#{id}")
+      rescue ServiceNotAvailable
+        return false
+      end
+      !!(@config = @response if @response &&
+                                  @response['procInfo'] &&
+                                  @response['procInfo']['alive'])
+    end
+
+    # Send a request to Mongo Orchestration, specifying the request type.
+    #
+    # @example Get a list of servers.
+    #   standalone.request('GET', 'servers')
+    #
+    # @param [ String ] method The request type.
+    # @param [ String ] path The path for the request.
+    # @param [ Hash ] options Options for the request.
+    #
+    # @return [ HTTParty::Response ] The response from the request.
+    #
+    # @since 2.0.0
+    def request(method, path, options = {})
+      http_request(method, path, options)
+    end
+
+    private
+
+    def http_request(method, path = nil, options = {})
+      dispatch do
+        method = method.downcase.to_sym unless method.class == Symbol
+        options[:body] = options[:body].to_json if options.has_key?(:body)
+        HTTParty.send(method, [@base_path, path].compact.join('/'), options)
+      end
+    end
+
+    def get(path = nil, options = {})
+      http_request(__method__, path, options)
+    end
+
+    def post(path = nil, options = {})
+      http_request(__method__, path, options)
+    end
+
+    def dispatch
+      begin
+        @response = yield
+      rescue ArgumentError, Errno::ECONNREFUSED
+        raise ServiceNotAvailable.new unless ok?
+      end
+    end
+
+    def ok?
+      @response && @response.code/100 == 2
+    end
+  end
+end

data/spec/support/mongo_orchestration/standalone.rb

@@ -0,0 +1,57 @@
+# Copyright (C) 2009-2014 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module MongoOrchestration
+  class Standalone
+    include Requestable
+
+    attr_reader :client
+    attr_reader :id
+
+    # Stop this resource.
+    #
+    # @since 2.0.0
+    def stop
+      request_content = { body: { action: 'stop' } }
+      @config = post("#{orchestration}/#{id}", request_content)
+      self
+    end
+
+    private
+
+    def create(options = {})
+      unless alive?
+        @config = post(orchestration,
+                      { body: initialize_config[:request_content] })
+        @id = @config['id']
+        @client = Mongo::Client.new("#{@config['mongodb_uri']}")
+      end
+      self
+    end
+
+    def orchestration
+      initialize_config[:orchestration]
+    end
+
+    def initialize_config
+     {
+        orchestration: 'servers',
+        request_content: {
+                          name: 'mongod',
+                          procParams: { journal: true }
+                         }
+      }
+    end
+  end
+end

data/spec/support/sdam/rs/discover_arbiters.yml

@@ -0,0 +1,41 @@
+description: "Discover arbiters"
+
+uri: "mongodb://a/?replicaSet=rs"
+
+phases: [
+
+    {
+        responses: [
+
+                ["a:27017", {
+
+                    ok: 1,
+                    ismaster: true,
+                    hosts: ["a:27017"],
+                    arbiters: ["b:27017"],
+                    setName: "rs"
+                }]
+        ],
+
+        outcome: {
+
+            servers: {
+
+                "a:27017": {
+
+                    type: "RSPrimary",
+                    setName: "rs"
+                },
+
+                "b:27017": {
+
+                    type: "Unknown",
+                    setName:
+                }
+            },
+
+            topologyType: "ReplicaSetWithPrimary",
+            setName: "rs"
+        }
+    }
+]