def test_put(self):
fs = grid_fs_accessor.GridFsAccessor()
self.fs.put.return_value = 'file_id'
res = fs.put('data', kwarg_1='a', kwarg_2='b')
self.fs.put.assert_called_with('data', kwarg_1='a', kwarg_2='b')
self.assertEqual(res, 'file_id')
self.mock_logger.debug.assert_called_with(
'Saved file with file_id: file_id')
def test_find(self):
fs = grid_fs_accessor.GridFsAccessor()
self.fs.find_one.return_value = 'data'
res = fs.find_one('query', 'session', 'arg', bar='bar')
self.fs.find_one.assert_called_with('arg',
filter='query',
session='session',
bar='bar')
self.assertEqual(res, 'data')
def test_insert_image_and_thumbnail(self, fn, expected_fn):
content = b'image_content'
self.mock_streamer.load.return_value.dump.return_value = {
'image': content,
'filename': 'mock_fn',
'mimetype': 'mime',
'thumbnail': 'mock_thumbnail'
}
grid_fs_accessor.GridFsAccessor().insert_image_and_thumbnail(
content, fn)
self.mock_streamer.load.assert_called_with(content)
self.fs.put.assert_any_call(content,
_id='mock_id',
filename=expected_fn,
mimetype='mime')
self.fs.put.assert_any_call('mock_thumbnail',
_id='mock_id',
filename=f'thumbnail_{expected_fn}',
mimetype='mime')
def test_delete(self):
grid_fs_accessor.GridFsAccessor().delete('file_id')
self.fs.delete.assert_called_with('file_id')
self.mock_logger.debug.assert_called_with(
'Removing file with file_id: file_id')
def test_get(self):
fs = grid_fs_accessor.GridFsAccessor()
self.fs.get.return_value = 'data'
res = fs.get('file_id')
self.fs.get.assert_called_with('file_id')
self.assertEqual(res, 'data')
def test_init(self, db, exp_db, uri, exp_uri):
self.mock_env.mongodb_uri = uri
fs = grid_fs_accessor.GridFsAccessor(db=db)
self.assertEqual(fs.db, exp_db)
self.assertEqual(fs.uri, exp_uri)
self.mock_gridfs.GridFS.assert_called_with(exp_db)