154 lines
2.9 KiB
Ruby
154 lines
2.9 KiB
Ruby
require_relative 'helper'
|
|
|
|
Testing Mongoid::GridFs do
|
|
##
|
|
#
|
|
GridFS =
|
|
GridFs =
|
|
Mongoid::GridFS
|
|
|
|
prepare do
|
|
GridFS::File.destroy_all
|
|
GridFS::Chunk.destroy_all
|
|
end
|
|
|
|
##
|
|
#
|
|
context '#put' do
|
|
|
|
test 'default' do
|
|
filename = __FILE__
|
|
basename = File.basename(filename)
|
|
|
|
g = assert{ GridFS.put(filename) }
|
|
|
|
assert{ g.filename =~ %r| #{ object_id_re } / #{ basename } \Z|imox }
|
|
assert{ g.content_type == "application/x-ruby" }
|
|
assert{ g.data == IO.read(filename) }
|
|
end
|
|
|
|
test 'with a :filename' do
|
|
filename = 'path/info/a.rb'
|
|
|
|
g = assert{ GridFS.put(__FILE__, :filename => filename) }
|
|
|
|
assert{ g.filename == filename }
|
|
end
|
|
|
|
end
|
|
|
|
|
|
##
|
|
#
|
|
context '#get' do
|
|
|
|
test 'default' do
|
|
id = assert{ GridFS::File.last.id }
|
|
g = assert{ GridFs.get(id) }
|
|
end
|
|
|
|
end
|
|
|
|
##
|
|
#
|
|
context '#delete' do
|
|
|
|
test 'default' do
|
|
id = assert{ GridFS::File.last.id }
|
|
g = assert{ GridFs.get(id) }
|
|
assert{ GridFs.delete(id) }
|
|
assert_raises( Mongoid::Errors::DocumentNotFound){ GridFs.get(id) }
|
|
end
|
|
|
|
end
|
|
|
|
##
|
|
#
|
|
context '[] and []=' do
|
|
|
|
test 'default' do
|
|
path = 'a.rb'
|
|
data = IO.read(__FILE__)
|
|
|
|
sio = SIO.new(path, data)
|
|
|
|
g = assert{ GridFs[path] = sio and GridFs[path] }
|
|
|
|
assert{ g.data == data }
|
|
assert{ g.content_type == "application/x-ruby" }
|
|
|
|
before = GridFs::File.count
|
|
|
|
assert{ GridFs[path] = SIO.new(path, 'foobar') }
|
|
assert{ GridFs[path].data == 'foobar' }
|
|
|
|
after = GridFs::File.count
|
|
|
|
created = after - before
|
|
|
|
assert{ created.zero? }
|
|
end
|
|
|
|
##
|
|
#
|
|
context 'data uris' do
|
|
|
|
test 'default' do
|
|
id = assert{ GridFS::File.last.id }
|
|
g = assert{ GridFs.get(id) }
|
|
|
|
content_type = g.content_type
|
|
base64 = [g.to_s].pack('m').chomp
|
|
|
|
data_uri = "data:#{ content_type };base64,".concat(base64)
|
|
|
|
assert{ g.data_uri == data_uri }
|
|
end
|
|
|
|
end
|
|
|
|
##
|
|
#
|
|
context 'namespaces' do
|
|
test 'default' do
|
|
assert{ GridFs.namespace.prefix == 'fs' }
|
|
assert{ GridFs.file_model.collection_name == 'fs.files' }
|
|
assert{ GridFs.chunk_model.collection_name == 'fs.chunks' }
|
|
end
|
|
|
|
test 'new' do
|
|
ns = GridFs.namespace_for(:ns)
|
|
|
|
assert{ ns.prefix == 'ns' }
|
|
|
|
assert{ ns.file_model < Mongoid::Document }
|
|
assert{ ns.file_model.collection_name == 'ns.files' }
|
|
|
|
assert{ ns.chunk_model < Mongoid::Document }
|
|
assert{ ns.chunk_model.collection_name == 'ns.chunks' }
|
|
|
|
assert{ ns.file_model.destroy_all }
|
|
|
|
count = GridFs::File.count
|
|
|
|
assert{ ns.file_model.count == 0}
|
|
assert{ ns.put __FILE__ }
|
|
assert{ ns.file_model.count == 1}
|
|
|
|
assert{ count == GridFs::File.count }
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
##
|
|
#
|
|
|
|
protected
|
|
def object_id_re
|
|
object_id = defined?(Moped) ? Moped::BSON::ObjectId.new : BSON::ObjectId.new
|
|
|
|
%r| \w{#{ object_id.to_s.size }} |iomx
|
|
end
|
|
end
|