kbaum-mongo 0.18.3p

data/test/db_connection_test.rb

@@ -0,0 +1,16 @@
+require 'test/test_helper'
+
+# NOTE: assumes Mongo is running
+class DBConnectionTest < Test::Unit::TestCase
+
+  include Mongo
+
+  def test_no_exceptions
+    host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+    port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+    db = Connection.new(host, port).db('ruby-mongo-demo')
+    coll = db.collection('test')
+    coll.remove
+    db.error
+  end
+end

data/test/db_test.rb

@@ -0,0 +1,272 @@
+require 'test/test_helper'
+require 'digest/md5'
+require 'stringio'
+require 'logger'
+
+class TestPKFactory
+  def create_pk(row)
+    row['_id'] ||= Mongo::ObjectID.new
+    row
+  end
+end
+
+# NOTE: assumes Mongo is running
+class DBTest < Test::Unit::TestCase
+
+  include Mongo
+
+  @@host  = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+  @@port  = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+  @@conn  = Connection.new(@@host, @@port)
+  @@db    = @@conn.db('ruby-mongo-test')
+  @@users = @@db.collection('system.users')
+
+  def test_close
+    @@conn.close
+    assert !@@conn.connected?
+    begin
+      @@db.collection('test').insert('a' => 1)
+      fail "expected 'NilClass' exception"
+    rescue => ex
+      assert_match /NilClass/, ex.to_s
+    ensure
+      @@db = Connection.new(@@host, @@port).db('ruby-mongo-test')
+      @@users = @@db.collection('system.users')
+    end
+  end
+  
+  def test_logger
+    output = StringIO.new
+    logger = Logger.new(output)
+    logger.level = Logger::DEBUG
+    conn = Connection.new(@host, @port, :logger => logger)
+    assert_equal logger, conn.logger
+    
+    conn.logger.debug 'testing'
+    assert output.string.include?('testing')
+  end
+
+  def test_full_coll_name
+    coll = @@db.collection('test')
+    assert_equal 'ruby-mongo-test.test', @@db.full_collection_name(coll.name)
+  end
+
+  def test_collection_names
+    @@db.collection("test").insert("foo" => 5)
+    @@db.collection("test.mike").insert("bar" => 0)
+
+    colls = @@db.collection_names()
+    assert colls.include?("test")
+    assert colls.include?("test.mike")
+    colls.each { |name|
+      assert !name.include?("$")
+    }
+  end
+
+  def test_collections
+    @@db.collection("test.durran").insert("foo" => 5)
+    @@db.collection("test.les").insert("bar" => 0)
+
+    colls = @@db.collections()
+    assert_not_nil colls.select { |coll| coll.name == "test.durran" }
+    assert_not_nil colls.select { |coll| coll.name == "test.les" }
+    assert_equal [], colls.select { |coll| coll.name == "does_not_exist" }
+
+    assert_kind_of Collection, colls[0]
+  end
+
+  def test_pair
+    @@conn.close
+    @@users = nil
+    @@conn  = Connection.new({:left => "this-should-fail", :right => [@@host, @@port]})
+    @@db    = @@conn['ruby-mongo-test']
+    assert @@conn.connected?
+  ensure
+    unless @@conn.connected?
+      @@conn = Connection.new(@@host, @@port) 
+      @@db   = @@conn.db('ruby-mongo-test')
+    end
+    @@users = @@db.collection('system.users')
+  end
+
+  def test_pk_factory
+    db = Connection.new(@@host, @@port).db('ruby-mongo-test', :pk => TestPKFactory.new)
+    coll = db.collection('test')
+    coll.remove
+
+    insert_id = coll.insert('name' => 'Fred', 'age' => 42)
+    # new id gets added to returned object
+    row = coll.find_one({'name' => 'Fred'})
+    oid = row['_id']
+    assert_not_nil oid
+    assert_equal insert_id, oid
+
+    oid = ObjectID.new
+    data = {'_id' => oid, 'name' => 'Barney', 'age' => 41}
+    coll.insert(data)
+    row = coll.find_one({'name' => data['name']})
+    db_oid = row['_id']
+    assert_equal oid, db_oid
+    assert_equal data, row
+
+    coll.remove
+  end
+
+  def test_pk_factory_reset
+    conn = Connection.new(@@host, @@port)
+    db   = conn.db('ruby-mongo-test')
+    db.pk_factory = Object.new # first time
+    begin
+      db.pk_factory = Object.new
+      fail "error: expected exception"
+    rescue => ex
+      assert_match /Cannot change/, ex.to_s
+    ensure
+      conn.close
+    end
+  end
+
+  def test_authenticate
+    @@db.add_user('spongebob', 'squarepants')
+    assert_raise Mongo::AuthenticationError do
+      assert !@@db.authenticate('nobody', 'nopassword')
+    end
+    assert_raise Mongo::AuthenticationError do
+      assert !@@db.authenticate('spongebob' , 'squareliederhosen')
+    end
+    assert @@db.authenticate('spongebob', 'squarepants')
+    @@db.logout
+    @@db.remove_user('spongebob')
+  end
+
+  def test_authenticate_with_connection_uri
+    @@db.add_user('spongebob', 'squarepants')
+    assert Mongo::Connection.from_uri("mongodb://spongebob:squarepants@localhost/#{@@db.name}")
+
+    assert_raise Mongo::AuthenticationError do
+      Mongo::Connection.from_uri("mongodb://wrong:info@localhost/#{@@db.name}")
+    end
+  end
+
+  def test_logout
+    assert @@db.logout
+  end
+
+  def test_error
+    @@db.reset_error_history
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert_nil @@db.previous_error
+
+    @@db.send(:command, :forceerror => 1)
+    assert @@db.error?
+    assert_not_nil @@db.error
+    assert_not_nil @@db.previous_error
+
+    @@db.send(:command, :forceerror => 1)
+    assert @@db.error?
+    assert @@db.error
+    prev_error = @@db.previous_error
+    assert_equal 1, prev_error['nPrev']
+    assert_equal prev_error["err"], @@db.error
+
+    @@db.collection('test').find_one
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert @@db.previous_error
+    assert_equal 2, @@db.previous_error['nPrev']
+
+    @@db.reset_error_history
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert_nil @@db.previous_error
+  end
+
+  def test_check_command_response
+    command = {:forceerror => 1}
+    assert_raise OperationFailure do 
+      @@db.command(command, false, true)
+    end
+  end
+
+  def test_last_status
+    @@db['test'].remove
+    @@db['test'].save("i" => 1)
+
+    @@db['test'].update({"i" => 1}, {"$set" => {"i" => 2}})
+    assert @@db.last_status()["updatedExisting"]
+
+    @@db['test'].update({"i" => 1}, {"$set" => {"i" => 500}})
+    assert !@@db.last_status()["updatedExisting"]
+  end
+
+  def test_text_port_number_raises_no_errors
+    conn = Connection.new(@@host, @@port.to_s)
+    db   = conn['ruby-mongo-test']
+    assert db.collection('users').remove
+  end
+
+  def test_user_management
+    @@db.add_user("bob", "secret")
+    assert @@db.authenticate("bob", "secret")
+    @@db.logout
+    assert @@db.remove_user("bob")
+    assert_raise Mongo::AuthenticationError do
+      @@db.authenticate("bob", "secret")
+    end
+  end
+
+  def test_remove_non_existant_user
+    assert !@@db.remove_user("joe")
+  end
+
+  context "database profiling" do
+    setup do
+      @db  = @@conn['ruby-mongo-test-admin-functions']
+      @coll = @db['test']
+      @coll.remove
+      @r1 = @coll.insert('a' => 1) # collection not created until it's used
+    end
+
+    should "set default profiling level" do
+      assert_equal :off, @db.profiling_level
+    end
+
+    should "change profiling level" do
+      @db.profiling_level = :slow_only
+      assert_equal :slow_only, @db.profiling_level
+      @db.profiling_level = :off
+      assert_equal :off, @db.profiling_level
+      @db.profiling_level = :all
+      assert_equal :all, @db.profiling_level
+      begin
+        @db.profiling_level = :medium
+        fail "shouldn't be able to do this"
+      rescue
+      end
+    end
+
+    should "return profiling info" do
+      @db.profiling_level = :all
+      @coll.find()
+      @db.profiling_level = :off
+
+      info = @db.profiling_info
+      assert_kind_of Array, info
+      assert info.length >= 1
+      first = info.first
+      assert_kind_of String, first['info']
+      assert_kind_of Time, first['ts']
+      assert_kind_of Numeric, first['millis']
+    end
+
+    should "validate collection" do
+      doc = @db.validate_collection(@coll.name)
+      assert_not_nil doc
+      result = doc['result']
+      assert_not_nil result
+      assert_match /firstExtent/, result
+    end
+
+  end
+end

