diff --git a/python/tests/test_gilded_rose.py b/python/tests/test_gilded_rose.py index 761686b0..b69951c0 100644 --- a/python/tests/test_gilded_rose.py +++ b/python/tests/test_gilded_rose.py @@ -11,34 +11,40 @@ class GildedRoseTest(unittest.TestCase): self.assertEqual(item.sell_in, sell_in) self.assertEqual(item.quality, quality) + def update_gilded_rose_days(self, gilded_rose, days): + for day in range(days): + gilded_rose.update_quality() + def generate_and_update_gilded_rose(self, items, days): gilded_rose = GildedRose(items) - for day in range(days): - gilded_rose.update_gilded_rose() + self.update_gilded_rose_days(gilded_rose, days) return gilded_rose def test_foo_1day(self): items= [Item('foo', 1, 1)] - gilded_rose = GildedRose(items) - gilded_rose.update_quality() + self.generate_and_update_gilded_rose(items, 1) self.check_item_values(items[0], 'foo', 0, 0) def test_foo_nonnegative(self): items = [Item("foo", 0, 0)] - gilded_rose = GildedRose(items) - gilded_rose.update_quality() + self.generate_and_update_gilded_rose(items, 1) self.check_item_values(items[0], "foo", 0, 0) def test_multiple_items(self): items = [Item("foo", 0, 0), Item("bar", 1, 1), Item("baz", 2, 2)] - gilded_rose = GildedRose(items) - gilded_rose.update_quality() + self.generate_and_update_gilded_rose(items, 1) self.check_item_values(items[0], "foo", 0, 0) self.check_item_values(items[1], "bar", 0, 0) self.check_item_values(items[2], "baz", 1, 1) + def test_multiple_days(self): + items= [Item('foo', 2, 3)] + self.generate_and_update_gilded_rose(items, 2) + self.check_item_values(items[0], 'foo', 0, 1) + + if __name__ == '__main__':