mongoid-shell 0.1.0 → 0.2.0

data/CHANGELOG.md

@@ -1,3 +1,19 @@
+Next Release
+============
+
+* Your contribution here.
+
+0.2.0 (1/29/2013)
+=================
+
+* Added `Mongoid::Shell::Commands::Mongo` that generates a command line which connects to the session's primary node - [@dblock](https://github.com/dblock).
+* Added `Mongoid::Shell::Commands::Mongostat` that generates a command line for `mongostat` - [@dblock](https://github.com/dblock).
+* Added `Mongoid::Shell::Commands::Mongorestore` that generates a command line for `mongorestore` - [@dblock](https://github.com/dblock).
+* A `Mongoid::Shell::Errors::SessionNotConnectedError` error will be raised when a session is not connected - [@dblock](https://github.com/dblock).
+* Added support for `--out`, `--dbpath`, `--directoryperdb`, `--journal`, `--oplog`, `--repair`, `--forceTableScan`, `--dbpath`, `--ipv6` to `Mongoid::Shell::Commands::Mongodump` - [@dblock](https://github.com/dblock).
+* Added support for `--nodb`, `--norc`, `--quiet` and `--ipv6` to `Mongoid::Shell::Commands::Mongo` - [@dblock](https://github.com/dblock).
+* It's now possible to override built-in `db`, `username`, `password`, `host` and `primary` - [@dblock](https://github.com/dblock).
+
 0.1.0 (1/27/2013)
 ==================
 

data/README.md

@@ -1,7 +1,20 @@
 Mongoid::Shell [![Build Status](https://travis-ci.org/dblock/mongoid-shell.png?branch=master)](https://travis-ci.org/dblock/mongoid-shell)
 ==============
 
-Create mongo command-lines from Mongoid configuration. 
+Create mongo command-lines from Mongoid configuration.
+
+For example, connect to your production database without having to remember the entire command line using a `db:production:shell` Rake task.
+
+``` ruby
+namespace :db
+  namespace :production
+    task :shell
+      Mongoid.load! File.join(Rails.root, "config/mongoid.yml"), :production
+      system Mongoid::Shell::Commands::Mongo.new.to_s
+    end
+  end
+end
+```
 
 Commands can be created for the current default session or you can pass a session as an argument to a new command.
 
@@ -10,17 +23,54 @@ Mongoid::Shell::Commands::Mongodump.new # will use Mongoid.default_session
 Mongoid::Shell::Commands::Mongodump.new(session: Moped::Session.new([ "127.0.0.1:27017" ]))
 ```
 
+Commands accept parameters.
+
+``` ruby
+out = File.join(Dir.tmpdir, 'db_backup')
+mongodump = Mongoid::Shell::Commands::Mongodump.new(db: 'another_database', out: out)
+system mongodump.to_s # mongodump --db another_database --out /tmp/db_backup
+```
+
 Supported Commands
 ==================
 
-### Mongodump
+## Mongo
+
+The mongo shell is an interactive JavaScript shell for MongoDB. The `Mongoid::Shell::Commands::Mongo` class generates a command line to connect to MongoDB. A particularly useful feature is that it will always yield the address of the master node of a MongoDB replica set.
+
+``` ruby
+Mongoid::Shell::Commands::Mongo.new.to_s
+```
+
+Supports `--username`, `--password`, `--eval`, `--nodb`, `--norc`, `--quiet` and `--ipv6`.
+
+## Mongodump
+
+Mongodump is a utility for creating a binary export of the contents of a database.
 
 ``` ruby
 mongodump = Mongoid::Shell::Commands::Mongodump.new({ collection: 'test' })
-mongodump.to_s # mongodump --host localhost:27017 --db test --collection test
+mongodump.to_s # mongodump --db test --collection test
+```
+
+The `Mongoid::Shell::Commands::Mongodump` class supports `--db`, `--host`, `--username`, `--password`, `--query`, `--out`, `--collection`, `--directoryperdb`, `--journal`, `--oplog`, `--repair`, `--forceTableScan`, `--dbpath` and `--ipv6`.
+
+## Mongorestore
+
+The mongorestore tool imports content from binary database dump, created by mongodump into a specific database.
+
+``` ruby
+mongorestore = Mongoid::Shell::Commands::Mongorestore.new({ collection: 'test', restore: '/tmp/db_backup' })
+mongorestore.to_s # mongorestore --db test --collection test /tmp/db_backup
 ```
 
-Supports `--db`, `--host`, `--username`, `--password` and `--collection`.
+The `Mongoid::Shell::Commands::Mongorestore` class supports `--db`, `--host`, `--username`, `--password`, `--collection`, `--ipv6, `--dbpath`, `--directoryperdb`, `--journal`, `--objcheck`, `--filter`, `--drop`, `--oplogReplay` and `--keepIndexVersion`.
+
+## Mongostat
+
+The mongostat utility provides a quick overview of the status of a currently running mongod or mongos instance.
+
+The `Mongoid::Shell::Commands::Mongostat` class supports `--host`, `--username`, `--password`, `--rowcount`, `--discover`, `--noheaders`, `--http` and `--all`.
 
 Contributing
 ------------

data/lib/config/locales/en.yml

@@ -6,5 +6,16 @@ en:
           missing_session:
             message: "Missing session."
             summary: "The command requires a session."
-            resolution: "Your code is missing a parameter to `new` or `command_for`.
-              Typically you can obtain a session from `Mongoid.default_session`."
+            resolution: "Your code is missing a parameter to `new` or `command_for`.\n
+            \_Typically you can obtain a session from `Mongoid.default_session`.\n
+            \_Examples:\n
+            \_\_Mongoid::Shell::Commands::Mongodump.new(session: Mongoid.default_session).to_s\n
+            \_\_Mongoid::Shell::Commands::Mongodump.command_for(Mongoid.default_session).to_s"
+          session_not_connected:
+            message: "Session is not connected."
+            summary: "The command requires a connected session."
+            resolution: "Ensure that the database is up and can be reached."
+          missing_primary_node:
+            message: "Session does not have a primary node."
+            summary: "The command requires a primary node."
+            resolution: "Ensure that the database is up and can be reached."

data/lib/mongoid/shell/commands.rb

@@ -1,2 +1,5 @@
 require 'mongoid/shell/commands/base'
+require 'mongoid/shell/commands/mongo'
 require 'mongoid/shell/commands/mongodump'
+require 'mongoid/shell/commands/mongorestore'
+require 'mongoid/shell/commands/mongostat'

data/lib/mongoid/shell/commands/base.rb

@@ -30,9 +30,14 @@ module Mongoid
           args.map do |key, property|
             value = self.send(property)
             next unless value
-            # TODO: quote other special characters?
-            value = '"' + value + '"' if value.include? ' '
-            "#{key} #{value}"
+            if value.is_a? Boolean
+              key
+            else
+              value = value.to_s
+              # TODO: quote other special characters?
+              value = '"' + value + '"' if value.include? ' '
+              key[0] == '-' ? "#{key} #{value}" : value
+            end
           end
         end
 
@@ -43,4 +48,4 @@ module Mongoid
       end
     end
   end
-end
+end

data/lib/mongoid/shell/commands/mongo.rb

@@ -0,0 +1,36 @@
+module Mongoid
+  module Shell
+    module Commands
+      class Mongo < Mongoid::Shell::Commands::Base
+        include Mongoid::Shell::Properties::Primary
+        include Mongoid::Shell::Properties::Database
+        include Mongoid::Shell::Properties::Username
+        include Mongoid::Shell::Properties::Password
+
+        attr_accessor :eval, :nodb, :norc, :quiet, :ipv6
+
+        def initialize(attrs = {})
+          super
+        end
+
+        def host_port_and_db
+          [ primary, db ].compact.join("/")
+        end
+
+        def vargs
+          super({
+            'db address' => :host_port_and_db,
+            '--username' => :username,
+            '--password' => :password,
+            '--eval' => :eval,
+            '--nodb' => :nodb,
+            '--norc' => :norc,
+            '--quiet' => :quiet,
+            '--ipv6' => :ipv6,
+          })
+        end
+
+      end
+    end
+  end
+end

data/lib/mongoid/shell/commands/mongodump.rb

@@ -7,7 +7,7 @@ module Mongoid
         include Mongoid::Shell::Properties::Username
         include Mongoid::Shell::Properties::Password
 
-        attr_accessor :collection, :query
+        attr_accessor :collection, :query, :out, :directoryperdb, :journal, :oplog, :repair, :forceTableScan, :dbpath, :ipv6
 
         def initialize(attrs = {})
           super
@@ -16,15 +16,23 @@ module Mongoid
         def vargs
           super({
             '--host' => :host,
-            '--db' => :database_name,
+            '--db' => :db,
             '--username' => :username,
             '--password' => :password,
             '--collection' => :collection,
-            '--query' => :query
+            '--query' => :query,
+            '--out' => :out,
+            '--directoryperdb' => :directoryperdb,
+            '--journal' => :journal,
+            '--oplog' => :oplog,
+            '--repair' => :repair,
+            '--forceTableScan' => :forceTableScan,
+            '--dbpath' => :dbpath,
+            '--ipv6' => :ipv6
           })
         end
 
       end
     end
   end
-end
+end

data/lib/mongoid/shell/commands/mongorestore.rb

@@ -0,0 +1,39 @@
+module Mongoid
+  module Shell
+    module Commands
+      class Mongorestore < Mongoid::Shell::Commands::Base
+        include Mongoid::Shell::Properties::Host
+        include Mongoid::Shell::Properties::Database
+        include Mongoid::Shell::Properties::Username
+        include Mongoid::Shell::Properties::Password
+
+        attr_accessor :collection, :ipv6, :dbpath, :directoryperdb, :journal, :objcheck, :filter, :drop, :oplogReplay, :keepIndexVersion, :restore
+
+        def initialize(attrs = {})
+          super
+        end
+
+        def vargs
+          super({
+            '--host' => :host,
+            '--db' => :db,
+            '--username' => :username,
+            '--password' => :password,
+            '--collection' => :collection,
+            '--ipv6' => :ipv6,
+            '--dbpath' => :dbpath,
+            '--directoryperdb' => :directoryperdb,
+            '--journal' => :journal,
+            '--objcheck' => :objcheck,
+            '--filter' => :filter,
+            '--drop' => :drop,
+            '--oplogReplay' => :oplogReplay,
+            '--keepIndexVersion' => :keepIndexVersion,
+            'directory or filename to restore from' => :restore
+          })
+        end
+
+      end
+    end
+  end
+end

data/lib/mongoid/shell/commands/mongostat.rb

@@ -0,0 +1,31 @@
+module Mongoid
+  module Shell
+    module Commands
+      class Mongostat < Mongoid::Shell::Commands::Base
+        include Mongoid::Shell::Properties::Host
+        include Mongoid::Shell::Properties::Username
+        include Mongoid::Shell::Properties::Password
+
+        attr_accessor :rowcount, :discover, :all, :http, :noheaders
+
+        def initialize(attrs = {})
+          super
+        end
+
+        def vargs
+          super({
+            '--host' => :host,
+            '--username' => :username,
+            '--password' => :password,
+            '--rowcount' => :rowcount,
+            '--discover' => :discover,
+            '--noheaders' => :noheaders,
+            '--http' => :http,
+            '--all' => :all
+          })
+        end
+
+      end
+    end
+  end
+end

data/lib/mongoid/shell/errors.rb

@@ -1,2 +1,4 @@
 require 'mongoid/shell/errors/base'
 require 'mongoid/shell/errors/missing_session_error'
+require 'mongoid/shell/errors/session_not_connected_error'
+require 'mongoid/shell/errors/missing_primary_node_error'

data/lib/mongoid/shell/errors/missing_primary_node_error.rb

@@ -0,0 +1,11 @@
+module Mongoid
+  module Shell
+    module Errors
+      class MissingPrimaryNodeError < Mongoid::Shell::Errors::Base
+        def initialize
+          super(compose_message("missing_primary_node"))
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/shell/errors/session_not_connected_error.rb

@@ -0,0 +1,11 @@
+module Mongoid
+  module Shell
+    module Errors
+      class SessionNotConnectedError < Mongoid::Shell::Errors::Base
+        def initialize
+          super(compose_message("session_not_connected"))
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/shell/properties.rb

@@ -1,4 +1,5 @@
 require 'mongoid/shell/properties/host'
+require 'mongoid/shell/properties/primary'
 require 'mongoid/shell/properties/database'
 require 'mongoid/shell/properties/username'
 require 'mongoid/shell/properties/password'

data/lib/mongoid/shell/properties/database.rb

@@ -3,12 +3,14 @@ module Mongoid
     module Properties
       module Database
 
+        attr_accessor :db
+
         # current database name
-        def database_name
-          session.send(:current_database).name
+        def db
+          @db || session.send(:current_database).name
         end
 
       end
     end
   end
-end
+end

data/lib/mongoid/shell/properties/host.rb

@@ -3,12 +3,18 @@ module Mongoid
     module Properties
       module Host
 
+        attr_accessor :host
+
         # database host
         def host
-          session.cluster.nodes.first.address
+          @host || begin
+            node = session.cluster.nodes.first
+            raise Mongoid::Shell::Errors::SessionNotConnectedError unless node
+            node.address == "localhost:27017" ? nil : node.address
+          end
         end
 
       end
     end
   end
-end
+end

data/lib/mongoid/shell/properties/password.rb

@@ -3,13 +3,17 @@ module Mongoid
     module Properties
       module Password
 
+        attr_accessor :password
+
         # current password
         def password
-          return nil unless session.context.cluster.auth && session.context.cluster.auth.first
-          session.context.cluster.auth.first[1][1]
+          @password || begin
+            return nil unless session.context.cluster.auth && session.context.cluster.auth.first
+            session.context.cluster.auth.first[1][1]
+          end
         end
 
       end
     end
   end
-end
+end

data/lib/mongoid/shell/properties/primary.rb

@@ -0,0 +1,21 @@
+module Mongoid
+  module Shell
+    module Properties
+      module Primary
+
+        attr_accessor :primary
+
+        # primary database host
+        def primary
+          @primary || begin
+            raise Mongoid::Shell::Errors::SessionNotConnectedError unless session.cluster.nodes.any?
+            node = session.cluster.nodes.find(&:primary?)
+            raise Mongoid::Shell::Errors::MissingPrimaryNodeError unless node
+            node.address == "localhost:27017" ? nil : node.address
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/mongoid/shell/properties/username.rb

@@ -3,13 +3,17 @@ module Mongoid
     module Properties
       module Username
 
+        attr_accessor :username
+
         # current username
         def username
-          return nil unless session.context.cluster.auth && session.context.cluster.auth.first
-          session.context.cluster.auth.first[1][0]
+          @username || begin
+            return nil unless session.context.cluster.auth && session.context.cluster.auth.first
+            session.context.cluster.auth.first[1][0]
+          end
         end
 
       end
     end
   end
-end
+end

data/lib/mongoid/shell/version.rb

@@ -1,5 +1,5 @@
 module Mongoid
   module Shell
-    VERSION = '0.1.0'
+    VERSION = '0.2.0'
   end
 end

data/spec/mongoid/commands/base_spec.rb

@@ -6,12 +6,12 @@ describe Mongoid::Shell::Commands::Base do
       Mongoid.stub(:default_session).and_return(nil)
     end
     it "raises an exception when a session is missing" do
-      expect { 
+      expect {
         Mongoid::Shell::Commands::Base.new({})
       }.to raise_error Mongoid::Shell::Errors::MissingSessionError, /Missing session./
     end
     it "raises an exception when options are missing" do
-      expect { 
+      expect {
         Mongoid::Shell::Commands::Base.new(nil)
       }.to raise_error Mongoid::Shell::Errors::MissingSessionError, /Missing session./
     end

data/spec/mongoid/commands/mongo_spec.rb

@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe Mongoid::Shell::Commands::Mongo do
+  include MopedSessionHelper
+  context "local" do
+    it "defaults to local" do
+      Mongoid::Shell::Commands::Mongo.new.to_s.should == "mongo mongoid_shell_tests"
+    end
+    it "includes eval" do
+      Mongoid::Shell::Commands::Mongo.new({
+        eval: 'find x'
+      }).to_s.should == 'mongo mongoid_shell_tests --eval "find x"'
+    end
+    it "overrides primary" do
+      Mongoid::Shell::Commands::Mongo.new({
+        eval: 'find x',
+        primary: 'my_primary'
+      }).to_s.should == 'mongo my_primary/mongoid_shell_tests --eval "find x"'
+    end
+    [ :nodb, :norc, :quiet, :ipv6 ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongo.new({
+          option => true
+        }).to_s.should == "mongo mongoid_shell_tests --#{option}"
+      end
+    end
+
+  end
+  context "sessions" do
+    context "default" do
+      before :each do
+        @session = Mongoid::Sessions.with_name(:default)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongo.new({
+          session: @session
+        }).to_s.should == "mongo mongoid_shell_tests"
+      end
+    end
+    context "a replica set" do
+      before :each do
+        @session = moped_session(:replica_set)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongo.new({
+          session: @session
+        }).to_s.should == "mongo dedicated1.myapp.com:27017/mongoid --username user --password password"
+      end
+    end
+    context "url" do
+      before :each do
+        @session = moped_session(:url)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongo.new({
+          session: @session
+        }).to_s.should == "mongo 59.1.22.1:27017/mongoid --username user --password password"
+      end
+    end
+  end
+end

data/spec/mongoid/commands/mongodump_spec.rb

@@ -1,19 +1,66 @@
 require 'spec_helper'
 
 describe Mongoid::Shell::Commands::Mongodump do
-  context "default session" do
+  include MopedSessionHelper
+  context "local" do
     it "defaults to local" do
-      Mongoid::Shell::Commands::Mongodump.new.to_s.should == "mongodump --host localhost:27017 --db mongoid_shell_tests"
+      Mongoid::Shell::Commands::Mongodump.new.to_s.should == "mongodump --db mongoid_shell_tests"
     end
     it "includes collection" do
-      Mongoid::Shell::Commands::Mongodump.new({ 
+      Mongoid::Shell::Commands::Mongodump.new({
         collection: 'test'
-      }).to_s.should == "mongodump --host localhost:27017 --db mongoid_shell_tests --collection test"
+      }).to_s.should == "mongodump --db mongoid_shell_tests --collection test"
     end
     it "includes query" do
-      Mongoid::Shell::Commands::Mongodump.new({ 
-        query: 'find x' 
-      }).to_s.should == 'mongodump --host localhost:27017 --db mongoid_shell_tests --query "find x"'
+      Mongoid::Shell::Commands::Mongodump.new({
+        query: 'find x'
+      }).to_s.should == 'mongodump --db mongoid_shell_tests --query "find x"'
+    end
+    [ :out, :dbpath ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongodump.new({
+          option => '/this is a folder'
+        }).to_s.should == "mongodump --db mongoid_shell_tests --#{option} \"/this is a folder\""
+      end
+    end
+    [ :directoryperdb, :journal, :oplog, :repair, :forceTableScan, :dbpath, :ipv6 ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongodump.new({
+          option => true
+        }).to_s.should == "mongodump --db mongoid_shell_tests --#{option}"
+      end
+    end
+  end
+  context "sessions" do
+    context "default" do
+      before :each do
+        @session = moped_session(:default)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongodump.new({
+          session: @session
+        }).to_s.should == "mongodump --db mongoid_shell_tests"
+      end
+    end
+    context "a replica set" do
+      before :each do
+        @session = moped_session(:replica_set)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongodump.new({
+          session: @session
+        }).to_s.should == "mongodump --host dedicated1.myapp.com:27017 --db mongoid --username user --password password"
+      end
+    end
+    context "url" do
+      before :each do
+        @session = moped_session(:url)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongodump.new({
+          session: @session
+        }).to_s.should == "mongodump --host 59.1.22.1:27017 --db mongoid --username user --password password"
+      end
     end
   end
 end

data/spec/mongoid/commands/mongorestore_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+
+describe Mongoid::Shell::Commands::Mongorestore do
+  include MopedSessionHelper
+  context "local" do
+    it "defaults to local" do
+      Mongoid::Shell::Commands::Mongorestore.new.to_s.should == "mongorestore --db mongoid_shell_tests"
+    end
+    it "includes collection" do
+      Mongoid::Shell::Commands::Mongorestore.new({
+        collection: 'test',
+        restore: 'folder'
+      }).to_s.should == "mongorestore --db mongoid_shell_tests --collection test folder"
+    end
+    it "can override the database, username, password and host" do
+      Mongoid::Shell::Commands::Mongorestore.new({
+        host: 'my_host',
+        username: 'my_username',
+        password: 'my_password',
+        collection: 'test',
+        db: 'my_db',
+        restore: 'folder'
+      }).to_s.should == "mongorestore --host my_host --db my_db --username my_username --password my_password --collection test folder"
+    end
+    [ :collection, :dbpath, :filter ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongorestore.new({
+          option => 'var arg',
+          restore: 'a folder'
+        }).to_s.should == "mongorestore --db mongoid_shell_tests --#{option} \"var arg\" \"a folder\""
+      end
+    end
+    [ :ipv6, :directoryperdb, :journal, :objcheck, :drop, :oplogReplay, :keepIndexVersion ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongorestore.new({
+          option => true
+        }).to_s.should == "mongorestore --db mongoid_shell_tests --#{option}"
+      end
+    end
+  end
+  context "sessions" do
+    context "default" do
+      before :each do
+        @session = moped_session(:default)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongorestore.new({
+          session: @session,
+          restore: "a folder"
+        }).to_s.should == "mongorestore --db mongoid_shell_tests \"a folder\""
+      end
+    end
+    context "a replica set" do
+      before :each do
+        @session = moped_session(:replica_set)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongorestore.new({
+          session: @session,
+          restore: "a folder"
+        }).to_s.should == "mongorestore --host dedicated1.myapp.com:27017 --db mongoid --username user --password password \"a folder\""
+      end
+    end
+    context "url" do
+      before :each do
+        @session = moped_session(:url)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongorestore.new({
+          session: @session,
+          restore: "a folder"
+        }).to_s.should == "mongorestore --host 59.1.22.1:27017 --db mongoid --username user --password password \"a folder\""
+      end
+    end
+  end
+end

data/spec/mongoid/commands/mongostat_spec.rb

@@ -0,0 +1,54 @@
+require 'spec_helper'
+
+describe Mongoid::Shell::Commands::Mongostat do
+  include MopedSessionHelper
+  context "local" do
+    it "defaults to local" do
+      Mongoid::Shell::Commands::Mongostat.new.to_s.should == "mongostat"
+    end
+    it "rowcount" do
+      Mongoid::Shell::Commands::Mongostat.new({
+        :rowcount => 10
+      }).to_s.should == "mongostat --rowcount 10"
+    end
+    [ :http, :discover, :all, :noheaders ].each do |option|
+      it "includes #{option}" do
+        Mongoid::Shell::Commands::Mongostat.new({
+          option => true
+        }).to_s.should == "mongostat --#{option}"
+      end
+    end
+  end
+  context "sessions" do
+    context "default" do
+      before :each do
+        @session = Mongoid::Sessions.with_name(:default)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongostat.new({
+          session: @session
+        }).to_s.should == "mongostat"
+      end
+    end
+    context "a replica set" do
+      before :each do
+        @session = moped_session(:replica_set)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongostat.new({
+          session: @session
+        }).to_s.should == "mongostat --host dedicated1.myapp.com:27017 --username user --password password"
+      end
+    end
+    context "url" do
+      before :each do
+        @session = moped_session(:url)
+      end
+      it "includes username and password" do
+        Mongoid::Shell::Commands::Mongostat.new({
+          session: @session
+        }).to_s.should == "mongostat --host 59.1.22.1:27017 --username user --password password"
+      end
+    end
+  end
+end

data/spec/mongoid/properties/host_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe Mongoid::Shell::Properties::Host do
+  subject do
+    klass = Class.new do
+      include Mongoid::Shell::Properties::Host
+
+      def session
+        Mongoid.default_session
+      end
+    end
+    klass.new
+  end
+  it "raises an exception when the session is not connected" do
+    Mongoid.default_session.cluster.stub(:nodes).and_return([])
+    expect {
+      subject.host
+    }.to raise_error Mongoid::Shell::Errors::SessionNotConnectedError, /Session is not connected./
+  end
+end

data/spec/mongoid/properties/primary_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe Mongoid::Shell::Properties::Primary do
+  subject do
+    klass = Class.new do
+      include Mongoid::Shell::Properties::Primary
+
+      def session
+        Mongoid.default_session
+      end
+    end
+    klass.new
+  end
+  it "raises an exception when the session is not connected" do
+    Mongoid.default_session.cluster.nodes.first.stub(:primary?).and_return(false)
+    expect {
+      subject.primary
+    }.to raise_error Mongoid::Shell::Errors::MissingPrimaryNodeError, /Session does not have a primary node./
+  end
+end

data/spec/spec_helper.rb

@@ -6,6 +6,12 @@ require 'rspec'
 require 'mongoid'
 require 'mongoid-shell'
 
+[ "support/helpers/*.rb" ].each do |path|
+  Dir["#{File.dirname(__FILE__)}/#{path}"].each do |file|
+    require file
+  end
+end
+
 Mongoid.configure do |config|
   config.connect_to('mongoid_shell_tests')
 end

data/spec/support/config/mongoid.yml

@@ -0,0 +1,37 @@
+# Tell Mongoid which environment this configuration is for.
+production:
+  # This starts the session configuration settings. You may have as
+  # many sessions as you like, but you must have at least 1 named
+  # 'default'.
+  sessions:
+    # Define the default session.
+    default:
+      # A session can have any number of hosts. Usually 1 for a single
+      # server setup, and at least 3 for a replica set. Hosts must be
+      # an array of host:port pairs. This session is single server.
+      hosts:
+        - flame.mongohq.com:27017
+      # Define the default database name.
+      database: mongoid
+      # Since this database points at a session connected to MongoHQ, we must
+      # provide the authentication details.
+      username: user
+      password: password
+    # This defines a secondary session at a replica set.
+    replica_set:
+      # This configuration is a 3 node replica set.
+      hosts:
+        - dedicated1.myapp.com:27017
+        - dedicated2.myapp.com:27017
+        - dedicated3.myapp.com:27017
+      database: mongoid
+      # We can set session specific options, like reads executing
+      # on secondary nodes, and defaulting the session to safe mode.
+      options:
+        consistency: :eventual
+        safe: true
+      username: user
+      password: password
+    # This configuration shows an authenticated replica set via a uri.
+    url:
+      uri: mongodb://user:password@59.1.22.1:27017,59.1.22.2:27017/mongoid

data/spec/support/helpers/moped_session_helper.rb

@@ -0,0 +1,11 @@
+module MopedSessionHelper
+  # returns a Moped session with stubbed address resolution
+  def moped_session(name)
+    config = File.join(File.dirname(__FILE__), "../../support/config/mongoid.yml")
+    Mongoid.load! config, :production
+    Moped::Node.any_instance.stub(:resolve_address)
+    session = Mongoid::Sessions.with_name(name)
+    session.cluster.nodes.last.stub(:primary?).and_return(true)
+    session
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongoid-shell
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-27 00:00:00.000000000 Z
+date: 2013-01-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mongoid
@@ -61,22 +61,35 @@ files:
 - lib/mongoid-shell.rb
 - lib/mongoid/shell/commands.rb
 - lib/mongoid/shell/commands/base.rb
+- lib/mongoid/shell/commands/mongo.rb
 - lib/mongoid/shell/commands/mongodump.rb
+- lib/mongoid/shell/commands/mongorestore.rb
+- lib/mongoid/shell/commands/mongostat.rb
 - lib/mongoid/shell/errors.rb
 - lib/mongoid/shell/errors/base.rb
+- lib/mongoid/shell/errors/missing_primary_node_error.rb
 - lib/mongoid/shell/errors/missing_session_error.rb
+- lib/mongoid/shell/errors/session_not_connected_error.rb
 - lib/mongoid/shell/properties.rb
 - lib/mongoid/shell/properties/database.rb
 - lib/mongoid/shell/properties/host.rb
 - lib/mongoid/shell/properties/password.rb
+- lib/mongoid/shell/properties/primary.rb
 - lib/mongoid/shell/properties/username.rb
 - lib/mongoid/shell/version.rb
 - lib/mongoid_shell.rb
 - mongoid-shell.gemspec
 - spec/mongoid/commands/base_spec.rb
+- spec/mongoid/commands/mongo_spec.rb
 - spec/mongoid/commands/mongodump_spec.rb
+- spec/mongoid/commands/mongorestore_spec.rb
+- spec/mongoid/commands/mongostat_spec.rb
 - spec/mongoid/mongoid_shell_spec.rb
+- spec/mongoid/properties/host_spec.rb
+- spec/mongoid/properties/primary_spec.rb
 - spec/spec_helper.rb
+- spec/support/config/mongoid.yml
+- spec/support/helpers/moped_session_helper.rb
 homepage: http://github.com/dblock/mongoid-shell
 licenses:
 - MIT
@@ -92,7 +105,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 408535153
+      hash: 4237627171487641625
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: