mongo 2.21.0 → 2.21.2

data/spec/mongo/grid/fs_bucket_spec.rb

@@ -1,1269 +0,0 @@
-# frozen_string_literal: true
-# rubocop:todo all
-
-require 'spec_helper'
-
-describe Mongo::Grid::FSBucket do
-
-  let(:fs) do
-    described_class.new(client.database, options)
-  end
-
-  # A different instance so that fs creates indexes correctly
-  let(:support_fs) do
-    described_class.new(client.database, options)
-  end
-
-  let(:client) do
-    authorized_client
-  end
-
-  let(:options) do
-    { }
-  end
-
-  let(:filename) do
-    'specs.rb'
-  end
-
-  let(:file) do
-    File.open(__FILE__)
-  end
-
-  before do
-    support_fs.files_collection.drop rescue nil
-    support_fs.chunks_collection.drop rescue nil
-  end
-
-  describe '#initialize' do
-
-    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
-
-    context 'when options are provided' do
-
-      let(:fs) do
-        described_class.new(authorized_client.database, options)
-      end
-
-      context 'when a write concern is set' do
-
-        context 'when the option :write is provided' do
-
-          let(:options) do
-            { write: { w: 2 } }
-          end
-
-          it 'sets the write concern' do
-            expect(fs.send(:write_concern).options).to eq(Mongo::WriteConcern.get(w: 2).options)
-          end
-        end
-      end
-
-      context 'when a read preference is set' do
-
-        context 'when given as a hash with symbol keys' do
-          let(:options) do
-            { read: { mode: :secondary } }
-          end
-
-          it 'returns the read preference as a BSON::Document' do
-            expect(fs.send(:read_preference)).to be_a(BSON::Document)
-            expect(fs.send(:read_preference)).to eq('mode' => :secondary)
-          end
-        end
-
-        context 'when given as a BSON::Document' do
-          let(:options) do
-            BSON::Document.new(read: { mode: :secondary })
-          end
-
-          it 'returns the read preference as set' do
-            expect(fs.send(:read_preference)).to eq(options[:read])
-          end
-        end
-      end
-
-      context 'when a read preference is not set' do
-
-        let(:database) do
-          authorized_client.with(read: { mode: :secondary }).database
-        end
-
-        let(:fs) do
-          described_class.new(database, options)
-        end
-
-        it 'uses the read preference of the database' do
-          expect(fs.read_preference).to be(database.read_preference)
-        end
-      end
-
-      context 'when a write stream is opened' do
-
-        let(:stream) do
-          fs.open_upload_stream('test.txt')
-        end
-
-        let(:fs) do
-          described_class.new(authorized_client.database, options)
-        end
-
-        context 'when a write option is specified' do
-
-          let(:options) do
-            { write: { w: 2 } }
-          end
-
-          it 'passes the write concern to the write stream' do
-            expect(stream.write_concern.options).to eq(Mongo::WriteConcern.get(options[:write]).options)
-          end
-        end
-
-        context 'when disable_md5 is not specified' do
-
-          it 'does not set the option on the write stream' do
-            expect(stream.options[:disable_md5]).to be_nil
-          end
-        end
-
-        context 'when disable_md5 is specified' do
-
-          context 'when disable_md5 is true' do
-
-            let(:options) do
-              { disable_md5: true }
-            end
-
-            it 'passes the option to the write stream' do
-              expect(stream.options[:disable_md5]).to be(true)
-            end
-          end
-
-          context 'when disable_md5 is false' do
-
-            let(:options) do
-              { disable_md5: false }
-            end
-
-            it 'passes the option to the write stream' do
-              expect(stream.options[:disable_md5]).to be(false)
-            end
-          end
-        end
-      end
-    end
-  end
-
-  describe '#find' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database)
-    end
-
-    context 'when there is no selector provided' do
-
-      let(:files) do
-        [
-            Mongo::Grid::File.new('hello world!', :filename => 'test.txt'),
-            Mongo::Grid::File.new('goodbye world!', :filename => 'test1.txt')
-        ]
-      end
-
-      before do
-        files.each do |file|
-          fs.insert_one(file)
-        end
-      end
-
-      it 'returns a collection view' do
-        expect(fs.find).to be_a(Mongo::Collection::View)
-      end
-
-      it 'iterates over the documents in the result' do
-        fs.find.each do |document|
-          expect(document).to_not be_nil
-        end
-      end
-    end
-
-    context 'when provided a filter' do
-
-      let(:view) do
-        fs.find(filename: 'test.txt')
-      end
-
-      it 'returns a collection view for the filter' do
-        expect(view.filter).to eq('filename' => 'test.txt')
-      end
-    end
-
-    context 'when options are provided' do
-
-      let(:view) do
-        fs.find({filename: 'test.txt'}, options)
-      end
-
-      context 'when provided allow_disk_use' do
-        context 'when allow_disk_use is true' do
-          let(:options) { { allow_disk_use: true } }
-
-          it 'sets allow_disk_use on the view' do
-            expect(view.options[:allow_disk_use]).to be true
-          end
-        end
-
-        context 'when allow_disk_use is false' do
-          let(:options) { { allow_disk_use: false } }
-
-          it 'sets allow_disk_use on the view' do
-            expect(view.options[:allow_disk_use]).to be false
-          end
-        end
-      end
-
-      context 'when provided batch_size' do
-
-        let(:options) do
-          { batch_size: 5 }
-        end
-
-        it 'sets the batch_size on the view' do
-          expect(view.batch_size).to eq(options[:batch_size])
-        end
-      end
-
-      context 'when provided limit' do
-
-        let(:options) do
-          { limit: 5 }
-        end
-
-        it 'sets the limit on the view' do
-          expect(view.limit).to eq(options[:limit])
-        end
-      end
-
-      context 'when provided no_cursor_timeout' do
-
-        let(:options) do
-          { no_cursor_timeout: true }
-        end
-
-        it 'sets the no_cursor_timeout on the view' do
-          expect(view.options[:no_cursor_timeout]).to eq(options[:no_cursor_timeout])
-        end
-      end
-
-      context 'when provided skip' do
-
-        let(:options) do
-          { skip: 5 }
-        end
-
-        it 'sets the skip on the view' do
-          expect(view.skip).to eq(options[:skip])
-        end
-      end
-
-      context 'when provided sort' do
-
-        let(:options) do
-          { sort:  { 'x' => Mongo::Index::ASCENDING } }
-        end
-
-        it 'sets the sort on the view' do
-          expect(view.sort).to eq(options[:sort])
-        end
-      end
-    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
-
-    let(:from_db) do
-      fs.find_one(:filename => 'test.txt')
-    end
-
-    let(:from_db_upload_date) do
-      from_db.info.upload_date.strftime("%Y-%m-%d %H:%M:%S")
-    end
-
-    let(:file_info_upload_date) do
-      file.info.upload_date.strftime("%Y-%m-%d %H:%M:%S")
-    end
-
-    it 'returns the assembled file from the db' do
-      expect(from_db.filename).to eq(file.info.filename)
-    end
-
-    it 'maps the file info correctly' do
-      expect(from_db.info.length).to eq(file.info.length)
-      expect(from_db_upload_date).to eq(file_info_upload_date)
-    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
-
-    let(:support_file) do
-      Mongo::Grid::File.new('Hello!', :filename => 'support_test.txt')
-    end
-
-    context 'when inserting the file once' do
-
-      let!(:result) do
-        fs.insert_one(file)
-      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.info.filename)
-      end
-
-      it 'includes the chunks and data with the file' do
-        expect(from_db.data).to eq('Hello!')
-      end
-
-      it 'returns the file id' do
-        expect(result).to eq(file.id)
-      end
-    end
-
-    context 'when the files collection is empty' do
-      before do
-        fs.database[fs.files_collection.name].indexes
-      end
-
-      let(:operation) do
-        expect(fs.files_collection).to receive(:indexes).and_call_original
-        expect(fs.chunks_collection).to receive(:indexes).and_call_original
-        fs.insert_one(file)
-      end
-
-      let(:chunks_index) do
-        fs.database[fs.chunks_collection.name].indexes.get(:files_id => 1, :n => 1)
-      end
-
-      let(:files_index) do
-        fs.database[fs.files_collection.name].indexes.get(:filename => 1, :uploadDate => 1)
-      end
-
-      it 'tries to create indexes' do
-        expect(fs).to receive(:create_index_if_missing!).twice.and_call_original
-        operation
-      end
-
-      it 'creates an index on the files collection' do
-        operation
-        expect(files_index[:name]).to eq('filename_1_uploadDate_1')
-      end
-
-      it 'creates an index on the chunks collection' do
-        operation
-        expect(chunks_index[:name]).to eq('files_id_1_n_1')
-      end
-
-      context 'when a write operation is called more than once' do
-
-        let(:file2) do
-          Mongo::Grid::File.new('Goodbye!', :filename => 'test2.txt')
-        end
-
-        it 'only creates the indexes the first time' do
-          RSpec::Mocks.with_temporary_scope do
-            expect(fs).to receive(:create_index_if_missing!).twice.and_call_original
-            operation
-          end
-          RSpec::Mocks.with_temporary_scope do
-            expect(fs).not_to receive(:create_index_if_missing!)
-            expect(fs.insert_one(file2)).to be_a(BSON::ObjectId)
-          end
-        end
-      end
-    end
-
-    context 'when the index creation encounters an error' do
-
-      before do
-        fs.chunks_collection.indexes.create_one(Mongo::Grid::FSBucket::CHUNKS_INDEX, :unique => false)
-      end
-
-      it 'should not raise an error to the user' do
-        expect {
-          fs.insert_one(file)
-        }.not_to raise_error
-      end
-    end
-
-    context 'when the files collection is not empty' do
-
-      before do
-        support_fs.insert_one(support_file)
-        fs.insert_one(file)
-      end
-
-      let(:files_index) do
-        fs.database[fs.files_collection.name].indexes.get(:filename => 1, :uploadDate => 1)
-      end
-
-      it 'assumes indexes already exist' do
-        expect(files_index[:name]).to eq('filename_1_uploadDate_1')
-      end
-    end
-
-    context 'when inserting the file more than once' do
-
-      it 'raises an error' do
-        expect {
-          fs.insert_one(file)
-          fs.insert_one(file)
-        }.to raise_error(Mongo::Error::BulkWriteError)
-      end
-    end
-
-    context 'when the file exceeds the max bson size' do
-
-      let(:fs) do
-        described_class.new(authorized_client.database)
-      end
-
-      let(:file) do
-        str = 'y' * 16777216
-        Mongo::Grid::File.new(str, :filename => 'large-file.txt')
-      end
-
-      before do
-        fs.insert_one(file)
-      end
-
-      it 'successfully inserts the file' do
-        expect(
-          fs.find_one(:filename => 'large-file.txt').chunks
-        ).to eq(file.chunks)
-      end
-    end
-  end
-
-  describe '#delete_one' do
-
-    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
-
-  describe '#delete' do
-
-    let(:file_id) do
-      fs.upload_from_stream(filename, file)
-    end
-
-    before do
-      fs.delete(file_id)
-    end
-
-    let(:from_db) do
-      fs.find_one(:filename => filename)
-    end
-
-    it 'removes the file from the db' do
-      expect(from_db).to be_nil
-    end
-
-    context 'when a custom file id is used' do
-
-      let(:custom_file_id) do
-        fs.upload_from_stream(filename, file, file_id: 'Custom ID')
-      end
-
-      before do
-        fs.delete(custom_file_id)
-      end
-
-      let(:from_db) do
-        fs.find_one(:filename => filename)
-      end
-
-      it 'removes the file from the db' do
-        expect(from_db).to be_nil
-      end
-    end
-  end
-
-  context 'when a read stream is opened' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database, options)
-    end
-
-    let(:io) do
-      StringIO.new
-    end
-
-    describe '#open_download_stream' do
-
-      let!(:file_id) do
-        fs.open_upload_stream(filename) do |stream|
-          stream.write(file)
-        end.file_id
-      end
-
-      context 'when a block is provided' do
-
-        let!(:stream) do
-          fs.open_download_stream(file_id) do |stream|
-            io.write(stream.read)
-          end
-        end
-
-        it 'returns a Stream::Read object' do
-          expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Read)
-        end
-
-        it 'closes the stream after the block completes' do
-          expect(stream.closed?).to be(true)
-        end
-
-        it 'yields the stream to the block' do
-          expect(io.size).to eq(file.size)
-        end
-      end
-
-      context 'when a block is not provided' do
-
-        let!(:stream) do
-          fs.open_download_stream(file_id)
-        end
-
-        it 'returns a Stream::Read object' do
-          expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Read)
-        end
-
-        it 'does not close the stream' do
-          expect(stream.closed?).to be(false)
-        end
-
-        it 'does not yield the stream to the block' do
-          expect(io.size).to eq(0)
-        end
-      end
-
-      context 'when a custom file id is provided' do
-
-        let(:file) do
-          File.open(__FILE__)
-        end
-
-        let!(:file_id) do
-          fs.open_upload_stream(filename, file_id: 'Custom ID') do |stream|
-            stream.write(file)
-          end.file_id
-        end
-
-        context 'when a block is provided' do
-
-          let!(:stream) do
-            fs.open_download_stream(file_id) do |stream|
-              io.write(stream.read)
-            end
-          end
-
-          it 'yields the stream to the block' do
-            expect(io.size).to eq(file.size)
-          end
-        end
-
-        context 'when a block is not provided' do
-
-          let!(:stream) do
-            fs.open_download_stream(file_id)
-          end
-
-          it 'returns a Stream::Read object' do
-            expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Read)
-          end
-
-          it 'does not close the stream' do
-            expect(stream.closed?).to be(false)
-          end
-
-          it 'does not yield the stream to the block' do
-            expect(io.size).to eq(0)
-          end
-        end
-      end
-    end
-
-    describe '#download_to_stream' do
-
-      context 'sessions' do
-
-        let(:options) do
-          { session: session }
-        end
-
-        let(:file_id) do
-          fs.open_upload_stream(filename) do |stream|
-            stream.write(file)
-          end.file_id
-        end
-
-        let(:operation) do
-          fs.download_to_stream(file_id, io)
-        end
-
-        let(:client) do
-          authorized_client
-        end
-
-        it_behaves_like 'an operation using a session'
-      end
-
-      context 'when the file is found' do
-
-        let!(:file_id) do
-          fs.open_upload_stream(filename) do |stream|
-            stream.write(file)
-          end.file_id
-        end
-
-        before do
-          fs.download_to_stream(file_id, io)
-        end
-
-        it 'writes to the provided stream' do
-          expect(io.size).to eq(file.size)
-        end
-
-        it 'does not close the stream' do
-          expect(io.closed?).to be(false)
-        end
-
-        context 'when the file has length 0' do
-
-          let(:file) do
-            StringIO.new('')
-          end
-
-          let(:from_db) do
-            fs.open_upload_stream(filename) { |s| s.write(file) }
-            fs.find_one(:filename => filename)
-          end
-
-          it 'can read the file back' do
-            expect(from_db.data.size).to eq(file.size)
-          end
-        end
-      end
-
-      context 'when there is no files collection document found' do
-
-        it 'raises an exception' do
-          expect{
-            fs.download_to_stream(BSON::ObjectId.new, io)
-          }.to raise_exception(Mongo::Error::FileNotFound)
-        end
-      end
-
-      context 'when a file has an id that is not an ObjectId' do
-
-        before do
-          fs.insert_one(file)
-          fs.download_to_stream(file_id, io)
-        end
-
-        let(:file_id) do
-          'non-object-id'
-        end
-
-        let(:file) do
-          Mongo::Grid::File.new(File.open(__FILE__).read,
-                                :filename => filename,
-                                :_id => file_id)
-        end
-
-        it 'reads the file successfully' do
-          expect(io.size).to eq(file.data.size)
-        end
-      end
-    end
-
-    context 'when a read preference is specified' do
-
-      let(:fs) do
-        described_class.new(authorized_client.database, options)
-      end
-
-      let(:options) do
-        { read: { mode: :secondary } }
-      end
-
-      let(:stream) do
-        fs.open_download_stream(BSON::ObjectId)
-      end
-
-      it 'sets the read preference on the Stream::Read object' do
-        expect(stream.read_preference).to be_a(BSON::Document)
-        expect(stream.read_preference).to eq(BSON::Document.new(options[:read]))
-      end
-    end
-
-    describe '#download_to_stream_by_name' do
-
-
-      let(:files) do
-        [
-            StringIO.new('hello 1'),
-            StringIO.new('hello 2'),
-            StringIO.new('hello 3'),
-            StringIO.new('hello 4')
-        ]
-      end
-
-      context ' when using a session' do
-
-        let(:options) do
-          { session: session }
-        end
-
-        let(:operation) do
-          fs.download_to_stream_by_name('test.txt', io)
-        end
-
-        let(:client) do
-          authorized_client
-        end
-
-        before do
-          files.each do |file|
-            authorized_client.database.fs.upload_from_stream('test.txt', file)
-          end
-        end
-
-        let(:io) do
-          StringIO.new
-        end
-
-        it_behaves_like 'an operation using a session'
-      end
-
-      context 'when not using a session' do
-
-        before do
-          files.each do |file|
-            fs.upload_from_stream('test.txt', file)
-          end
-        end
-
-        let(:io) do
-          StringIO.new
-        end
-
-        context 'when revision is not specified' do
-
-          let!(:result) do
-            fs.download_to_stream_by_name('test.txt', io)
-          end
-
-          it 'returns the most recent version' do
-            expect(io.string).to eq('hello 4')
-          end
-        end
-
-        context 'when revision is 0' do
-
-          let!(:result) do
-            fs.download_to_stream_by_name('test.txt', io, revision: 0)
-          end
-
-          it 'returns the original stored file' do
-            expect(io.string).to eq('hello 1')
-          end
-        end
-
-        context 'when revision is negative' do
-
-          let!(:result) do
-            fs.download_to_stream_by_name('test.txt', io, revision: -2)
-          end
-
-          it 'returns that number of versions from the most recent' do
-            expect(io.string).to eq('hello 3')
-          end
-        end
-
-        context 'when revision is positive' do
-
-          let!(:result) do
-            fs.download_to_stream_by_name('test.txt', io, revision: 1)
-          end
-
-          it 'returns that number revision' do
-            expect(io.string).to eq('hello 2')
-          end
-        end
-
-        context 'when the file revision is not found' do
-
-          it 'raises a FileNotFound error' do
-            expect {
-              fs.download_to_stream_by_name('test.txt', io, revision: 100)
-            }.to raise_exception(Mongo::Error::InvalidFileRevision)
-          end
-        end
-
-        context 'when the file is not found' do
-
-          it 'raises a FileNotFound error' do
-            expect {
-              fs.download_to_stream_by_name('non-existent.txt', io)
-            }.to raise_exception(Mongo::Error::FileNotFound)
-          end
-        end
-      end
-    end
-
-    describe '#open_download_stream_by_name' do
-
-      let(:files) do
-        [
-            StringIO.new('hello 1'),
-            StringIO.new('hello 2'),
-            StringIO.new('hello 3'),
-            StringIO.new('hello 4')
-        ]
-      end
-
-      let(:io) do
-        StringIO.new
-      end
-
-      context ' when using a session' do
-
-        let(:options) do
-          { session: session }
-        end
-
-        let(:operation) do
-          fs.download_to_stream_by_name('test.txt', io)
-        end
-
-        let(:client) do
-          authorized_client
-        end
-
-        before do
-          files.each do |file|
-            authorized_client.database.fs.upload_from_stream('test.txt', file)
-          end
-        end
-
-        let(:io) do
-          StringIO.new
-        end
-
-        it_behaves_like 'an operation using a session'
-      end
-
-      context 'when not using a session' do
-
-        before do
-          files.each do |file|
-            fs.upload_from_stream('test.txt', file)
-          end
-        end
-
-        context 'when a block is provided' do
-
-          let(:stream) do
-            fs.open_download_stream_by_name('test.txt') do |stream|
-              io.write(stream.read)
-            end
-          end
-
-          it 'returns a Stream::Read object' do
-            expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Read)
-          end
-
-          it 'closes the stream after the block completes' do
-            expect(stream.closed?).to be(true)
-          end
-
-          it 'yields the stream to the block' do
-            stream
-            expect(io.size).to eq(files[0].size)
-          end
-
-          context 'when revision is not specified' do
-
-            let!(:result) do
-              fs.open_download_stream_by_name('test.txt') do |stream|
-                io.write(stream.read)
-              end
-            end
-
-            it 'returns the most recent version' do
-              expect(io.string).to eq('hello 4')
-            end
-          end
-
-          context 'when revision is 0' do
-
-            let!(:result) do
-              fs.open_download_stream_by_name('test.txt', revision: 0) do |stream|
-                io.write(stream.read)
-              end
-            end
-
-            it 'returns the original stored file' do
-              expect(io.string).to eq('hello 1')
-            end
-          end
-
-          context 'when revision is negative' do
-
-            let!(:result) do
-              fs.open_download_stream_by_name('test.txt', revision: -2) do |stream|
-                io.write(stream.read)
-              end
-            end
-
-            it 'returns that number of versions from the most recent' do
-              expect(io.string).to eq('hello 3')
-            end
-          end
-
-          context 'when revision is positive' do
-
-            let!(:result) do
-              fs.open_download_stream_by_name('test.txt', revision: 1) do |stream|
-                io.write(stream.read)
-              end
-            end
-
-            it 'returns that number revision' do
-              expect(io.string).to eq('hello 2')
-            end
-          end
-
-          context 'when the file revision is not found' do
-
-            it 'raises a FileNotFound error' do
-              expect {
-                fs.open_download_stream_by_name('test.txt', revision: 100)
-              }.to raise_exception(Mongo::Error::InvalidFileRevision)
-            end
-          end
-
-          context 'when the file is not found' do
-
-            it 'raises a FileNotFound error' do
-              expect {
-                fs.open_download_stream_by_name('non-existent.txt')
-              }.to raise_exception(Mongo::Error::FileNotFound)
-            end
-          end
-        end
-
-        context 'when a block is not provided' do
-
-          let!(:stream) do
-            fs.open_download_stream_by_name('test.txt')
-          end
-
-          it 'returns a Stream::Read object' do
-            expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Read)
-          end
-
-          it 'does not close the stream' do
-            expect(stream.closed?).to be(false)
-          end
-
-          it 'does not yield the stream to the block' do
-            expect(io.size).to eq(0)
-          end
-        end
-      end
-    end
-  end
-
-  context 'when a write stream is opened' do
-
-    let(:stream) do
-      fs.open_upload_stream(filename)
-    end
-
-    describe '#open_upload_stream' do
-
-      context 'when a block is not provided' do
-
-        it 'returns a Stream::Write object' do
-          expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Write)
-        end
-
-        it 'creates an ObjectId for the file' do
-          expect(stream.file_id).to be_a(BSON::ObjectId)
-        end
-
-        context 'when a custom file ID is provided' do
-
-          let(:stream) do
-            fs.open_upload_stream(filename, file_id: 'Custom ID')
-          end
-
-          it 'returns a Stream::Write object' do
-            expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Write)
-          end
-
-          it 'creates an ObjectId for the file' do
-            expect(stream.file_id).to eq('Custom ID')
-          end
-        end
-      end
-
-      context 'when a block is provided' do
-
-        context 'when a session is not used' do
-
-          let!(:stream) do
-            fs.open_upload_stream(filename) do |stream|
-              stream.write(file)
-            end
-          end
-
-          let(:result) do
-            fs.find_one(filename: filename)
-          end
-
-          it 'returns the stream' do
-            expect(stream).to be_a(Mongo::Grid::FSBucket::Stream::Write)
-          end
-
-          it 'creates an ObjectId for the file' do
-            expect(stream.file_id).to be_a(BSON::ObjectId)
-          end
-
-          it 'yields the stream to the block' do
-            expect(result.data.size).to eq(file.size)
-          end
-
-          it 'closes the stream when the block completes' do
-            expect(stream.closed?).to be(true)
-          end
-        end
-      end
-
-    end
-
-    describe '#upload_from_stream' do
-
-      let!(:result) do
-        fs.upload_from_stream(filename, file)
-      end
-
-      let(:file_from_db) do
-        fs.find_one(:filename => filename)
-      end
-
-      it 'writes to the provided stream' do
-        expect(file_from_db.data.length).to eq(file.size)
-      end
-
-      it 'does not close the stream' do
-        expect(file.closed?).to be(false)
-      end
-
-      it 'returns the id of the file' do
-        expect(result).to be_a(BSON::ObjectId)
-      end
-
-      context 'when the io stream raises an error' do
-
-        let(:stream) do
-          fs.open_upload_stream(filename)
-        end
-
-        before do
-          allow(fs).to receive(:open_upload_stream).and_yield(stream)
-        end
-
-        context 'when stream#abort does not raise an OperationFailure' do
-
-          before do
-            expect(stream).to receive(:abort).and_call_original
-            file.close
-          end
-
-          it 'raises the original IOError' do
-            expect {
-              fs.upload_from_stream(filename, file)
-            }.to raise_exception(IOError)
-          end
-
-          it 'closes the stream' do
-            begin; fs.upload_from_stream(filename, file); rescue; end
-            expect(stream.closed?).to be(true)
-          end
-        end
-
-        context 'when stream#abort raises an OperationFailure' do
-
-          before do
-            allow(stream).to receive(:abort).and_raise(Mongo::Error::OperationFailure)
-            file.close
-          end
-
-          it 'raises the original IOError' do
-            expect {
-              fs.upload_from_stream(filename, file)
-            }.to raise_exception(IOError)
-          end
-        end
-      end
-    end
-
-    context 'when options are provided when opening the write stream' do
-
-      let(:stream) do
-        fs.open_upload_stream(filename, stream_options)
-      end
-
-      context 'when a custom file id is provided' do
-
-        let(:stream_options) do
-          { file_id: 'Custom ID' }
-        end
-
-        it 'sets the file id on the stream' do
-          expect(stream.file_id).to eq('Custom ID')
-        end
-      end
-
-      context 'when a write option is specified' do
-
-        let(:stream_options) do
-          { write: { w: 2 } }
-        end
-
-        it 'sets the write concern on the write stream' do
-          expect(stream.write_concern.options).to eq(Mongo::WriteConcern.get(stream_options[:write]).options)
-        end
-      end
-
-      context 'when there is a chunk size set on the FSBucket' do
-
-        let(:stream_options) do
-          {  }
-        end
-
-        let(:options) do
-          { chunk_size: 100 }
-        end
-
-        it 'sets the chunk size as the default on the write stream' do
-          expect(stream.options[:chunk_size]).to eq(options[:chunk_size])
-        end
-      end
-
-      context 'when a chunk size option is specified' do
-
-        let(:stream_options) do
-          { chunk_size: 50 }
-        end
-
-        it 'sets the chunk size on the write stream' do
-          expect(stream.options[:chunk_size]).to eq(stream_options[:chunk_size])
-        end
-
-        context 'when there is a chunk size set on the FSBucket' do
-
-          let(:options) do
-            { chunk_size: 100 }
-          end
-
-          let(:fs) do
-            described_class.new(authorized_client.database, options)
-          end
-
-          it 'uses the chunk size set on the write stream' do
-            expect(stream.options[:chunk_size]).to eq(stream_options[:chunk_size])
-          end
-
-        end
-      end
-
-      context 'when a file metadata option is specified' do
-
-        let(:stream_options) do
-          { metadata: { some_field: 1 } }
-        end
-
-        it 'sets the file metadata option on the write stream' do
-          expect(stream.options[:metadata]).to eq(stream_options[:metadata])
-        end
-      end
-
-      context 'when a content type option is specified' do
-
-        let(:stream_options) do
-          { content_type: 'text/plain' }
-        end
-
-        it 'sets the content type on the write stream' do
-          expect(stream.options[:content_type]).to eq(stream_options[:content_type])
-        end
-      end
-
-      context 'when a aliases option is specified' do
-
-        let(:stream_options) do
-          { aliases: [ 'another-name.txt' ] }
-        end
-
-        it 'sets the alias option on the write stream' do
-          expect(stream.options[:aliases]).to eq(stream_options[:aliases])
-        end
-      end
-    end
-  end
-end