animehunter-mongo 0.9

data/tests/test_db_connection.rb

@@ -0,0 +1,18 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'test/unit'
+
+# NOTE: assumes Mongo is running
+class DBConnectionTest < Test::Unit::TestCase
+
+  include XGen::Mongo::Driver
+
+  def test_no_exceptions
+    host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+    port = ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT
+    db = Mongo.new(host, port).db('ruby-mongo-demo')
+    coll = db.collection('test')
+    coll.clear
+    db.error
+  end
+end

data/tests/test_grid_store.rb

@@ -0,0 +1,284 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'test/unit'
+require 'mongo'
+require 'mongo/gridfs'
+
+class GridStoreTest < Test::Unit::TestCase
+
+  include XGen::Mongo::Driver
+  include XGen::Mongo::GridFS
+
+  @@db = Mongo.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
+                   ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT).db('ruby-mongo-test')
+  @@files = @@db.collection('fs.files')
+  @@chunks = @@db.collection('fs.chunks')
+
+  def setup
+    @@chunks.clear
+    @@files.clear
+    GridStore.open(@@db, 'foobar', 'w') { |f| f.write("hello, world!") }
+  end
+
+  def teardown
+    @@chunks.clear
+    @@files.clear
+    @@db.error
+  end
+
+  def test_exist
+    assert GridStore.exist?(@@db, 'foobar')
+    assert !GridStore.exist?(@@db, 'does_not_exist')
+    assert !GridStore.exist?(@@db, 'foobar', 'another_root')
+  end
+
+  def test_list
+    assert_equal ['foobar'], GridStore.list(@@db)
+    assert_equal ['foobar'], GridStore.list(@@db, 'fs')
+    assert_equal [], GridStore.list(@@db, 'my_fs')
+
+    GridStore.open(@@db, 'test', 'w') { |f| f.write("my file") }
+
+    assert_equal ['foobar', 'test'], GridStore.list(@@db)
+  end
+
+  def test_small_write
+    rows = @@files.find({'filename' => 'foobar'}).to_a
+    assert_not_nil rows
+    assert_equal 1, rows.length
+    row = rows[0]
+    assert_not_nil row
+
+    file_id = row['_id']
+    assert_kind_of ObjectID, file_id
+    rows = @@chunks.find({'files_id' => file_id}).to_a
+    assert_not_nil rows
+    assert_equal 1, rows.length
+  end
+
+  def test_small_file
+    rows = @@files.find({'filename' => 'foobar'}).to_a
+    assert_not_nil rows
+    assert_equal 1, rows.length
+    row = rows[0]
+    assert_not_nil row
+    assert_equal "hello, world!", GridStore.read(@@db, 'foobar')
+  end
+
+  def test_overwrite
+    GridStore.open(@@db, 'foobar', 'w') { |f| f.write("overwrite") }
+    assert_equal "overwrite", GridStore.read(@@db, 'foobar')
+  end
+
+  def test_read_length
+    assert_equal "hello", GridStore.read(@@db, 'foobar', 5)
+  end
+
+  # Also tests seek
+  def test_read_with_offset
+    assert_equal "world", GridStore.read(@@db, 'foobar', 5, 7)
+    assert_equal "world!", GridStore.read(@@db, 'foobar', nil, 7)
+  end
+
+  def test_seek
+    GridStore.open(@@db, 'foobar', 'r') { |f|
+      f.seek(0)
+      assert_equal 'h', f.getc.chr
+      f.seek(7)
+      assert_equal 'w', f.getc.chr
+      f.seek(4)
+      assert_equal 'o', f.getc.chr
+
+      f.seek(-1, IO::SEEK_END)
+      assert_equal '!', f.getc.chr
+      f.seek(-6, IO::SEEK_END)
+      assert_equal 'w', f.getc.chr
+
+      f.seek(0)
+      f.seek(7, IO::SEEK_CUR)
+      assert_equal 'w', f.getc.chr
+      f.seek(-1, IO::SEEK_CUR)
+      assert_equal 'w', f.getc.chr
+      f.seek(-4, IO::SEEK_CUR)
+      assert_equal 'o', f.getc.chr
+      f.seek(3, IO::SEEK_CUR)
+      assert_equal 'o', f.getc.chr
+    }
+  end
+
+  def test_multi_chunk
+    @@chunks.clear
+    @@files.clear
+
+    size = 512
+    GridStore.open(@@db, 'biggie', 'w') { |f|
+      f.chunk_size = size
+      f.write('x' * size)
+      f.write('y' * size)
+      f.write('z' * size)
+    }
+
+    assert_equal 3, @@chunks.count
+    assert_equal ('x' * size) + ('y' * size) + ('z' * size), GridStore.read(@@db, 'biggie')
+  end
+
+  def test_puts_and_readlines
+    GridStore.open(@@db, 'multiline', 'w') { |f|
+      f.puts "line one"
+      f.puts "line two\n"
+      f.puts "line three"
+    }
+
+    lines = GridStore.readlines(@@db, 'multiline')
+    assert_equal ["line one\n", "line two\n", "line three\n"], lines
+  end
+
+  def test_unlink
+    assert_equal 1, @@files.count
+    assert_equal 1, @@chunks.count
+    GridStore.unlink(@@db, 'foobar')
+    assert_equal 0, @@files.count
+    assert_equal 0, @@chunks.count
+  end
+
+  def test_append
+    GridStore.open(@@db, 'foobar', 'w+') { |f| f.write(" how are you?") }
+    assert_equal 1, @@chunks.count
+    assert_equal "hello, world! how are you?", GridStore.read(@@db, 'foobar')
+  end
+
+  def test_rewind_and_truncate_on_write
+    GridStore.open(@@db, 'foobar', 'w') { |f|
+      f.write("some text is inserted here")
+      f.rewind
+      f.write("abc")
+    }
+    assert_equal "abc", GridStore.read(@@db, 'foobar')
+  end
+
+  def test_tell
+    GridStore.open(@@db, 'foobar', 'r') { |f|
+      f.read(5)
+      assert_equal 5, f.tell
+    }
+  end
+
+  def test_empty_block_ok
+    GridStore.open(@@db, 'empty', 'w')
+  end
+
+  def test_save_empty_file
+    @@chunks.clear
+    @@files.clear
+    GridStore.open(@@db, 'empty', 'w') {} # re-write with zero bytes
+    assert_equal 1, @@files.count
+    assert_equal 0, @@chunks.count
+  end
+
+  def test_empty_file_eof
+    GridStore.open(@@db, 'empty', 'w')
+    GridStore.open(@@db, 'empty', 'r') { |f|
+      assert f.eof?
+    }
+  end
+
+  def test_cannot_change_chunk_size_on_read
+    begin
+      GridStore.open(@@db, 'foobar', 'r') { |f| f.chunk_size = 42 }
+      fail "should have seen error"
+    rescue => ex
+      assert_match /error: can only change chunk size/, ex.to_s
+    end
+  end
+
+  def test_cannot_change_chunk_size_after_data_written
+    begin
+      GridStore.open(@@db, 'foobar', 'w') { |f|
+        f.write("some text")
+        f.chunk_size = 42
+      }
+      fail "should have seen error"
+    rescue => ex
+      assert_match /error: can only change chunk size/, ex.to_s
+    end
+  end
+
+  def test_change_chunk_size
+    GridStore.open(@@db, 'new-file', 'w') { |f|
+      f.chunk_size = 42
+      f.write("foo")
+    }
+    GridStore.open(@@db, 'new-file', 'r') { |f|
+      assert f.chunk_size == 42
+    }
+  end
+
+  def test_chunk_size_in_option
+    GridStore.open(@@db, 'new-file', 'w', :chunk_size => 42) { |f| f.write("foo") }
+    GridStore.open(@@db, 'new-file', 'r') { |f|
+      assert f.chunk_size == 42
+    }
+  end
+
+  def test_md5
+    GridStore.open(@@db, 'new-file', 'w') { |f| f.write("hello world\n")}
+    GridStore.open(@@db, 'new-file', 'r') { |f|
+      assert f.md5 == '6f5902ac237024bdd0c176cb93063dc4'
+      begin
+        f.md5 = 'cant do this'
+        fail "should have seen error"
+      rescue => ex
+        true
+      end
+    }
+    GridStore.open(@@db, 'new-file', 'w') {}
+    GridStore.open(@@db, 'new-file', 'r') { |f|
+      assert f.md5 == 'd41d8cd98f00b204e9800998ecf8427e'
+    }
+  end
+
+  def test_upload_date
+    now = Time.now
+    orig_file_upload_date = nil
+    GridStore.open(@@db, 'foobar', 'r') { |f| orig_file_upload_date = f.upload_date }
+    assert_not_nil orig_file_upload_date
+    assert (orig_file_upload_date - now) < 5 # even a really slow system < 5 secs
+
+    sleep(2)
+    GridStore.open(@@db, 'foobar', 'w') { |f| f.write "new data" }
+    file_upload_date = nil
+    GridStore.open(@@db, 'foobar', 'r') { |f| file_upload_date = f.upload_date }
+    assert_equal orig_file_upload_date, file_upload_date
+  end
+
+  def test_content_type
+    ct = nil
+    GridStore.open(@@db, 'foobar', 'r') { |f| ct = f.content_type }
+    assert_equal GridStore::DEFAULT_CONTENT_TYPE, ct
+
+    GridStore.open(@@db, 'foobar', 'w+') { |f| f.content_type = 'text/html' }
+    ct2 = nil
+    GridStore.open(@@db, 'foobar', 'r') { |f| ct2 = f.content_type }
+    assert_equal 'text/html', ct2
+  end
+
+  def test_content_type_option
+    GridStore.open(@@db, 'new-file', 'w', :content_type => 'image/jpg') { |f| f.write('foo') }
+    ct = nil
+    GridStore.open(@@db, 'new-file', 'r') { |f| ct = f.content_type }
+    assert_equal 'image/jpg', ct
+  end
+
+  def test_unknown_mode
+    GridStore.open(@@db, 'foobar', 'x')
+    fail 'should have seen "illegal mode" error raised'
+  rescue => ex
+    assert_equal "error: illegal mode x", ex.to_s
+  end
+
+  def test_metadata
+    GridStore.open(@@db, 'foobar', 'r') { |f| assert_nil f.metadata }
+    GridStore.open(@@db, 'foobar', 'w+') { |f| f.metadata = {'a' => 1} }
+    GridStore.open(@@db, 'foobar', 'r') { |f| assert_equal({'a' => 1}, f.metadata) }
+  end
+
+end

