mongodb-mongo 0.2.0 → 0.2.1

data/lib/mongo/db.rb

@@ -15,7 +15,7 @@
 # ++
 
 require 'socket'
-require 'md5'
+require 'digest/md5'
 require 'mutex_m'
 require 'mongo/collection'
 require 'mongo/message'
@@ -144,10 +144,16 @@ module XGen
           auth['authenticate'] = 1
           auth['user'] = username
           auth['nonce'] = nonce
-          auth['key'] = MD5.md5("#{nonce}#{username}#{hash_password(password)}").to_s
+          auth['key'] = Digest::MD5.hexdigest("#{nonce}#{username}#{hash_password(password)}")
           ok?(db_command(auth))
         end
 
+        # Deauthorizes use for this database for this connection.
+        def logout
+          doc = db_command(:logout => 1)
+          raise "error logging out: #{doc.inspect}" unless ok?(doc)
+        end
+
         # Returns an array of collection names. Each name is of the form
         # "database_name.collection_name".
         def collection_names
@@ -425,7 +431,7 @@ module XGen
         private
 
         def hash_password(plaintext)
-          MD5.new("mongo#{plaintext}").to_s
+          Digest::MD5.hexdigest("mongo#{plaintext}")
         end
 
       end

data/mongo-ruby-driver.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = 'mongo'
-  s.version = '0.2.0'
+  s.version = '0.2.1'
   s.platform = Gem::Platform::RUBY
   s.summary = 'Simple pure-Ruby driver for the 10gen Mongo DB'
   s.description = 'A pure-Ruby driver for the 10gen Mongo DB. For more information about Mongo, see http://www.mongodb.org.'

data/tests/test_db.rb

@@ -1,5 +1,5 @@
 $LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
-require 'md5'
+require 'digest/md5'
 require 'mongo'
 require 'test/unit'
 
@@ -19,12 +19,17 @@ class DBTest < Test::Unit::TestCase
     @host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
     @port = ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT
     @db = Mongo.new(@host, @port).db('ruby-mongo-test')
+
     @spongebob = 'spongebob'
     @spongebob_password = 'squarepants'
+    @users = @db.collection('system.users')
+    @users.clear
+    @db.add_user(@spongebob, @spongebob_password)
   end
 
   def teardown
     if @db.connected?
+      @users.clear if @users
       @db.close
     end
   end
@@ -54,6 +59,7 @@ class DBTest < Test::Unit::TestCase
 
   def test_array
     @db.close
+    @users = nil
     @db = Mongo.new([["nosuch.example.com"], [@host, @port]]).db('ruby-mongo-test')
     assert @db.connected?
   end
@@ -86,45 +92,25 @@ class DBTest < Test::Unit::TestCase
   end
 
   def test_add_user
-    coll = @db.collection('system.users')
-    coll.clear
-    begin
-      assert_equal 0, coll.count
-      @db.add_user(@spongebob, @spongebob_password)
-      assert_equal 1, coll.count
-      doc = coll.find({}, :limit => 1).next_object
-      assert_equal @spongebob, doc['user']
-      assert_equal MD5.new("mongo#{@spongebob_password}").to_s, doc['pwd']
-    ensure
-      coll.clear
-    end
+    assert_equal 1, @users.count
+    doc = @users.find({}, :limit => 1).next_object
+    assert_equal @spongebob, doc['user']
+    assert_equal Digest::MD5.hexdigest("mongo#{@spongebob_password}"), doc['pwd']
   end
 
   def test_delete_user
-    coll = @db.collection('system.users')
-    coll.clear
-    begin
-      assert_equal 0, coll.count
-      @db.add_user(@spongebob, @spongebob_password)
-      assert_equal 1, coll.count
-      @db.delete_user(@spongebob)
-      assert_equal 0, coll.count
-    ensure
-      coll.clear
-    end
+    @db.delete_user(@spongebob)
+    assert_equal 0, @users.count
   end
 
   def test_authenticate
-    coll = @db.collection('system.users')
-    coll.clear
-    begin
-      @db.add_user(@spongebob, @spongebob_password)
-      assert !@db.authenticate('nobody', 'nopassword')
-      assert !@db.authenticate(@spongebob, 'squareliederhosen')
-      assert @db.authenticate(@spongebob, @spongebob_password)
-    ensure
-      coll.clear
-    end
+    assert !@db.authenticate('nobody', 'nopassword')
+    assert !@db.authenticate(@spongebob, 'squareliederhosen')
+    assert @db.authenticate(@spongebob, @spongebob_password)
+  end
+
+  def test_logout
+    @db.logout                  # only testing that we don't throw exception
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mongodb-mongo
 version: !ruby/object:Gem::Version 
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors: 
 - Jim Menard