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)