mongo-ruby-driver/test/unit/connection_test.rb

require 'test/test_helper'
include Mongo

context "Initialization: " do
  setup do
    def new_mock_socket
      socket = Object.new
      socket.stubs(:setsockopt).with(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
      socket
    end

    def new_mock_db
      db = Object.new
    end
  end

  context "given a single node" do
    setup do
      TCPSocket.stubs(:new).returns(new_mock_socket)
      @conn = Connection.new('localhost', 27017, :connect => false)

      admin_db = new_mock_db
      admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
      @conn.expects(:[]).with('admin').returns(admin_db)
      @conn.connect_to_master
    end

    should "set localhost and port to master" do
      assert_equal 'localhost', @conn.host
      assert_equal 27017, @conn.port
    end

    should "set connection pool to 1" do
      assert_equal 1, @conn.size
    end

    should "default slave_ok to false" do
      assert !@conn.slave_ok?
    end
  end

  context "initializing a paired connection" do
    should "require left and right nodes" do
      assert_raise MongoArgumentError do
        Connection.paired(['localhost', 27018], :connect => false)
      end

      assert_raise MongoArgumentError do
        Connection.paired(['localhost', 27018], :connect => false)
      end
    end

    should "store both nodes" do
      @conn = Connection.paired([['localhost', 27017], ['localhost', 27018]], :connect => false)

      assert_equal ['localhost', 27017], @conn.nodes[0]
      assert_equal ['localhost', 27018], @conn.nodes[1]
    end
  end

  context "initializing with a mongodb uri" do
    should "parse a simple uri" do
      @conn = Connection.from_uri("mongodb://localhost", :connect => false)
      assert_equal ['localhost', 27017], @conn.nodes[0]
    end

    should "parse a uri specifying multiple nodes" do
      @conn = Connection.from_uri("mongodb://localhost:27017,mydb.com:27018", :connect => false)
      assert_equal ['localhost', 27017], @conn.nodes[0]
      assert_equal ['mydb.com', 27018], @conn.nodes[1]
    end

    should "parse a uri specifying multiple nodes with auth" do
      @conn = Connection.from_uri("mongodb://kyle:s3cr3t@localhost:27017/app,mickey:m0u5e@mydb.com:27018/dsny", :connect => false)
      assert_equal ['localhost', 27017], @conn.nodes[0]
      assert_equal ['mydb.com', 27018], @conn.nodes[1]
      assert_equal ['kyle', 's3cr3t', 'app'], @conn.auths[0]
      assert_equal ['mickey', 'm0u5e', 'dsny'], @conn.auths[1]
    end

    should "attempt to connect" do
      TCPSocket.stubs(:new).returns(new_mock_socket)
      @conn = Connection.from_uri("mongodb://localhost", :connect => false)

      admin_db = new_mock_db
      admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
      @conn.expects(:[]).with('admin').returns(admin_db)
      @conn.connect_to_master
    end

    should "raise an error on invalid uris" do
      assert_raise MongoArgumentError do
        Connection.from_uri("mongo://localhost", :connect => false)
      end

      assert_raise MongoArgumentError do
        Connection.from_uri("mongodb://localhost:abc", :connect => false)
      end

      assert_raise MongoArgumentError do
        Connection.from_uri("mongodb://localhost:27017, my.db.com:27018, ", :connect => false)
      end
    end

    should "require all of username, password, and database if any one is specified" do
      assert_raise MongoArgumentError do
        Connection.from_uri("mongodb://localhost/db", :connect => false)
      end

      assert_raise MongoArgumentError do
        Connection.from_uri("mongodb://kyle:password@localhost", :connect => false)
      end
    end
  end
end