def remove(id): try: #user.remove({'stock':ticker}) stock = Stock(id=id) sym = stock['symbol'] stock.delete() return render_template('remove.html', ticker=sym) except: return render_template('uhoh.html')
def test_create_stock_no_duplicate(self): """Makes sure stock don't have the same stock_id when creating.""" Category.create(category_id='001', description='Testing Stock') create_stock('001', 'Testing stock', 1, '001', 9.99) from peewee import IntegrityError with self.assertRaises(IntegrityError): create_stock('001', 'Testing stock', 1, '001', 9.99) q = Stock.delete().where(Stock.stock_id == '001') q.execute() t = Category.delete().where(Category.category_id == '001') t.execute()
def test_create_stock(self): """Make sure it create a stock.""" Category.create(category_id='001', description='Testing Stock') create_stock('001', 'Testing stock', 1, '001', 9.99) s = Stock.select().where(Stock.stock_id == '001') t = Stock.delete().where(Stock.stock_id == '001') t.execute() t = Category.delete().where(Category.category_id == '001') t.execute() self.assertEqual(str(s), ("<class 'models.Stock'> SELECT `t1`.`id`," " `t1`.`stock_id`, `t1`.`description`," " `t1`.`quantity`, `t1`.`category_id`," " `t1`.`price` FROM `stock` AS t1" " WHERE (`t1`.`stock_id` = %s) ['001']"))
def test_create_incoming(self): """Create an incoming record.""" Category.create(category_id='001', description='Testing Stock') create_stock('001', 'Testing stock', 1, '001', 9.99) create_incoming_stock(stock="001", date="2015-07-22", quantity=13, price=13.99) p = IncomingStock.select().where(IncomingStock.stock == '001') q = IncomingStock.delete().where(IncomingStock.stock == '001') q.execute() s = Stock.select().where(Stock.stock_id == '001') t = Stock.delete().where(Stock.stock_id == '001') t.execute() t = Category.delete().where(Category.category_id == '001') t.execute() self.assertEqual(str(p), ("<class 'models.IncomingStock'> SELECT `t1`.`id`, `t1`.`stock_id`, " "`t1`.`date`, `t1`.`price`, `t1`.`cost` FROM `incomingstock` AS t1 WHERE " "(`t1`.`stock_id` = %s) ['001']"))