def test_query(self):
"""Test query method."""
c = RestifyCollection.query(self.connection, self.database_name,
self.collection_name)
self.assertTrue(c)
aq = RestifyCollection.query(self.connection, self.database_name,
self.collection_name, spec={'name': "Jesse"})
self.assertEqual(aq.result[0].age, 21)
self.assertEqual(aq.result[0].name, "Jesse")
self.assertEqual(len(aq.result), 1)
def api_collection(class_name):
if request.method == "GET":
query_object = request.args.get("where")
if query_object:
try:
query_object = json.loads(query_object)
except Exception, e:
return exceptions.BadRequest(e)
collection = RestifyCollection.query(db, db_name, class_name, spec=query_object)
else:
collection = RestifyCollection.query(db, db_name, class_name)
return jsonify(collection.to_dict())
def test_to_dict(self):
"""Test to_dict method."""
c = RestifyCollection.query(self.connection, self.database_name,
self.collection_name)
self.assertTrue(c)
fixture_result_dict = {
'result': [obj.to_dict() for obj in self.plenty_fixture_objects]
}
self.assertTrue(c.to_dict(), fixture_result_dict)
