def test_find_all_stocks(self): stocks = Stock.find_all(test_mode=True) stocks = stocks['stocks'] stock_1 = stocks[0] self._assert_first_stock(stock_1) stock_2 = stocks[1] self._assert_second_stock(stock_2)
def get_all_stocks(name=None): search_name = request.args.get('name') if search_name: stock = Stock.find_one({'name': search_name}, test_mode=app.testing) return jsonify(stock=stock) industry = request.args.get('industry') if industry: stocks = Stock.find({'industry': industry}, test_mode=app.testing) return jsonify(stocks=stocks) # Get all stocks stocks = Stock.find_all(test_mode=app.testing) return jsonify(stocks=stocks['stocks'])