def test_add_customer(self):
customer_1 = Customer(1, "James")
customer_2 = Customer(2, "John")
self.cd_rental.add_customer(customer_1)
self.cd_rental.add_customer(customer_2)
self.assertEqual(len(self.cd_rental.customers), 2)
def test_customer_id_not_found_2(self):
self.assertEqual(self.cd_rental.check_customer_id("001", 1),
"Try Again.")
customer_1 = Customer(1, "James")
self.cd_rental.add_customer(customer_1)
self.assertEqual(self.cd_rental.check_customer_id(1, 2), {})
def rental_contract_printed(step):
cd = Cd("Breaking Bad", "name dammit", False)
customer = Customer("Jones", 0)
cd_rental = Cd_rental()
cd_rental.add_customer(customer)
cd_rental.add_cd(cd)
cd_rental.check_out("Jones", "Breaking Bad")
assert_equal(cd_rental.print_contract("Jones", "Breaking Bad"),
(0, 'name dammit'))
def then_cd_recorded_as_rented(step):
cd = Cd("Fall", "name", False)
customer = Customer("James", 0)
cd_rental = Cd_rental()
cd_rental.add_customer(customer)
cd_rental.add_cd(cd)
cd_rental.check_out("James", "Fall")
assert_equal(cd_rental.get_rented_if("Fall"), True)
assert_equal(cd_rental.check_customer_id("James", 1), {'Fall': True})
def test_print_contract(self):
customer = Customer(1, "James")
cd = Cd("Breaking Bad", "Insert Disc", False)
self.cd_rental.add_customer(customer)
self.cd_rental.add_cd(cd)
self.cd_rental.check_out(1, "Breaking Bad")
self.assertEqual(self.cd_rental.print_contract(1, "Breaking Bad"),
("James", "Insert Disc"))
def test_record_rented(self):
customer = Customer(1, "James")
cd = Cd(1, "Insert Disc", False)
self.cd_rental.add_cd(cd)
self.cd_rental.add_customer(customer)
self.cd_rental.check_out(1, 1)
self.assertEqual(self.cd_rental.check_customer_id(1, 1), {1: True})
self.assertEqual(self.cd_rental.get_rented_if(001), True)
def test_violate_business_rules(self):
customer = Customer(1, "James")
customer2 = Customer(2, "John")
cd1 = Cd(1, "Insert Disc", False)
cd2 = Cd(2, "Insert Disc", False)
cd3 = Cd(3, "Insert Disc", False)
cd4 = Cd(4, "Insert Disc", False)
self.cd_rental.add_cd(cd1)
self.cd_rental.add_cd(cd2)
self.cd_rental.add_cd(cd3)
self.cd_rental.add_cd(cd4)
self.cd_rental.add_customer(customer)
self.cd_rental.add_customer(customer2)
self.cd_rental.check_out(1, 4)
self.cd_rental.check_out(1, 1)
self.cd_rental.check_out(1, 2)
self.assertEqual(self.cd_rental.check_out(1, 3),
"Violated CD Rental Limit business rule")
self.assertEqual(self.cd_rental.customers.get(2), {})
self.assertEqual(self.cd_rental.customers.get(23), None)
def test_check_customer_id(self):
customer_1 = Customer(1, "James")
self.cd_rental.add_customer(customer_1)
self.assertEqual(self.cd_rental.check_customer_id(1, 1), {})
def test_customer_object_can_be_created(self):
customer = Customer("001", "John")
self.assertEqual(customer.customer_id, "001")
self.assertEqual(customer.customer_name, "John")
def when_clerk_checks_out_cd(step):
cd = Cd("Arch Age", "some freaking name", False)
customer = Customer("Jane", 0)
cd_rental = Cd_rental()
cd_rental.add_customer(customer)
cd_rental.check_out(customer, cd)
def given_customer_has_id(step):
customer = Customer("Fame Game", {})
cd_rental = Cd_rental()
cd_rental.add_customer(customer)
assert_equal(cd_rental.check_customer_id("Fame Game", 1), {})