data/tests/test_message.rb

@@ -0,0 +1,35 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'test/unit'
+
+class MessageTest < Test::Unit::TestCase
+
+  include XGen::Mongo::Driver
+
+  def setup
+    @msg = Message.new(42)
+  end
+
+  def test_initial_info
+    assert_equal Message::HEADER_SIZE, @msg.buf.length
+    @msg.write_long(1029)
+    @msg.buf.rewind
+    assert_equal Message::HEADER_SIZE + 8, @msg.buf.get_int
+    @msg.buf.get_int            # skip message id
+    assert_equal 0, @msg.buf.get_int
+    assert_equal 42,  @msg.buf.get_int
+    assert_equal 1029, @msg.buf.get_long
+  end
+
+  def test_update_length
+    @msg.update_message_length
+    @msg.buf.rewind
+    assert_equal Message::HEADER_SIZE, @msg.buf.get_int
+  end
+
+  def test_long_length
+    @msg.write_long(1027)
+    assert_equal Message::HEADER_SIZE + 8, @msg.buf.length
+  end
+
+end

data/tests/test_mongo.rb

@@ -0,0 +1,78 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'test/unit'
+
+# NOTE: assumes Mongo is running
+class MongoTest < Test::Unit::TestCase
+
+  include XGen::Mongo::Driver
+
+  def setup
+    @host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+    @port = ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT
+    @mongo = Mongo.new(@host, @port)
+  end
+
+  def teardown
+    @mongo.db('ruby-mongo-test').error
+  end
+
+  def test_database_info
+    @mongo.drop_database('ruby-mongo-info-test')
+    @mongo.db('ruby-mongo-info-test').collection('info-test').insert('a' => 1)
+
+    info = @mongo.database_info
+    assert_not_nil info
+    assert_kind_of Hash, info
+    assert_not_nil info['ruby-mongo-info-test']
+    assert info['ruby-mongo-info-test'] > 0
+
+    @mongo.drop_database('ruby-mongo-info-test')
+  end
+
+  def test_database_names
+    @mongo.drop_database('ruby-mongo-info-test')
+    @mongo.db('ruby-mongo-info-test').collection('info-test').insert('a' => 1)
+
+    names = @mongo.database_names
+    assert_not_nil names
+    assert_kind_of Array, names
+    assert names.length >= 1
+    assert names.include?('ruby-mongo-info-test')
+
+    @mongo.drop_database('ruby-mongo-info-test')
+  end
+
+  def test_drop_database
+    db = @mongo.db('ruby-mongo-will-be-deleted')
+    coll = db.collection('temp')
+    coll.clear
+    coll.insert(:name => 'temp')
+    assert_equal 1, coll.count()
+    assert @mongo.database_names.include?('ruby-mongo-will-be-deleted')
+
+    @mongo.drop_database('ruby-mongo-will-be-deleted')
+    assert !@mongo.database_names.include?('ruby-mongo-will-be-deleted')
+  end
+
+  def test_pair
+    db = Mongo.new({:left => ['foo', 123]})
+    pair = db.instance_variable_get('@pair')
+    assert_equal 2, pair.length
+    assert_equal ['foo', 123], pair[0]
+    assert_equal ['localhost', Mongo::DEFAULT_PORT], pair[1]
+
+    db = Mongo.new({:right => 'bar'})
+    pair = db.instance_variable_get('@pair')
+    assert_equal 2, pair.length
+    assert_equal ['localhost', Mongo::DEFAULT_PORT], pair[0]
+    assert_equal ['bar', Mongo::DEFAULT_PORT], pair[1]
+
+    db = Mongo.new({:right => ['foo', 123], :left => 'bar'})
+    pair = db.instance_variable_get('@pair')
+    assert_equal 2, pair.length
+    assert_equal ['bar', Mongo::DEFAULT_PORT], pair[0]
+    assert_equal ['foo', 123], pair[1]
+  end
+
+end

