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)