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 init_app(self, app):
self.mongo = Mongo(app)
self.elastic = Elastic(app)
if 'DEFAULT_FILE_STORAGE' in app.config:
self.storage = import_by_path(app.config['DEFAULT_FILE_STORAGE'])()
self.storage.init_app(app)
else:
self.storage = self.driver
def test_combine_queries(self):
mongo = Mongo(None)
query_a = {'username': {'$exists': True}}
query_b = {'username': '******'}
combined = mongo.combine_queries(query_a, query_b)
self.assertEqual(
combined,
{'$and': [{'username': {'$exists': True}}, {'username': '******'}]}
)
def init_app(self, app):
app.data = self # app.data must be set for locks to work
self.mongo = Mongo(app)
self.driver = self.mongo.driver
self.storage = self.driver
self.elastic = Elastic(app,
serializer=SuperdeskJSONEncoder(),
skip_index_init=True,
retry_on_timeout=True)
def test_get_value_from_query(self):
mongo = Mongo(None)
simple_query = {'_id': 'abcdef012345678901234567'}
compound_query = {'$and': [
{'username': {'$exists': False}},
{'_id': 'abcdef012345678901234567'}
]}
self.assertEqual(mongo.get_value_from_query(simple_query, '_id'),
'abcdef012345678901234567')
self.assertEqual(mongo.get_value_from_query(compound_query, '_id'),
'abcdef012345678901234567')
def init_app(self, app):
self.mongo = Mongo(app)
self.elastic = Elastic(app)
self.custom = CustomDataLayer(app)
self.elastic.es.json_encoder = SuperdeskJsonEncoder
if 'DEFAULT_FILE_STORAGE' in app.config:
self.storage = import_by_path(app.config['DEFAULT_FILE_STORAGE'])()
self.storage.init_app(app)
else:
self.storage = self.driver
def test_query_contains_field(self):
mongo = Mongo(None)
simple_query = {'_id': 'abcdef012345678901234567'}
compound_query = {'$and': [
{'username': {'$exists': False}},
{'_id': 'abcdef012345678901234567'}
]}
self.assertTrue(mongo.query_contains_field(simple_query, '_id'))
self.assertFalse(mongo.query_contains_field(simple_query,
'fake-field'))
self.assertTrue(mongo.query_contains_field(compound_query, '_id'))
self.assertFalse(mongo.query_contains_field(compound_query,
'fake-field'))
def test_combine_queries(self):
mongo = Mongo(None)
query_a = {"username": {"$exists": True}}
query_b = {"username": "******"}
combined = mongo.combine_queries(query_a, query_b)
self.assertEqual(
combined,
{"$and": [{
"username": {
"$exists": True
}
}, {
"username": "******"
}]})
def test_query_contains_field(self):
mongo = Mongo(None)
simple_query = {"_id": "abcdef012345678901234567"}
compound_query = {
"$and": [
{
"username": {
"$exists": False
}
},
{
"_id": "abcdef012345678901234567"
},
]
}
self.assertTrue(mongo.query_contains_field(simple_query, "_id"))
self.assertFalse(mongo.query_contains_field(simple_query,
"fake-field"))
self.assertTrue(mongo.query_contains_field(compound_query, "_id"))
self.assertFalse(
mongo.query_contains_field(compound_query, "fake-field"))
def test_get_value_from_query(self):
mongo = Mongo(None)
simple_query = {"_id": "abcdef012345678901234567"}
compound_query = {
"$and": [
{
"username": {
"$exists": False
}
},
{
"_id": "abcdef012345678901234567"
},
]
}
self.assertEqual(mongo.get_value_from_query(simple_query, "_id"),
"abcdef012345678901234567")
self.assertEqual(
mongo.get_value_from_query(compound_query, "_id"),
"abcdef012345678901234567",
)
def test_json_encoder_class(self):
mongo = Mongo(None)
self.assertTrue((mongo.json_encoder_class(), MongoJSONEncoder))
self.assertTrue((mongo.json_encoder_class(), json.JSONEncoder))
def init_app(self, app):
self.mongo = Mongo(app)
self.elastic = Elastic(app)
self.aapmm = AAPMMDatalayer(app)
self.storage = self.driver
def init_app(self, app):
self.mongo = Mongo(app)
self.driver = self.mongo.driver
self.storage = self.driver
self.elastic = Elastic(app, serializer=SuperdeskJSONEncoder())
