class TestRegister(TestCase):
def setUp(self):
self.reg = Register()
def test_initial_register_state(self):
results = 0.00
self.assertEqual(results, self.reg.get_total())
def test_scan_item_ground_beef(self):
results = 3.99
self.assertEqual(results, self.reg.scan_item("Ground Beef", 1))
def test_scan_item_milk(self):
results = 4.99
self.assertEqual(results, self.reg.scan_item("Milk", 1))
def test_scan_unknown_item(self):
results = 0.00
self.assertEqual(results, self.reg.scan_item("Mystery Goo", 10))
def test_scan_multiple_items(self):
self.assertEqual(4.99, self.reg.scan_item("Milk", 1))
self.assertEqual(3.99, self.reg.scan_item("Ground Beef", 1))
def test_get_initial_total(self):
results = 0.00
self.assertEqual(results, self.reg.get_total())
def test_get_total_with_item(self):
self.reg.scan_item("Milk", 1)
results = 4.99
self.assertEqual(results, self.reg.get_total())
def test_scan_and_total_multiple_items(self):
self.assertEqual(4.99, self.reg.scan_item("Milk", 1))
self.assertEqual(3.99, self.reg.scan_item("Ground Beef", 1))
self.assertEqual(8.98, self.reg.get_total())
def test_gum_2_for_1(self):
self.reg.scan_item("Gum", 2)
results = .25
self.assertEqual(results, self.reg.get_total())
def test_void_an_item(self):
self.reg.scan_item("Milk", 1)
self.reg.void_item("Milk", 1)
result = 0.00
self.assertEqual(result, self.reg.get_total())
def test_void_an_item_with_one_leftover(self):
self.reg.scan_item("Milk", 2)
self.reg.void_item("Milk", 1)
results = 4.99
self.assertEqual(results, self.reg.get_total())
def test_total_is_calculated_right_after_removing_special_item1(self):
self.reg.scan_item("Gum", 3)
self.reg.void_item("Gum", 1)
results = .25
self.assertEqual(results, self.reg.get_total())
def test_total_is_calculated_right_after_removing_special_item2(self):
self.reg.scan_item("Gum", 3)
self.reg.void_item("Gum", 2)
results = .25
self.assertEqual(results, self.reg.get_total())
def test_many_item_multiple_discounts(self):
self.reg.scan_item("Gum", 2) # .25
self.reg.scan_item("Milk", 2) # 7.00
results = 7.25
self.assertEqual(results, self.reg.get_total())
def test_weighted_item_n_for_m(self):
results = 6.69
self.reg.scan_item("Ground Beef", 1.0) # 3.99
self.reg.scan_item("Chicken", 1.0) # 2.70
self.assertEqual(self.reg.get_total(), results) # test from discount engine
