def test_cancel(self):
week, year = this_week_year()
d = dinner(week,year)
assert d is not None
# now cancel
cancel_msg = "durumbar"
skip_week(week, year, cancel_msg)
before = dinner(week-1,year)
d2 = dinner(week,year)
assert d.house.name != d2.house.name
assert d2.house.name == cancel_msg
# make sure we cant double cancel
skip_week(week, year, cancel_msg)
d2 = dinner(week,year)
assert d.house.name != d2.house.name
assert d2.house.name == cancel_msg
before2 = dinner(week-1,year)
# make sure that stuff from earlier wasn't affected
assert before.house.name == before2.house.name
db.query(Cancellation).delete()
db.commit()
def test_swap(self):
week1,year1 = 16,2016
week2, year2 = 17,2016
week3, year3 = 18, 2016
d1 = dinner(week1,year1)
d2 = dinner(week2,year2)
d3 = dinner(week3,year3)
#Test valid input generally
assert d1 is not None
assert d2 is not None
assert d1 is not d2
#commit a change to db
swap_houses_to_db(week1,year1, week2,year2)
d1a = dinner(week1,year1)
d2a = dinner(week2,year2)
d3a = dinner(week3, year3)
#test that house swap has worked
assert d1a.house.name == d2.house.name
assert d2a.house.name == d1.house.name
assert d3a.house.name == d3.house.name
#Test what happens when swap is across years
week4, year4 = 52, 2017
week5, year5 = 1, 2018
week6, year6 = 2, 2018
d4 = dinner(week4,year4)
d5 = dinner(week5, year5)
d6 = dinner(week6, year6)
swap_houses_to_db(week4, year4, week5, year5)
d4a = dinner(week4, year4)
d5a = dinner(week5, year5)
d6a = dinner(week6, year6)
#Test that it worked
assert d4a.house.name == d5.house.name
assert d5a.house.name == d4.house.name
assert d6a.house.name == d6.house.name
db.query(Swap).delete()
db.commit()
