85 lines
2.2 KiB
Ruby
85 lines
2.2 KiB
Ruby
require 'test/test_helper'
|
|
|
|
class GridTest < Test::Unit::TestCase
|
|
|
|
def setup
|
|
@db ||= Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
|
|
ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT).db('ruby-mongo-test')
|
|
@files = @db.collection('test-fs.files')
|
|
@chunks = @db.collection('test-fs.chunks')
|
|
end
|
|
|
|
def teardown
|
|
@files.remove
|
|
@chunks.remove
|
|
end
|
|
|
|
context "A basic grid-stored file" do
|
|
setup do
|
|
@data = "GRIDDATA" * 50000
|
|
@grid = Grid.new(@db, 'test-fs')
|
|
@id = @grid.put(@data, 'sample', :metadata => {'app' => 'photos'})
|
|
end
|
|
|
|
should "retrieve the stored data" do
|
|
data = @grid.get(@id).data
|
|
assert_equal @data, data
|
|
end
|
|
|
|
should "store the filename" do
|
|
file = @grid.get(@id)
|
|
assert_equal 'sample', file.filename
|
|
end
|
|
|
|
should "store any relevant metadata" do
|
|
file = @grid.get(@id)
|
|
assert_equal 'photos', file.metadata['app']
|
|
end
|
|
|
|
should "delete the file and any chunks" do
|
|
@grid.delete(@id)
|
|
assert_raise GridError do
|
|
@grid.get(@id)
|
|
end
|
|
end
|
|
end
|
|
|
|
def read_and_write_stream(filename, read_length, opts={})
|
|
io = File.open(File.join(File.dirname(__FILE__), 'data', filename), 'r')
|
|
id = @grid.put(io, filename + read_length.to_s, opts)
|
|
file = @grid.get(id)
|
|
io.rewind
|
|
data = io.read
|
|
if data.respond_to?(:force_encoding)
|
|
data.force_encoding(:binary)
|
|
end
|
|
read_data = ""
|
|
while(chunk = file.read(read_length))
|
|
read_data << chunk
|
|
end
|
|
assert_equal data.length, read_data.length
|
|
end
|
|
|
|
context "Streaming: " do
|
|
setup do
|
|
@grid = Grid.new(@db, 'test-fs')
|
|
end
|
|
|
|
should "put and get a small io object with a small chunk size" do
|
|
read_and_write_stream('small_data.txt', 1, :chunk_size => 2)
|
|
end
|
|
|
|
should "put and get a small io object" do
|
|
read_and_write_stream('small_data.txt', 1)
|
|
end
|
|
|
|
should "put and get a large io object when reading smaller than the chunk size" do
|
|
read_and_write_stream('sample_file.pdf', 256 * 1024)
|
|
end
|
|
|
|
should "put and get a large io object when reading larger than the chunk size" do
|
|
read_and_write_stream('sample_file.pdf', 300 * 1024)
|
|
end
|
|
end
|
|
end
|