data/tests/test_objectid.rb

@@ -0,0 +1,98 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'test/unit'
+
+class ObjectIDTest < Test::Unit::TestCase
+
+  include XGen::Mongo::Driver
+
+  def setup
+    @t = 42
+    @o = ObjectID.new(nil, @t)
+  end
+
+  def test_index_for_time
+    t = 99
+    assert_equal 0, @o.index_for_time(t)
+    assert_equal 1, @o.index_for_time(t)
+    assert_equal 2, @o.index_for_time(t)
+    t = 100
+    assert_equal 0, @o.index_for_time(t)
+  end
+
+  def test_time_bytes
+    a = @o.to_a
+    assert_equal @t, a[0]
+    3.times { |i| assert_equal 0, a[i+1] }
+
+    t = 43
+    o = ObjectID.new(nil, t)
+    a = o.to_a
+    assert_equal t, a[0]
+    3.times { |i| assert_equal 0, a[i+1] }
+    assert_equal 1, o.index_for_time(t) # 0 was used for o
+  end
+
+  def test_different
+    o2 = ObjectID.new(nil, @t)
+    assert @o.to_a != o2.to_a
+  end
+
+  def test_eql?
+    o2 = ObjectID.new(@o.to_a)
+    assert @o.eql?(o2)
+    assert @o == o2
+  end
+
+  def test_to_s
+    s = @o.to_s
+    assert_equal 24, s.length
+    s =~ /^([0-9a-f]+)$/
+    assert_equal 24, $1.length
+  end
+
+  def test_save_and_restore
+    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')
+    coll = db.collection('test')
+
+    coll.clear
+    coll << {'a' => 1, '_id' => @o}
+
+    row = coll.find().collect.first
+    assert_equal 1, row['a']
+    assert_equal @o, row['_id']
+  end
+
+  def test_from_string
+    hex_str = @o.to_s
+    o2 = ObjectID.from_string(hex_str)
+    assert_equal hex_str, o2.to_s
+    assert_equal @o, o2
+    assert_equal @o.to_s, o2.to_s
+  end
+
+  def test_legal
+    assert !ObjectID.legal?(nil)
+    assert !ObjectID.legal?("fred")
+    assert !ObjectID.legal?("0000")
+    assert !ObjectID.legal?('000102030405060708090A0')
+    assert ObjectID.legal?('000102030405060708090A0B')
+    assert ObjectID.legal?('abcdefABCDEF123456789012')
+    assert !ObjectID.legal?('abcdefABCDEF12345678901x')
+  end
+
+  def test_from_string_leading_zeroes
+    hex_str = '000000000000000000abcdef'
+    o = ObjectID.from_string(hex_str)
+    assert_equal hex_str, o.to_s
+  end
+
+  def test_byte_order
+    hex_str = '000102030405060708090A0B'
+    o = ObjectID.from_string(hex_str)
+    assert_equal [0x07, 0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00, 0x0b, 0x0a, 0x09, 0x08], o.to_a
+  end
+
+end