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()