data/test/grid_file_system_test.rb

@@ -0,0 +1,210 @@
+require 'test/test_helper'
+include Mongo
+
+class GridFileSystemTest < Test::Unit::TestCase
+  context "GridFileSystem:" do
+    setup do
+      @con = Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
+        ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT)
+      @db = @con.db('mongo-ruby-test')
+    end
+
+    teardown do
+      @db['fs.files'].remove
+      @db['fs.chunks'].remove
+    end
+
+    context "When reading:" do
+      setup do
+        @chunks_data = "CHUNKS" * 50000
+        @grid = GridFileSystem.new(@db)
+        @grid.open('sample.file', 'w') do |f|
+          f.write @chunks_data
+        end
+
+        @grid = GridFileSystem.new(@db)
+      end
+
+      should "read sample data" do
+        data = @grid.open('sample.file', 'r') { |f| f.read }
+        assert_equal data.length, @chunks_data.length
+      end
+
+      should "return an empty string if length is zero" do
+        data = @grid.open('sample.file', 'r') { |f| f.read(0) }
+        assert_equal '', data
+      end
+
+      should "return the first n bytes" do
+        data = @grid.open('sample.file', 'r') {|f| f.read(288888) }
+        assert_equal 288888, data.length
+        assert_equal @chunks_data[0...288888], data
+      end
+
+      should "return the first n bytes even with an offset" do
+        data = @grid.open('sample.file', 'r') do |f| 
+          f.seek(1000)
+          f.read(288888)
+        end
+        assert_equal 288888, data.length
+        assert_equal @chunks_data[1000...289888], data
+      end
+    end
+
+    context "When writing:" do
+     setup do
+       @data   = "BYTES" * 50
+       @grid = GridFileSystem.new(@db)
+       @grid.open('sample', 'w') do |f|
+         f.write @data
+       end
+     end
+
+     should "read sample data" do
+       data = @grid.open('sample', 'r') { |f| f.read }
+       assert_equal data.length, @data.length
+     end
+
+     should "return the total number of bytes written" do
+       data = 'a' * 300000
+       assert_equal 300000, @grid.open('sample', 'w') {|f| f.write(data) }
+     end
+
+     should "more read sample data" do
+       data = @grid.open('sample', 'r') { |f| f.read }
+       assert_equal data.length, @data.length
+     end
+
+     should "raise exception if not opened for write" do
+       assert_raise GridError do
+         @grid.open('io', 'r') { |f| f.write('hello') }
+       end
+     end
+
+     context "and when overwriting the file" do
+       setup do
+         @old = @grid.open('sample', 'r')
+
+         @new_data = "DATA" * 10
+         sleep(2)
+         @grid.open('sample', 'w') do |f|
+           f.write @new_data
+         end
+
+         @new = @grid.open('sample', 'r')
+       end
+
+       should "have a newer upload date" do
+         assert @new.upload_date > @old.upload_date, "New data is not greater than old date."
+       end
+
+       should "have a different files_id" do
+         assert_not_equal @new.files_id, @old.files_id
+       end
+
+       should "contain the new data" do
+         assert_equal @new_data, @new.read, "Expected DATA"
+       end
+
+       context "and on a second overwrite" do
+         setup do
+           sleep(2)
+           new_data = "NEW" * 1000
+           @grid.open('sample', 'w') do |f|
+             f.write new_data
+           end
+
+           @ids = @db['fs.files'].find({'filename' => 'sample'}).map {|file| file['_id']}
+         end
+
+         should "write a third version of the file" do
+           assert_equal 3, @db['fs.files'].find({'filename' => 'sample'}).count
+           assert_equal 3, @db['fs.chunks'].find({'files_id' => {'$in' => @ids}}).count
+         end
+
+         should "remove all versions and their data on delete" do
+           @grid.delete('sample')
+           assert_equal 0, @db['fs.files'].find({'filename' => 'sample'}).count
+           assert_equal 0, @db['fs.chunks'].find({'files_id' => {'$in' => @ids}}).count
+         end
+       end
+     end
+   end
+
+   context "When writing chunks:" do
+      setup do
+        data   = "B" * 50000
+        @grid = GridFileSystem.new(@db)
+        @grid.open('sample', 'w', :chunk_size => 1000) do |f|
+          f.write data
+        end
+      end
+
+      should "write the correct number of chunks" do
+        file   = @db['fs.files'].find_one({:filename => 'sample'})
+        chunks = @db['fs.chunks'].find({'files_id' => file['_id']}).to_a
+        assert_equal 50, chunks.length
+      end
+    end
+
+    context "Positioning:" do
+      setup do
+        data = 'hello, world' + '1' * 5000 + 'goodbye!' + '2' * 1000 + '!'
+        @grid = GridFileSystem.new(@db)
+        @grid.open('hello', 'w', :chunk_size => 1000) do |f|
+          f.write data
+        end
+      end
+
+      should "seek within chunks" do
+        @grid.open('hello', 'r') do |f|
+          f.seek(0)
+          assert_equal 'h', f.read(1)
+          f.seek(7)
+          assert_equal 'w', f.read(1)
+          f.seek(4)
+          assert_equal 'o', f.read(1)
+          f.seek(0)
+          f.seek(7, IO::SEEK_CUR)
+          assert_equal 'w', f.read(1)
+          f.seek(-2, IO::SEEK_CUR)
+          assert_equal ' ', f.read(1)
+          f.seek(-4, IO::SEEK_CUR)
+          assert_equal 'l', f.read(1)
+          f.seek(3, IO::SEEK_CUR)
+          assert_equal 'w', f.read(1)
+        end
+      end
+
+      should "seek between chunks" do
+        @grid.open('hello', 'r') do |f|
+          f.seek(1000)
+          assert_equal '11111', f.read(5)
+
+          f.seek(5009)
+          assert_equal '111goodbye!222', f.read(14)
+
+          f.seek(-1, IO::SEEK_END)
+          assert_equal '!', f.read(1)
+          f.seek(-6, IO::SEEK_END)
+          assert_equal '2', f.read(1)
+        end
+      end
+
+      should "tell the current position" do
+        @grid.open('hello', 'r') do |f|
+          assert_equal 0, f.tell
+
+          f.seek(999)
+          assert_equal 999, f.tell
+        end
+      end
+
+      should "seek only in read mode" do
+        assert_raise GridError do
+          @grid.open('hello', 'w') {|f| f.seek(0) }
+        end
+      end
+    end
+  end
+end

