def recv_inventory_add(self, environ, start_response):
formdata = environ['QUERY_STRING']
results = urlparse.parse_qs(formdata)
if ( 'liquor' in results.keys() ):
liquorStr = results['liquor'][0]
else:
liquorStr = "#"
content_type = 'text/html'
data = liquorStr
fp = StringIO(data)
load_bulk_data.load_inventory(fp)
data = "Added Liquor: %s<p><a href='./'>Index</a>" % liquorStr
start_response('200 OK', list(html_headers) )
return [data]
def test_bulk_load_inventory_3():
db._reset_db()
db.add_bottle_type('Johnnie Walker', 'Black Label', 'blended scotch')
data = "Johnnie Walker,Black Label,1000 ml\na,b\n"
fp = StringIO(data) # make this look like a file handle
n = load_bulk_data.load_inventory(fp)
assert db.check_inventory('Johnnie Walker', 'Black Label')
assert n == 1, n
def comment_test(): db._reset_db() #1 mfg = 'Johnnie Walker' #2 name = 'Black Label' typ = 'Blended Scotch' amt = '1000ml' st_typ = "#this comment is not suitable for children\n" + mfg + "," + name + "," + typ + "\n#nor is this one." #3 st_amt = "#this comment is not suitable for children\n" + mfg + "," + name + "\n#nor is this one." fp_typ = StringIO(st_typ) #4 fp_amt = StringIO(st_amt) load_bulk_data.load_bottle_types(fp_typ) #5 assert db._check_bottle_type_exists(mfg, name) == True load_bulk_data.load_inventory(fp_amt) #6 assert db.check_inventory(mfg, name) == False
def wspace_test(): db._reset_db() #1 mfg = 'Johnnie Walker' #2 name = 'Black Label' typ = 'Blended Scotch' amt = '1000ml' st_typ = "\n \n \n" + mfg + "," + name + "," + typ + "\n\n\n" #3 st_amt = "\n \n \n" + mfg + "," + name + "\n\n\n" fp_typ = StringIO(st_typ) #4 fp_amt = StringIO(st_amt) load_bulk_data.load_bottle_types(fp_typ) #5 assert db._check_bottle_type_exists(mfg, name) == True load_bulk_data.load_inventory(fp_amt) #6 assert db.check_inventory(mfg, name) == False
def test_get_liquor_amount_2():
db._reset_db()
db.add_bottle_type('Johnnie Walker', 'Black Label', 'blended scotch')
data = "Johnnie Walker,Black Label,1000 ml"
fp = StringIO(data) # make this look like a file handle
n = load_bulk_data.load_inventory(fp)
amount = db.get_liquor_amount('Johnnie Walker', 'Black Label')
assert amount == 1000.0, amount
def test_get_liquor_amount_5():
db._reset_db()
db.add_bottle_type('Johnnie Walker', 'Black Label', 'blended scotch')
data = "Johnnie Walker,Black Label,1000 ml\nJohnnie Walker,Black Label,50 gallons"
fp = StringIO(data) # make this look like a file handle
n = load_bulk_data.load_inventory(fp)
try:
amount = db.get_liquor_amount('Johnnie Walker', 'Black Label')
assert 0
except:
pass # we expect to see an error about gallons
def start_load_inventory(path):
db._reset_db()
start_load_bottle_types('test-data/bottle-types-data-1.txt', False)
test_data_path = path
fp = open(test_data_path, 'r')
try:
n = load_bulk_data.load_inventory(fp)
except db.LiquorMissing:
print "\nCould not find liquor: Incorrect string for \'load_bulk_data.load_inventory\' in \'%s\'." % test_data_path
print "Correct usage: \'manufacturer name,liquor name,amount ml\'\n"
pass
except ValueError:
print "\nIncorrect string format for \'load_bulk_data.load_inventory\' in \'%s\'." % test_data_path
print "Correct usage: \'manufacturer name,liquor name,liquor type\'\n"
pass
print "\n%s" % db._inventory_db
def load_bulk_data_inventory(self,file_name):
return load_bulk_data.load_inventory(file_name)
def rpc_inventory_add( self, item ):
data = item
fp = StringIO(data)
load_bulk_data.load_inventory(fp)
