mongo 1.12.5 → 2.0.0

data/spec/mongo/grid/file/chunk_spec.rb

@@ -0,0 +1,226 @@
+require 'spec_helper'
+
+describe Mongo::Grid::File::Chunk do
+
+  let(:data) do
+    BSON::Binary.new('testing')
+  end
+
+  let(:file_id) do
+    BSON::ObjectId.new
+  end
+
+  let(:metadata) do
+    Mongo::Grid::File::Metadata.new(:files_id => file_id)
+  end
+
+  describe '#==' do
+
+    let(:chunk) do
+      described_class.new(:data => data, :files_id => file_id, :n => 5)
+    end
+
+    context 'when the other is not a chunk' do
+
+      it 'returns false' do
+        expect(chunk).to_not eq('test')
+      end
+    end
+
+    context 'when the other object is a chunk' do
+
+      context 'when the documents are equal' do
+
+        it 'returns true' do
+          expect(chunk).to eq(chunk)
+        end
+      end
+
+      context 'when the documents are not equal' do
+
+        let(:other) do
+          described_class.new(:data => data, :files_id => file_id, :n => 6)
+        end
+
+        it 'returns false' do
+          expect(chunk).to_not eq(other)
+        end
+      end
+    end
+  end
+
+  describe '.assemble' do
+
+    let(:data_size) do
+      Mongo::Grid::File::Chunk::DEFAULT_SIZE * 3
+    end
+
+    let(:raw_data) do
+      'testing'
+    end
+
+    let(:data) do
+      BSON::Binary.new(raw_data)
+    end
+
+    let(:assembled) do
+      described_class.assemble(chunks)
+    end
+
+    before do
+      (1..data_size).each{ |i| raw_data << '1' }
+    end
+
+    let(:chunks) do
+      described_class.split(raw_data, metadata)
+    end
+
+    it 'returns the chunks assembled into the raw data' do
+      expect(assembled).to eq(raw_data)
+    end
+  end
+
+  describe '#document' do
+
+    let(:chunk) do
+      described_class.new(:data => data, :files_id => file_id, :n => 5)
+    end
+
+    let(:document) do
+      chunk.document
+    end
+
+    it 'sets the data' do
+      expect(document[:data]).to eq(data)
+    end
+
+    it 'sets the files_id' do
+      expect(document[:files_id]).to eq(file_id)
+    end
+
+    it 'sets the position' do
+      expect(document[:n]).to eq(5)
+    end
+
+    it 'sets an object id' do
+      expect(document[:_id]).to be_a(BSON::ObjectId)
+    end
+
+    context 'when asking for the document multiple times' do
+
+      it 'returns the same document' do
+        expect(document[:_id]).to eq(chunk.document[:_id])
+      end
+    end
+  end
+
+  describe '#initialize' do
+
+    let(:chunk) do
+      described_class.new(:data => data, :files_id => file_id, :n => 5)
+    end
+
+    it 'sets the document' do
+      expect(chunk.data).to eq(data)
+    end
+
+    it 'sets a default id' do
+      expect(chunk.id).to be_a(BSON::ObjectId)
+    end
+  end
+
+  describe '#to_bson' do
+
+    let(:chunk) do
+      described_class.new(:data => data, :files_id => file_id, :n => 5)
+    end
+
+    let(:document) do
+      chunk.document
+    end
+
+    it 'returns the document as bson' do
+      expect(chunk.to_bson).to eq(document.to_bson)
+    end
+  end
+
+  describe '.split' do
+
+    context 'when the data is smaller than the default size' do
+
+      let(:raw_data) do
+        'testing'
+      end
+
+      let(:data) do
+        BSON::Binary.new(raw_data)
+      end
+
+      let(:chunks) do
+        described_class.split(raw_data, metadata)
+      end
+
+      let(:chunk) do
+        chunks.first
+      end
+
+      it 'returns a single chunk' do
+        expect(chunks.size).to eq(1)
+      end
+
+      it 'sets the correct chunk position' do
+        expect(chunk.n).to eq(0)
+      end
+
+      it 'sets the correct chunk data' do
+        expect(chunk.data).to eq(data)
+      end
+    end
+
+    context 'when the data is larger that the default size' do
+
+      let(:data_size) do
+        Mongo::Grid::File::Chunk::DEFAULT_SIZE * 3
+      end
+
+      let(:raw_data) do
+        'testing'
+      end
+
+      let(:data) do
+        BSON::Binary.new(raw_data)
+      end
+
+      let(:assembled) do
+        full_data = ''
+        chunks.each do |chunk|
+          full_data << chunk.data.data
+        end
+        full_data
+      end
+
+      before do
+        (1..data_size).each{ |i| raw_data << '1' }
+      end
+
+      let(:chunks) do
+        described_class.split(raw_data, metadata)
+      end
+
+      it 'returns the correct number of chunks' do
+        expect(chunks.size).to eq(4)
+      end
+
+      it 'sets the correct chunk positions' do
+        expect(chunks[0].n).to eq(0)
+        expect(chunks[1].n).to eq(1)
+        expect(chunks[2].n).to eq(2)
+        expect(chunks[3].n).to eq(3)
+      end
+
+      it 'does to miss any bytes' do
+        expect(assembled).to eq(raw_data)
+      end
+    end
+  end
+end

