class TestGuest(unittest.TestCase):
def setUp(self):
self.guest_1 = Guest("Pierre", 50, "Gotta Go")
self.guest_2 = Guest("Alexander", 40, "On My Radio")
self.guest_3 = Guest("Pepe", 30, "Divorce a I'ltalienne")
self.guest_4 = Guest("Edi", 5, "Gotta Go")
self.room_1 = Room("Bongo", 3, 20)
self.room_2 = Room("Studio 24", 5, 30)
self.song_1 = Song("Gotta Go", "Agnostic Front", 3.2)
self.song_2 = Song("On My Radio", "Selecter", 3.52)
self.song_3 = Song("Divorce a I'ltalienne", "Mungo's Hifi", 3.46)
def test_guest_has_name(self):
self.assertEqual("Pierre", self.guest_1.name)
def test_guest_has_money(self):
self.assertEqual(30, self.guest_3.cash)
def test_guest_paying_entry_decreases_cash(self):
self.guest_1.pay_entry_fee(self.guest_1, self.room_1)
self.assertEqual(30, self.guest_1.cash)
def test_if_guest_fav_song_is_in_room(self):
self.room_1.add_song_to_list(self.song_1, self.room_1)
self.room_1.add_song_to_list(self.song_2, self.room_1)
self.assertEqual(
"Whoo!", self.guest_1.fav_song_in_room(self.guest_1, self.room_1))
def test_guest_can_add_to_tab(self):
guest_1 = Guest("Ted Mosby", 35.00, "Saturday night", "Disco Room")
fee = Room("Disco Room", 15.00)
order = Menu("Beer", 3.50)
guest_1.pay_entry_fee(fee)
guest_1.buy_from_menu(order)
self.assertEqual(18.50, guest_1.tab)
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("Ted Mosby", 25.00, "Saturday night", "Disco Room")
def test_guest_has_name(self):
self.assertEqual("Ted Mosby", self.guest.name)
def test_guest_has_cash(self):
self.assertEqual(25.00, self.guest.wallet)
def test_guest_can_pay_entry(self):
fee = Room("Disco Room", 15.00)
self.guest.pay_entry_fee(fee)
self.assertEqual(10.00, self.guest.wallet)
def test_guest_has_fave_song(self):
self.assertEqual("Saturday night", self.guest.song)
def test_guest_can_buy_from_bar(self):
order = Menu("Beer", 3.00)
self.guest.buy_from_menu(order)
self.assertEqual(22.00, self.guest.wallet)
def test_let_guest_know_they_are_skint(self):
guest_1 = Guest("Marshall Eriksson", 30.00,
"Night fever", "Disco Room")
fee = Room("Disco Room", 15.00)
self.guest.pay_entry_fee(fee)
order_1 = Menu("Beer", 3.00)
order_2 = Menu("Beer", 3.00)
order_3 = Menu("Beer", 3.00)
order_4 = Menu("Cocktail", 6.00)
guest_1.pay_entry_fee(fee)
guest_1.buy_from_menu(order_1)
guest_1.buy_from_menu(order_2)
guest_1.buy_from_menu(order_3)
guest_1.buy_from_menu(order_4)
self.assertEqual("Skint!", self.guest.is_guest_skint(guest_1))
def test_let_guest_know_they_still_have_cash(self):
guest_1 = Guest("Robin Scherbatsky", 50.00,
"Heart of glass", "Disco Room")
fee = Room("Disco Room", 15.00)
order_1 = Menu("Beer", 3.00)
order_2 = Menu("Beer", 3.00)
order_3 = Menu("Scotch", 5.00)
order_4 = Menu("Cocktail", 6.00)
guest_1.pay_entry_fee(fee)
guest_1.buy_from_menu(order_1)
guest_1.buy_from_menu(order_2)
guest_1.buy_from_menu(order_3)
guest_1.buy_from_menu(order_4)
self.assertEqual("Still loaded!", self.guest.is_guest_skint(guest_1))
def test_let_guest_know_they_still_have_cash(self):
guest_1 = Guest("Robin Scherbatsky", 50.00,
"Heart of glass", "Disco Room")
fee = Room("Disco Room", 15.00)
order_1 = Menu("Beer", 3.00)
order_2 = Menu("Beer", 3.00)
order_3 = Menu("Scotch", 5.00)
order_4 = Menu("Cocktail", 6.00)
guest_1.pay_entry_fee(fee)
guest_1.buy_from_menu(order_1)
guest_1.buy_from_menu(order_2)
guest_1.buy_from_menu(order_3)
guest_1.buy_from_menu(order_4)
self.assertEqual("Still loaded!", self.guest.is_guest_skint(guest_1))
def test_let_guest_know_they_are_skint(self):
guest_1 = Guest("Marshall Eriksson", 30.00,
"Night fever", "Disco Room")
fee = Room("Disco Room", 15.00)
self.guest.pay_entry_fee(fee)
order_1 = Menu("Beer", 3.00)
order_2 = Menu("Beer", 3.00)
order_3 = Menu("Beer", 3.00)
order_4 = Menu("Cocktail", 6.00)
guest_1.pay_entry_fee(fee)
guest_1.buy_from_menu(order_1)
guest_1.buy_from_menu(order_2)
guest_1.buy_from_menu(order_3)
guest_1.buy_from_menu(order_4)
self.assertEqual("Skint!", self.guest.is_guest_skint(guest_1))
class TestGuest(unittest.TestCase):
def setUp(self):
self.room = Room("Super Stars", 100.00, 2, 5.00)
self.guest = Guest("Joey", 50.00, "Don't Stop Believin'")
def test_guest_has_name(self):
self.assertEqual("Joey", self.guest.name)
def test_guest_has_money(self):
self.assertEqual(50.00, self.guest.wallet)
def test_guest_has_favourite_Song(self):
self.assertEqual("Don't Stop Believin'", self.guest.favourite_song)
def test_can_pay_entry_fee_sufficient_funds(self):
value = self.room.entry_fee
self.guest.pay_entry_fee(value)
self.assertEqual(45.00, self.guest.wallet)
class TestGuest(unittest.TestCase):
def setUp(self):
self.song = Song("Castle Theme")
self.guest = Guest("Mario", self.song, 100.00)
self.room = Room("Nintendo")
def test_guest_has_name(self):
self.assertEqual("Mario", self.guest.name)
def test_guest_has_favourite_song(self):
self.assertEqual(self.song, self.guest.favourite_song)
def test_guest_has_wallet(self):
self.assertEqual(100.00, self.guest.wallet)
def test_pay_entry_fee_removes_money_from_wallet(self):
self.guest.pay_entry_fee(self.room)
self.assertEqual(90.00, self.guest.wallet)
def test_guest_can_cheer(self):
self.assertEqual("Woop!", self.guest.cheer())
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("Patrick", 20, "Imagine")
self.room = Room(5, 10)
self.room_2 = Room(5, 30)
def test_guest_has_name(self):
self.assertEqual("Patrick", self.guest.name)
def test_guest_has_wallet(self):
self.assertEqual(20, self.guest.wallet)
def test_guest_has_favourite_song(self):
self.assertEqual(20, self.guest.wallet)
def test_guest_can_pay_entry_fee(self):
self.guest.pay_entry_fee(self.room)
self.assertEqual(10, self.guest.wallet)
def test_guest_cannot_pay_if_no_money(self):
self.assertEqual(False, self.guest.pay_entry_fee(self.room_2))
self.assertEqual(20, self.guest.wallet)