def test_tank_refill_successfully(session):
init_tanks(session)
beans_tank = Tank.query.filter(Tank.name == 'Beans Tank').first()
water_tank = Tank.query.filter(Tank.name == 'Water Tank').first()
milk_tank = Tank.query.filter(Tank.name == 'Milk Tank').first()
grounds_tank = Tank.query.filter(Tank.name == 'Grounds Tank').first()
assert refill("Water Tank", 1000) == "Water Tank refilled successfuly!"
assert water_tank.current_amount == 1000
assert refill("Milk Tank", 500) == "Milk Tank refilled successfuly!"
assert milk_tank.current_amount == 500
assert refill("Beans Tank", 500) == "Beans Tank refilled successfuly!"
assert beans_tank.current_amount == 500
assert refill("Grounds Tank", 0) == "Grounds Tank refilled successfuly!"
assert grounds_tank.current_amount == 0
def test_no_need_for_refill(session):
init_tanks(session)
grounds_tank = Tank.query.filter(Tank.name == 'Grounds Tank').first()
grounds_tank.current_amount = 0
grounds_tank_start_lvl = grounds_tank.current_amount
assert refill("Grounds Tank", 0) == \
'Grounds level at minimum - no need for refill'
assert grounds_tank.current_amount == grounds_tank_start_lvl
def test_negative_value(session):
init_tanks(session)
beans_tank = Tank.query.filter(Tank.name == 'Beans Tank').first()
beans_tank_start_lvl = beans_tank.current_amount
water_tank = Tank.query.filter(Tank.name == 'Water Tank').first()
water_tank_start_lvl = water_tank.current_amount
milk_tank = Tank.query.filter(Tank.name == 'Milk Tank').first()
milk_tank_start_lvl = milk_tank.current_amount
grounds_tank = Tank.query.filter(Tank.name == 'Grounds Tank').first()
grounds_tank_start_lvl = grounds_tank.current_amount
assert refill("Water Tank", -1) == 'Negative value is forbidden'
assert water_tank.current_amount == water_tank_start_lvl
assert refill("Milk Tank", -1) == 'Negative value is forbidden'
assert milk_tank.current_amount == milk_tank_start_lvl
assert refill("Beans Tank", -1) == 'Negative value is forbidden'
assert beans_tank.current_amount == beans_tank_start_lvl
assert refill("Grounds Tank", -1) == 'Negative value is forbidden'
assert grounds_tank.current_amount == grounds_tank_start_lvl
def service():
form = ServiceForm()
if form.validate_on_submit():
for field in form:
if field.name is not "submit":
try:
refill_amount = int(field.data)
tank_type = field.label.text
result = refill(tank_type, refill_amount)
flash(f'{result}', 'info')
except ValueError:
pass
# because 0 is not accepted in wtforms as number but False/None
except TypeError:
if type(field.data) is None:
result = refill(field.label.text, 0)
flash(f'{result}', 'info')
return redirect(url_for('main.service'))
tanks = Tank.query.all()
return render_template(
'service.html', title='Service menu', form=form, tanks=tanks)
def test_tank_overfill(session):
init_tanks(session)
beans_tank = Tank.query.filter(Tank.name == 'Beans Tank').first()
beans_tank_start_lvl = beans_tank.current_amount
water_tank = Tank.query.filter(Tank.name == 'Water Tank').first()
water_tank_start_lvl = water_tank.current_amount
milk_tank = Tank.query.filter(Tank.name == 'Milk Tank').first()
milk_tank_start_lvl = milk_tank.current_amount
grounds_tank = Tank.query.filter(Tank.name == 'Grounds Tank').first()
grounds_tank_start_lvl = grounds_tank.current_amount
assert refill("Water Tank", 1200) == \
"You are trying to overfill the Water Tank. Nice try."
assert refill("Beans Tank", 700) == \
"You are trying to overfill the Beans Tank. Nice try."
assert refill("Milk Tank", 501) == \
"You are trying to overfill the Milk Tank. Nice try."
assert refill("Grounds Tank", 200) == "You are trying to add more Grounds."
# check if refill function dont change data
assert beans_tank.current_amount == beans_tank_start_lvl
assert water_tank.current_amount == water_tank_start_lvl
assert grounds_tank.current_amount == grounds_tank_start_lvl
assert milk_tank.current_amount == milk_tank_start_lvl