class GridFSMediaStorageTestCase(unittest.TestCase):
def setUp(self):
self.app = eve.Eve(__name__, {'DOMAIN': {}})
self.app.config['SERVER_NAME'] = 'localhost'
self.app.config['DOMAIN'] = {'upload': {}}
self.app.config['MONGO_DBNAME'] = 'sptests'
self.app.data = SuperdeskDataLayer(self.app)
self.media = SuperdeskGridFSMediaStorage(self.app)
self.app.register_blueprint(bp)
self.app.upload_url = upload_url
def test_url_for_media(self):
_id = bson.ObjectId(sha('test')[:24])
with self.app.app_context():
url = self.media.url_for_media(_id)
self.assertEqual('http://localhost/upload-raw/%s' % _id, url)
def test_url_for_media_content_type(self):
_id_str = '1' * 24
_id = bson.ObjectId(_id_str)
with self.app.app_context():
url = self.media.url_for_media(_id, "image/jpeg")
self.assertEqual('http://localhost/upload-raw/{}.jpg'.format(_id_str),
url)
def test_put_media_with_id(self):
data = io.StringIO("test data")
filename = 'x'
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
with self.app.app_context():
self.media.put(data, filename, 'text/plain', _id=str(_id))
kwargs = {
'content_type': 'text/plain',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_put_into_folder(self):
data = b'test data'
filename = 'x'
folder = 'gridtest'
gridfs = self._mock_gridfs()
with self.app.app_context():
self.media.put(data,
filename=filename,
content_type='text/plain',
folder=folder)
kwargs = {
'content_type': 'text/plain',
'filename': '{}/{}'.format(folder, filename),
'metadata': None
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_find_files(self):
gridfs = self._mock_gridfs()
upload_date = {'$lte': utcnow(), '$gte': utcnow() - timedelta(hours=1)}
folder = 'gridtest'
query_filename = {'filename': {'$regex': '^{}/'.format(folder)}}
query_upload_date = {'uploadDate': upload_date}
with self.app.app_context():
self.media.find(folder=folder, upload_date=upload_date)
gridfs.find.assert_called_once_with(
{'$and': [query_filename, query_upload_date]})
self.media.find(folder=folder)
gridfs.find.assert_called_with(query_filename)
self.media.find(upload_date=upload_date)
gridfs.find.assert_called_with(query_upload_date)
self.media.find()
gridfs.find.assert_called_with({})
def _mock_gridfs(self):
gridfs = Mock()
gridfs.put = Mock(return_value='y')
gridfs.find = Mock(return_value=[])
self.media._fs['MONGO'] = gridfs
return gridfs
class GridFSMediaStorageTestCase(unittest.TestCase):
def setUp(self):
self.app = eve.Eve(__name__, {'DOMAIN': {}})
self.app.config['MEDIA_PREFIX'] = 'http://localhost/upload-raw'
self.app.config['DOMAIN'] = {'upload': {}}
self.app.config['MONGO_DBNAME'] = 'sptests'
self.app.data = SuperdeskDataLayer(self.app)
self.media = SuperdeskGridFSMediaStorage(self.app)
self.app.register_blueprint(bp)
self.app.upload_url = upload_url
def test_url_for_media(self):
_id = bson.ObjectId(sha('test')[:24])
with self.app.app_context():
url = self.media.url_for_media(_id)
self.assertEqual('http://localhost/upload-raw/%s' % _id, url)
def test_url_for_media_content_type(self):
_id_str = '1' * 24
_id = bson.ObjectId(_id_str)
with self.app.app_context():
url = self.media.url_for_media(_id, "image/jpeg")
self.assertEqual('http://localhost/upload-raw/{}.jpg'.format(_id_str),
url)
def test_put_media_with_id(self):
data = io.StringIO("test data")
filename = 'x'
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
with self.app.app_context():
self.media.put(data, filename, 'text/plain', _id=str(_id))
kwargs = {
'content_type': 'text/plain',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_put_into_folder(self):
data = b'test data'
filename = 'x'
folder = 'gridtest'
gridfs = self._mock_gridfs()
with self.app.app_context():
self.media.put(data,
filename=filename,
content_type='text/plain',
folder=folder)
kwargs = {
'content_type': 'text/plain',
'filename': '{}/{}'.format(folder, filename),
'metadata': None
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_find_files(self):
gridfs = self._mock_gridfs()
upload_date = {'$lte': utcnow(), '$gte': utcnow() - timedelta(hours=1)}
folder = 'gridtest'
query_filename = {'filename': {'$regex': '^{}/'.format(folder)}}
query_upload_date = {'uploadDate': upload_date}
with self.app.app_context():
self.media.find(folder=folder, upload_date=upload_date)
gridfs.find.assert_called_once_with(
{'$and': [query_filename, query_upload_date]})
self.media.find(folder=folder)
gridfs.find.assert_called_with(query_filename)
self.media.find(upload_date=upload_date)
gridfs.find.assert_called_with(query_upload_date)
self.media.find()
gridfs.find.assert_called_with({})
def test_custom_id(self):
data = b'foo'
with self.app.app_context():
self.media.put(data, _id='foo')
_file = self.media.get('foo')
assert data == _file.read()
def _mock_gridfs(self):
gridfs = Mock()
gridfs.put = Mock(return_value='y')
gridfs.find = Mock(return_value=[])
self.media._fs['MONGO'] = gridfs
return gridfs
def test_mimetype_detect(self):
# keep default mimetype
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b'bytes are here'
filename = 'extensionless'
content_type = 'text/css'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type': content_type,
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
# get mimetype from the filename
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b'bytes are here'
filename = 'styles.css'
content_type = 'application/pdf'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type': 'text/css',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b'bytes are here'
filename = 'styles.JpG'
content_type = 'application/pdf'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type': 'image/jpeg',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
# get mimetype from the file
fixtures_path = os.path.join(os.path.dirname(__file__), 'fixtures')
with open(os.path.join(fixtures_path, "file_example-jpg.jpg"),
'rb') as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = 'extensionless'
content_type = 'dummy/text'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type': 'image/jpeg',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-xls.xls"),
'rb') as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = 'extensionless'
content_type = 'dummy/text'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type': 'application/vnd.ms-excel',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-xlsx.xlsx"),
'rb') as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = 'extensionless'
content_type = 'dummy/text'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type':
'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-docx.docx"),
'rb') as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = 'extensionless'
content_type = 'dummy/text'
with self.app.app_context():
self.media.put(content, filename, content_type, _id=str(_id))
kwargs = {
'content_type':
'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
class GridFSMediaStorageTestCase(unittest.TestCase):
def setUp(self):
self.app = eve.Eve(__name__, {"DOMAIN": {}})
self.app.config["MEDIA_PREFIX"] = "http://localhost/upload-raw"
self.app.config["DOMAIN"] = {"upload": {}}
self.app.config["MONGO_DBNAME"] = "sptests"
self.app.data = SuperdeskDataLayer(self.app)
self.media = SuperdeskGridFSMediaStorage(self.app)
self.app.register_blueprint(bp)
self.app.upload_url = upload_url
def test_url_for_media(self):
_id = bson.ObjectId(sha("test")[:24])
with self.app.app_context():
url = self.media.url_for_media(_id)
self.assertEqual("http://localhost/upload-raw/%s" % _id, url)
def test_url_for_media_content_type(self):
_id_str = "1" * 24
_id = bson.ObjectId(_id_str)
with self.app.app_context():
url = self.media.url_for_media(_id, "image/jpeg")
self.assertEqual("http://localhost/upload-raw/{}.jpg".format(_id_str), url)
def test_put_media_with_id(self):
data = io.StringIO("test data")
filename = "x"
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
with self.app.app_context():
self.media.put(data, filename=filename, content_type="text/plain", _id=str(_id))
kwargs = {
"content_type": "text/plain",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_put_into_folder(self):
data = b"test data"
filename = "x"
folder = "gridtest"
gridfs = self._mock_gridfs()
with self.app.app_context():
self.media.put(data, filename=filename, content_type="text/plain", folder=folder)
kwargs = {"content_type": "text/plain", "filename": "{}/{}".format(folder, filename), "metadata": None}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_find_files(self):
gridfs = self._mock_gridfs()
upload_date = {"$lte": utcnow(), "$gte": utcnow() - timedelta(hours=1)}
folder = "gridtest"
query_filename = {"filename": {"$regex": "^{}/".format(folder)}}
query_upload_date = {"uploadDate": upload_date}
with self.app.app_context():
self.media.find(folder=folder, upload_date=upload_date)
gridfs.find.assert_called_once_with({"$and": [query_filename, query_upload_date]})
self.media.find(folder=folder)
gridfs.find.assert_called_with(query_filename)
self.media.find(upload_date=upload_date)
gridfs.find.assert_called_with(query_upload_date)
self.media.find()
gridfs.find.assert_called_with({})
def test_custom_id(self):
data = b"foo"
with self.app.app_context():
self.media.put(data, _id="foo")
_file = self.media.get("foo")
assert data == _file.read()
def _mock_gridfs(self):
gridfs = Mock()
gridfs.put = Mock(return_value="y")
gridfs.find = Mock(return_value=[])
self.media._fs["MONGO"] = gridfs
return gridfs
def test_mimetype_detect(self):
# keep default mimetype
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b"bytes are here"
filename = "extensionless"
content_type = "text/css"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": content_type,
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
# get mimetype from the filename
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b"bytes are here"
filename = "styles.css"
content_type = "application/pdf"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "text/css",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
content = b"bytes are here"
filename = "styles.JpG"
content_type = "application/pdf"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "image/jpeg",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
# get mimetype from the file
fixtures_path = os.path.join(os.path.dirname(__file__), "fixtures")
with open(os.path.join(fixtures_path, "file_example-jpg.jpg"), "rb") as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = "extensionless"
content_type = "dummy/text"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "image/jpeg",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-xls.xls"), "rb") as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = "extensionless"
content_type = "dummy/text"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "application/vnd.ms-excel",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-xlsx.xlsx"), "rb") as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = "extensionless"
content_type = "dummy/text"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
with open(os.path.join(fixtures_path, "file_example-docx.docx"), "rb") as content:
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
filename = "extensionless"
content_type = "dummy/text"
with self.app.app_context():
self.media.put(content, filename=filename, content_type=content_type, _id=str(_id))
kwargs = {
"content_type": "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
"filename": filename,
"metadata": None,
"_id": _id,
}
gridfs.put.assert_called_once_with(content, **kwargs)
class GridFSMediaStorageTestCase(unittest.TestCase):
def setUp(self):
self.app = eve.Eve(__name__, {'DOMAIN': {}})
self.app.config['MEDIA_PREFIX'] = 'http://localhost/upload-raw'
self.app.config['DOMAIN'] = {'upload': {}}
self.app.config['MONGO_DBNAME'] = 'sptests'
self.app.data = SuperdeskDataLayer(self.app)
self.media = SuperdeskGridFSMediaStorage(self.app)
self.app.register_blueprint(bp)
self.app.upload_url = upload_url
def test_url_for_media(self):
_id = bson.ObjectId(sha('test')[:24])
with self.app.app_context():
url = self.media.url_for_media(_id)
self.assertEqual('http://localhost/upload-raw/%s' % _id, url)
def test_url_for_media_content_type(self):
_id_str = '1' * 24
_id = bson.ObjectId(_id_str)
with self.app.app_context():
url = self.media.url_for_media(_id, "image/jpeg")
self.assertEqual('http://localhost/upload-raw/{}.jpg'.format(_id_str), url)
def test_put_media_with_id(self):
data = io.StringIO("test data")
filename = 'x'
gridfs = self._mock_gridfs()
_id = bson.ObjectId()
with self.app.app_context():
self.media.put(data, filename, 'text/plain', _id=str(_id))
kwargs = {
'content_type': 'text/plain',
'filename': filename,
'metadata': None,
'_id': _id,
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_put_into_folder(self):
data = b'test data'
filename = 'x'
folder = 'gridtest'
gridfs = self._mock_gridfs()
with self.app.app_context():
self.media.put(data, filename=filename, content_type='text/plain', folder=folder)
kwargs = {
'content_type': 'text/plain',
'filename': '{}/{}'.format(folder, filename),
'metadata': None
}
gridfs.put.assert_called_once_with(data, **kwargs)
def test_find_files(self):
gridfs = self._mock_gridfs()
upload_date = {'$lte': utcnow(), '$gte': utcnow() - timedelta(hours=1)}
folder = 'gridtest'
query_filename = {'filename': {'$regex': '^{}/'.format(folder)}}
query_upload_date = {'uploadDate': upload_date}
with self.app.app_context():
self.media.find(folder=folder, upload_date=upload_date)
gridfs.find.assert_called_once_with({'$and': [query_filename, query_upload_date]})
self.media.find(folder=folder)
gridfs.find.assert_called_with(query_filename)
self.media.find(upload_date=upload_date)
gridfs.find.assert_called_with(query_upload_date)
self.media.find()
gridfs.find.assert_called_with({})
def test_custom_id(self):
data = b'foo'
with self.app.app_context():
self.media.put(data, _id='foo')
_file = self.media.get('foo')
assert data == _file.read()
def _mock_gridfs(self):
gridfs = Mock()
gridfs.put = Mock(return_value='y')
gridfs.find = Mock(return_value=[])
self.media._fs['MONGO'] = gridfs
return gridfs