data/spec/mongo/grid/file/metadata_spec.rb

@@ -0,0 +1,92 @@
+require 'spec_helper'
+
+describe Mongo::Grid::File::Metadata do
+
+  describe '#==' do
+
+    let(:upload_date) do
+      Time.now.utc
+    end
+
+    let(:metadata) do
+      described_class.new(:filename => 'test.txt', :length => 7, :uploadDate => upload_date)
+    end
+
+    context 'when the other is not metadata' do
+
+      it 'returns false' do
+        expect(metadata).to_not eq('test')
+      end
+    end
+
+    context 'when the other object is metadata' do
+
+      context 'when the documents are equal' do
+
+        it 'returns true' do
+          expect(metadata).to eq(metadata)
+        end
+      end
+
+      context 'when the documents are not equal' do
+
+        let(:other) do
+          described_class.new(:filename => 'testing.txt')
+        end
+
+        it 'returns false' do
+          expect(metadata).to_not eq(other)
+        end
+      end
+    end
+  end
+
+  describe '#initialize' do
+
+    context 'when provided only a filename and length' do
+
+      let(:metadata) do
+        described_class.new(:filename => 'test.txt', :length => 7)
+      end
+
+      it 'sets the default id' do
+        expect(metadata.id).to be_a(BSON::ObjectId)
+      end
+
+      it 'sets the upload date' do
+        expect(metadata.upload_date).to be_a(Time)
+      end
+
+      it 'sets the chunk size' do
+        expect(metadata.chunk_size).to eq(Mongo::Grid::File::Chunk::DEFAULT_SIZE)
+      end
+
+      it 'sets the content type' do
+        expect(metadata.content_type).to eq(Mongo::Grid::File::Metadata::DEFAULT_CONTENT_TYPE)
+      end
+    end
+  end
+
+  describe '#inspect' do
+
+    let(:metadata) do
+      described_class.new(:filename => 'test.txt', :length => 7)
+    end
+
+    it 'includes the chunk size' do
+      expect(metadata.inspect).to include(metadata.chunk_size.to_s)
+    end
+
+    it 'includes the filename' do
+      expect(metadata.inspect).to include(metadata.filename)
+    end
+
+    it 'includes the md5' do
+      expect(metadata.inspect).to include(metadata.md5.to_s)
+    end
+
+    it 'includes the id' do
+      expect(metadata.inspect).to include(metadata.id.to_s)
+    end
+  end
+end

data/spec/mongo/grid/file_spec.rb

@@ -0,0 +1,172 @@
+require 'spec_helper'
+
+describe Mongo::Grid::File do
+
+  describe '#==' do
+
+    let(:file) do
+      described_class.new('test', :filename => 'test.txt')
+    end
+
+    context 'when the object is not a file' do
+
+      it 'returns false' do
+        expect(file).to_not eq('testing')
+      end
+    end
+
+    context 'when the object is a file' do
+
+      context 'when the objects are equal' do
+
+        it 'returns true' do
+          expect(file).to eq(file)
+        end
+      end
+
+      context 'when the objects are not equal' do
+
+        let(:other) do
+          described_class.new('tester', :filename => 'test.txt')
+        end
+
+        it 'returns false' do
+          expect(file).to_not eq(other)
+        end
+      end
+    end
+  end
+
+  describe '#initialize' do
+
+    let(:data_size) do
+      Mongo::Grid::File::Chunk::DEFAULT_SIZE * 3
+    end
+
+    let(:data) do
+      'testing'
+    end
+
+    before do
+      (1..data_size).each{ |i| data << '1' }
+    end
+
+    context 'when provided data and metadata' do
+
+      let(:file) do
+        described_class.new(data, :filename => 'test.txt')
+      end
+
+      it 'sets the data' do
+        expect(file.data).to eq(data)
+      end
+
+      it 'creates the chunks' do
+        expect(file.chunks.size).to eq(4)
+      end
+    end
+
+    context 'when data is a ruby file' do
+
+      let(:ruby_file) do
+        File.open(__FILE__)
+      end
+
+      let(:data) do
+        ruby_file.read
+      end
+
+      let(:file) do
+        described_class.new(data, :filename => File.basename(ruby_file.path))
+      end
+
+      it 'sets the data' do
+        expect(file.data).to eq(data)
+      end
+
+      it 'creates the chunks' do
+        expect(file.chunks.size).to eq(4)
+      end
+    end
+
+    context 'when using idiomatic ruby field names' do
+
+      let(:time) do
+        Time.now.utc
+      end
+
+      let(:file) do
+        described_class.new(
+          data,
+          :filename => 'test.txt',
+          :chunk_size => 100,
+          :upload_date => time,
+          :content_type => 'text/plain'
+        )
+      end
+
+      it 'normalizes the chunk size name' do
+        expect(file.chunk_size).to eq(100)
+      end
+
+      it 'normalizes the upload date name' do
+        expect(file.upload_date).to eq(time)
+      end
+
+      it 'normalizes the content type name' do
+        expect(file.content_type).to eq('text/plain')
+      end
+    end
+
+    context 'when provided chunks and metadata' do
+
+      let(:file_id) do
+        BSON::ObjectId.new
+      end
+
+      let(:metadata) do
+        BSON::Document.new(
+          :_id => file_id,
+          :uploadDate => Time.now.utc,
+          :filename => 'test.txt',
+          :chunkSize => Mongo::Grid::File::Chunk::DEFAULT_SIZE,
+          :length => data.length,
+          :contentType => Mongo::Grid::File::Metadata::DEFAULT_CONTENT_TYPE
+        )
+      end
+
+      let(:chunks) do
+        Mongo::Grid::File::Chunk.split(
+          data, Mongo::Grid::File::Metadata.new(metadata)
+        ).map{ |chunk| chunk.document }
+      end
+
+      let(:file) do
+        described_class.new(chunks, metadata)
+      end
+
+      it 'sets the chunks' do
+        expect(file.chunks.size).to eq(4)
+      end
+
+      it 'assembles to data' do
+        expect(file.data).to eq(data)
+      end
+
+      it 'sets the metadata' do
+        expect(file.metadata.id).to eq(metadata[:_id])
+      end
+    end
+  end
+
+  describe '#inspect' do
+
+    let(:file) do
+      described_class.new('Hi', :filename => 'test.txt')
+    end
+
+    it 'includes the filename' do
+      expect(file.inspect).to include('test.txt')
+    end
+  end
+end

