mongo-ruby-driver/tests/test_grid_store.rb

262 lines
6.8 KiB
Ruby
Raw Normal View History

2009-01-29 16:23:50 +00:00
$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
def setup
@host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
@port = ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT
@db = Mongo.new(@host, @port).db('ruby-mongo-utils-test')
2009-01-30 21:44:29 +00:00
@files = @db.collection('gridfs.files')
@chunks = @db.collection('gridfs.chunks')
2009-01-29 16:23:50 +00:00
@chunks.clear
@files.clear
GridStore.open(@db, 'foobar', 'w') { |f| f.write("hello, world!") }
end
def teardown
if @db && @db.connected?
@chunks.clear
@files.clear
@db.close
end
end
2009-01-30 21:44:29 +00:00
def test_exist
assert GridStore.exist?(@db, 'foobar')
assert !GridStore.exist?(@db, 'does_not_exist')
assert !GridStore.exist?(@db, 'foobar', 'another_root')
end
2009-01-29 16:23:50 +00:00
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
2009-01-30 21:44:29 +00:00
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
2009-01-29 16:23:50 +00:00
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
2009-01-30 21:44:29 +00:00
# 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
2009-01-29 16:23:50 +00:00
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
2009-01-29 16:23:50 +00:00
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
2009-01-30 21:44:29 +00:00
GridStore.open(@db, 'empty', 'w') {} # re-write with zero bytes
2009-01-29 16:23:50 +00:00
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_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
2009-01-30 21:44:29 +00:00
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
2009-01-29 16:23:50 +00:00
end