def setUp(self): self.app = flask.Flask(__name__) self.app.config['SERVER_NAME'] = 'localhost' self.app.data = Mongo(self.app) self.app.config['DOMAIN'] = {'upload': {}} self.media = KoldoctaGridFSMediaStorage(self.app) self.app.register_blueprint(bp)
def setUp(self): self.app = flask.Flask(__name__) self.app.config['SERVER_NAME'] = 'localhost' self.app.data = Mongo(self.app) self.app.config['DOMAIN'] = {'upload': {}} self.media = KoldoctaGridFSMediaStorage(self.app) self.app.register_blueprint(bp)
class GridFSMediaStorageTestCase(unittest.TestCase): def setUp(self): self.app = flask.Flask(__name__) self.app.config['SERVER_NAME'] = 'localhost' self.app.data = Mongo(self.app) self.app.config['DOMAIN'] = {'upload': {}} self.media = KoldoctaGridFSMediaStorage(self.app) self.app.register_blueprint(bp) def test_media_id(self): filename = 'some-file' media_id = self.media.media_id(filename) self.assertIsInstance(media_id, bson.ObjectId) self.assertEqual(media_id, self.media.media_id(filename)) def test_url_for_media(self): _id = self.media.media_id('test') with self.app.app_context(): url = self.media.url_for_media(_id) self.assertEqual('http://localhost/upload/%s/raw' % _id, url) def test_put_media_with_id(self): data = io.StringIO(u"test data") filename = 'x' gridfs = Mock() gridfs.put = Mock(return_value='y') self.media._fs['MONGO'] = 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)
class GridFSMediaStorageTestCase(unittest.TestCase): def setUp(self): self.app = flask.Flask(__name__) self.app.config['SERVER_NAME'] = 'localhost' self.app.data = Mongo(self.app) self.app.config['DOMAIN'] = {'upload': {}} self.media = KoldoctaGridFSMediaStorage(self.app) self.app.register_blueprint(bp) def test_media_id(self): filename = 'some-file' media_id = self.media.media_id(filename) self.assertIsInstance(media_id, bson.ObjectId) self.assertEqual(media_id, self.media.media_id(filename)) def test_url_for_media(self): _id = self.media.media_id('test') with self.app.app_context(): url = self.media.url_for_media(_id) self.assertEqual('http://localhost/upload/%s/raw' % _id, url) def test_put_media_with_id(self): data = io.StringIO(u"test data") filename = 'x' gridfs = Mock() gridfs.put = Mock(return_value='y') self.media._fs['MONGO'] = 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)