data/spec/mongo/grid/fs_spec.rb

@@ -0,0 +1,129 @@
+require 'spec_helper'
+
+describe Mongo::Grid::FS do
+
+  describe '#initialize' do
+
+    let(:fs) do
+      described_class.new(authorized_client.database)
+    end
+
+    let(:index) do
+      fs.chunks_collection.indexes.get(:files_id => 1, :n => 1)
+    end
+
+    it 'sets the files collection' do
+      expect(fs.files_collection.name).to eq('fs.files')
+    end
+
+    it 'sets the chunks collection' do
+      expect(fs.chunks_collection.name).to eq('fs.chunks')
+    end
+
+    it 'creates the index on the chunks collection' do
+      expect(index[:name]).to eq('files_id_1_n_1')
+    end
+  end
+
+  describe '#find_one' do
+
+    let(:fs) do
+      described_class.new(authorized_client.database)
+    end
+
+    let(:file) do
+      Mongo::Grid::File.new('hello world!', :filename => 'test.txt')
+    end
+
+    before do
+      fs.insert_one(file)
+    end
+
+    after do
+      fs.files_collection.find.delete_many
+      fs.chunks_collection.find.delete_many
+    end
+
+    let(:from_db) do
+      fs.find_one(:filename => 'test.txt')
+    end
+
+    it 'returns the assembled file from the db' do
+      expect(from_db.filename).to eq(file.metadata.filename)
+    end
+  end
+
+  describe '#insert_one' do
+
+    let(:fs) do
+      described_class.new(authorized_client.database)
+    end
+
+    let(:file) do
+      Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
+    end
+
+    context 'when inserting the file once' do
+
+      before do
+        fs.insert_one(file)
+      end
+
+      after do
+        fs.files_collection.find.delete_many
+        fs.chunks_collection.find.delete_many
+      end
+
+      let(:from_db) do
+        fs.find_one(:filename => 'test.txt')
+      end
+
+      it 'inserts the file into the database' do
+        expect(from_db.filename).to eq(file.metadata.filename)
+      end
+
+      it 'includes the chunks and data with the file' do
+        expect(from_db.data).to eq('Hello!')
+      end
+    end
+
+    context 'when inserting the file more than once' do
+
+      after do
+        fs.files_collection.find.delete_many
+        fs.chunks_collection.find.delete_many
+      end
+
+      it 'raises an error' do
+        expect {
+          fs.insert_one(file)
+          fs.insert_one(file)
+        }.to raise_error(Mongo::Error::OperationFailure)
+      end
+    end
+  end
+
+  describe '#delete_one' do
+
+    let(:fs) do
+      described_class.new(authorized_client.database)
+    end
+
+    let(:file) do
+      Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
+    end
+
+    before do
+      fs.insert_one(file)
+      fs.delete_one(file)
+    end
+
+    let(:from_db) do
+      fs.find_one(:filename => 'test.txt')
+    end
+
+    it 'removes the file from the db' do
+      expect(from_db).to be_nil
+    end
+  end
+end