def test_stock_transaction_sanity(self):
stock_db_connection = get_default_db_connection()
reset_table(stock_db_connection)
stock_transaction_table = StockTransactionTable(stock_db_connection)
stock_transaction = StockTransaction()
stock_transaction.set_symbol("601398")
stock_transaction.set_buy_or_sell("buy")
stock_transaction.set_quantity(100)
stock_transaction.set_price(4.51)
stock_transaction.set_date(date(2015, 11, 10))
stock_transaction_table.add_stock_transaction(stock_transaction)
stock_transaction = StockTransaction()
stock_transaction.set_symbol("601857")
stock_transaction.set_buy_or_sell("buy")
stock_transaction.set_quantity(100)
stock_transaction.set_price(4.51)
stock_transaction.set_date(date(2015, 11, 10))
stock_transaction_table.add_stock_transaction(stock_transaction)
stock_transaction = \
stock_transaction_table.get_stock_transaction_by_trans_id(1)
stock_transaction.set_quantity("500")
stock_transaction_table.update_stock_transaction(stock_transaction)
stock_transaction = \
stock_transaction_table.get_stock_transaction_by_trans_id(1)
stock_transaction_table.delete_stock_transaction(stock_transaction)
return
def update_stock_transaction(self):
trans_id = int(self.entryTransId.get())
stock_transaction_table = StockTransactionTable()
stock_transaction = \
stock_transaction_table.get_stock_transaction_by_trans_id(trans_id)
symbol = self.entrySymbol.get()
buy_or_sell = self.entryBuyOrSell.get()
quantity = int(self.entryQuantity.get())
price = float(self.entryPrice.get())
date = self.entryDate.get()
stock_transaction.set_symbol(symbol)
stock_transaction.set_buy_or_sell(buy_or_sell)
stock_transaction.set_quantity(quantity)
stock_transaction.set_price(price)
stock_transaction_table.update_stock_transaction(stock_transaction)
self.refresh_list_box()
return