data/test/grid_io_test.rb

@@ -0,0 +1,78 @@
+require 'test/test_helper'
+include Mongo
+
+class GridIOTest < Test::Unit::TestCase
+
+  context "GridIO" do
+    setup do
+      @db ||= Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
+        ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT).db('ruby-mongo-test')
+      @files  = @db.collection('fs.files')
+      @chunks = @db.collection('fs.chunks')
+      @chunks.create_index([['files_id', Mongo::ASCENDING], ['n', Mongo::ASCENDING]])
+    end
+
+    teardown do
+      @files.remove
+      @chunks.remove
+    end
+
+    context "Options" do
+      setup do
+        @filename = 'test'
+        @mode     = 'w'
+      end
+
+      should "set default 256k chunk size" do
+        file = GridIO.new(@files, @chunks, @filename, @mode)
+        assert_equal 256 * 1024, file.chunk_size
+      end
+
+      should "set chunk size" do
+        file = GridIO.new(@files, @chunks, @filename, @mode, :chunk_size => 1000)
+        assert_equal 1000, file.chunk_size
+      end
+    end
+
+    context "Grid MD5 check" do
+      should "run in safe mode" do
+        file = GridIO.new(@files, @chunks, 'smallfile', 'w', :safe => true)
+        file.write("DATA" * 100)
+        assert file.close
+        assert_equal file.server_md5, file.client_md5
+      end
+
+      should "validate with a large file" do
+        io = File.open(File.join(File.dirname(__FILE__), 'data', 'sample_file.pdf'), 'r')
+        file = GridIO.new(@files, @chunks, 'bigfile', 'w', :safe => true)
+        file.write(io)
+        assert file.close
+        assert_equal file.server_md5, file.client_md5
+      end
+    end
+
+    context "Content types" do
+
+      if defined?(MIME)
+        should "determine common content types from the extension" do
+          file = GridIO.new(@files, @chunks, 'sample.pdf', 'w')
+          assert_equal 'application/pdf', file.content_type
+
+          file = GridIO.new(@files, @chunks, 'sample.txt', 'w')
+          assert_equal 'text/plain', file.content_type
+        end
+      end
+
+      should "default to binary/octet-stream when type is unknown" do
+        file = GridIO.new(@files, @chunks, 'sample.l33t', 'w')
+        assert_equal 'binary/octet-stream', file.content_type
+      end
+
+      should "use any provided content type by default" do
+        file = GridIO.new(@files, @chunks, 'sample.l33t', 'w', :content_type => 'image/jpg')
+        assert_equal 'image/jpg', file.content_type
+      end
+    end
+  end
+
+end