def test_obs_buys5000():
stocks = ['DIS', 'NTDOY', 'ATVI', 'SGAMY', 'UBSFY']
sql_ins = 'INSERT INTO buy_sell (b_type,username,price,t_account,stocktype,quantity) VALUES'
for stock in stocks:
price = mainapp.get_delayed_price(stock)
sql_ins += '(\'BUY\',\'admin\',' + str(
price) + ',\'Bank Stock Inventory\',\'' + stock + '\',5000),'
sql_ins -= ','
sql_delete = 'drop table buy_sell'
sql_table = 'create table buy_sell( '
sql_table += 'buy_sellid int auto_increment primary key, '
sql_table += 'b_type varchar(45), '
sql_table += 'username varchar(240), '
sql_table += 'price float, '
sql_table += 't_account varchar(240), '
sql_table += 'stocktype varchar(240), '
sql_table += 'quantity int);'
#reset the table starting with a buy of 5000
mainapp.query_db(sql_delete)
mainapp.query_db(sql_table)
mainapp.query_db(sql_ins)
#Check that the OBS begins with 5000 of each stock type in its inventory
for stock in stocks:
test_query = 'SELECT * FROM buy_sell WHERE username = \'admin\' AND stocktype =\'' + stock + '\''
ret = mainapp.query_db(test_query)
assert 'BUY' == str(ret[0][1])
assert 'admin' == str(ret[0][2])
assert stock == str(ret[0][5])
assert 5000 == int(ret[0][6])
def test_obs_buys5000():
price = mainapp.get_delayed_price()
sql = 'SELECT * FROM buy_sell where bid = 1 AND username = \'admin\''
sql_delete = 'drop table buy_sell'
sql_ins = 'INSERT INTO buy_sell (b_type,username,t_account,price,quantity) values(\'BUY\',\'admin\',\'Bank Stock Inventory\',' + str(
price) + ',5000)'
ret = mainapp.query_db(sql)
sql_table = 'create table buy_sell( '
sql_table += 'bid int auto_increment primary key, '
sql_table += 'b_type varchar(10), '
sql_table += 'username varchar(50), '
sql_table += 't_account varchar(50), '
sql_table += 'price numeric(10,2), '
sql_table += 'quantity int)'
#reset the table starting with a buy of 5000
mainapp.query_db(sql_delete)
mainapp.query_db(sql_table)
mainapp.query_db(sql_ins)
#check that first buy is an OBS buy at 5000
ret = mainapp.query_db(sql)
assert 'BUY' == str(ret[0][1])
assert 'admin' == str(ret[0][2])
assert 5000 == int(ret[0][5])
def test_get_delayed_price():
assert type(mainapp.get_delayed_price()) == type(1.0)
def test_get_delayed_price():
assert 0.0 == mainapp.get_delayed_price('BLAH')
assert 0.0 != mainapp.get_delayed_price('SGAMY')
def test_get_delayed_price():
stocks = ['DIS', 'NTDOY', 'ATVI', 'SGAMY', 'UBSFY']
for stock in stocks:
res = mainapp.get_delayed_price(stock)
assert type(res) == type(1.0)