338 lines
8.9 KiB
Ruby
338 lines
8.9 KiB
Ruby
require 'test/test_helper'
|
|
require 'mongo/gridfs'
|
|
|
|
class GridStoreTest < Test::Unit::TestCase
|
|
|
|
include Mongo
|
|
include GridFS
|
|
|
|
@@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')
|
|
|
|
def setup
|
|
@@chunks.remove
|
|
@@files.remove
|
|
GridStore.open(@@db, 'foobar', 'w') { |f| f.write("hello, world!") }
|
|
end
|
|
|
|
def teardown
|
|
@@chunks.remove
|
|
@@files.remove
|
|
@@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
|
|
|
|
def test_read_with_and_without_length
|
|
GridStore.open(@@db, 'read-types', 'w') do |f|
|
|
f.write('hello, there')
|
|
end
|
|
|
|
GridStore.open(@@db, 'read-types', 'r') do |f|
|
|
assert_equal 'hello, ', f.read(7)
|
|
assert_equal 'there', f.read
|
|
end
|
|
end
|
|
|
|
def test_access_length
|
|
assert_equal 13, GridStore.new(@@db, 'foobar').length
|
|
end
|
|
|
|
# Also tests seek
|
|
def test_read_with_offset
|
|
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.remove
|
|
@@files.remove
|
|
|
|
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
|
|
end
|
|
|
|
def test_binary
|
|
file = File.open(File.join(File.dirname(__FILE__), 'data', 'data.tar.gz'), 'r')
|
|
GridStore.open(@@db, 'zip', 'w') do |f|
|
|
f.write(file.read)
|
|
end
|
|
|
|
file.rewind
|
|
data = file.read
|
|
if data.respond_to?(:force_encoding)
|
|
data.force_encoding(:binary)
|
|
end
|
|
GridStore.open(@@db, 'zip', 'r') do |f|
|
|
assert_equal data.length, f.read.length
|
|
end
|
|
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_unlink_alternate_root_collection
|
|
GridStore.default_root_collection = 'gridfs'
|
|
GridStore.open(@@db, 'foobar', 'w') do |f|
|
|
f.puts "Hello"
|
|
end
|
|
assert GridStore.exist?(@@db, 'foobar')
|
|
|
|
GridStore.default_root_collection = 'fs'
|
|
GridStore.unlink(@@db, 'foobar')
|
|
assert !GridStore.exist?(@@db, 'foobar')
|
|
|
|
GridStore.default_root_collection = 'gridfs'
|
|
GridStore.unlink(@@db, 'foobar')
|
|
assert !GridStore.exist?(@@db, 'foobar')
|
|
end
|
|
|
|
def test_mv
|
|
assert_equal 1, @@files.count
|
|
assert_equal 1, @@chunks.count
|
|
GridStore.mv(@@db, 'foobar', 'bazqux')
|
|
assert_equal 1, @@files.count
|
|
assert_equal 1, @@chunks.count
|
|
assert !GridStore.exist?(@@db, 'foobar')
|
|
assert GridStore.exist?(@@db, 'bazqux')
|
|
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.remove
|
|
@@files.remove
|
|
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
|