async def test_write_file_like(self): one = motor_asyncio.AsyncIOMotorGridIn(self.db.fs) await one.write(b"hello world") await one.close() two = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, one._id) three = motor_asyncio.AsyncIOMotorGridIn(self.db.fs) await three.write(two) await three.close() four = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, three._id) self.assertEqual(b"hello world", (await four.read()))
async def test_iteration(self): fs = motor_asyncio.AsyncIOMotorGridFSBucket(self.db) _id = await fs.upload_from_stream("filename", b"foo") g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, _id) # Iteration is prohibited. self.assertRaises(TypeError, iter, g)
async def test_attributes(self): f = motor_asyncio.AsyncIOMotorGridIn(self.db.fs, filename="test", foo="bar", content_type="text") await f.close() g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, f._id) attr_names = ( "_id", "filename", "name", "name", "content_type", "length", "chunk_size", "upload_date", "aliases", "metadata", ) for attr_name in attr_names: self.assertRaises(InvalidOperation, getattr, g, attr_name) await g.open() for attr_name in attr_names: getattr(g, attr_name)
async def test_basic(self): f = motor_asyncio.AsyncIOMotorGridIn(self.db.fs, filename="test") await f.write(b"hello world") await f.close() self.assertEqual(1, (await self.db.fs.files.count_documents({}))) self.assertEqual(1, (await self.db.fs.chunks.count_documents({}))) g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, f._id) self.assertEqual(b"hello world", (await g.read())) f = motor_asyncio.AsyncIOMotorGridIn(self.db.fs, filename="test") await f.close() self.assertEqual(2, (await self.db.fs.files.count_documents({}))) self.assertEqual(1, (await self.db.fs.chunks.count_documents({}))) g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, f._id) self.assertEqual(b"", (await g.read()))
async def test_grid_out_file_document(self): one = motor_asyncio.AsyncIOMotorGridIn(self.db.fs) await one.write(b"foo bar") await one.close() file_document = await self.db.fs.files.find_one() two = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, file_document=file_document) self.assertEqual(b"foo bar", (await two.read())) file_document = await self.db.fs.files.find_one() three = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, 5, file_document) self.assertEqual(b"foo bar", (await three.read())) gridout = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, file_document={}) with self.assertRaises(NoFile): await gridout.open()
async def test_gridout_open_exc_info(self): g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, "_id that doesn't exist") try: await g.open() except NoFile: _, _, tb = sys.exc_info() # The call tree should include PyMongo code we ran on a thread. formatted = "\n".join(traceback.format_tb(tb)) self.assertTrue("_ensure_file" in formatted)
async def test_alternate_collection(self): await self.db.alt.files.delete_many({}) await self.db.alt.chunks.delete_many({}) f = motor_asyncio.AsyncIOMotorGridIn(self.db.alt) await f.write(b"hello world") await f.close() self.assertEqual(1, (await self.db.alt.files.count_documents({}))) self.assertEqual(1, (await self.db.alt.chunks.count_documents({}))) g = motor_asyncio.AsyncIOMotorGridOut(self.db.alt, f._id) self.assertEqual(b"hello world", (await g.read()))
async def test_gridout_open_exc_info(self): if sys.version_info < (3, ): raise SkipTest("Requires Python 3") g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, "_id that doesn't exist") try: await g.open() except NoFile: _, _, tb = sys.exc_info() # The call tree should include PyMongo code we ran on a thread. formatted = '\n'.join(traceback.format_tb(tb)) self.assertTrue('_ensure_file' in formatted)
async def test_alternate_collection(self): await self.db.alt.files.delete_many({}) await self.db.alt.chunks.delete_many({}) f = motor_asyncio.AsyncIOMotorGridIn(self.db.alt) await f.write(b"hello world") await f.close() self.assertEqual(1, (await self.db.alt.files.count_documents({}))) self.assertEqual(1, (await self.db.alt.chunks.count_documents({}))) g = motor_asyncio.AsyncIOMotorGridOut(self.db.alt, f._id) self.assertEqual(b"hello world", (await g.read())) # test that md5 still works... self.assertEqual("5eb63bbbe01eeed093cb22bb8f5acdc3", g.md5)
async def test_grid_out_default_opts(self): self.assertRaises(TypeError, motor_asyncio.AsyncIOMotorGridOut, "foo") gout = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, 5) with self.assertRaises(NoFile): await gout.open() a = motor_asyncio.AsyncIOMotorGridIn(self.db.fs) await a.close() b = await motor_asyncio.AsyncIOMotorGridOut(self.db.fs, a._id).open() self.assertEqual(a._id, b._id) self.assertEqual(0, b.length) self.assertEqual(None, b.content_type) self.assertEqual(255 * 1024, b.chunk_size) self.assertTrue(isinstance(b.upload_date, datetime.datetime)) self.assertEqual(None, b.aliases) self.assertEqual(None, b.metadata)
async def test_readchunk(self): in_data = b"a" * 10 f = motor_asyncio.AsyncIOMotorGridIn(self.db.fs, chunkSize=3) await f.write(in_data) await f.close() g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, f._id) # This is starting to look like Lisp. self.assertEqual(3, len((await g.readchunk()))) self.assertEqual(2, len((await g.read(2)))) self.assertEqual(1, len((await g.readchunk()))) self.assertEqual(3, len((await g.read(3)))) self.assertEqual(1, len((await g.readchunk()))) self.assertEqual(0, len((await g.readchunk())))
async def test_attributes(self): f = motor_asyncio.AsyncIOMotorGridIn(self.db.fs, filename="test", foo="bar", content_type="text") await f.close() g = motor_asyncio.AsyncIOMotorGridOut(self.db.fs, f._id) attr_names = ('_id', 'filename', 'name', 'name', 'content_type', 'length', 'chunk_size', 'upload_date', 'aliases', 'metadata', 'md5') for attr_name in attr_names: self.assertRaises(InvalidOperation, getattr, g, attr_name) await g.open() for attr_name in attr_names: getattr(g, attr_name)