class ProductRepositoryTest(unittest.TestCase): def setUp(self): fireTruck = Product('f1234', 'Fire Truck', 'red', 8.95, 'medium') barbieClassic = Product('b7654', 'Barbie Classic', 'yellow', 15.95, 'small') toyConvertible = Product('p1122', 'Toy Porsche Convertible', 'red', 230, 'na') self.repository = ProductRepository() self.repository.add(fireTruck) self.repository.add(barbieClassic) self.repository.add(toyConvertible) def testFindByColor(self): foundProducts = self.repository.selectBy(ColorSpec('red')) self.assertEquals(2, len(foundProducts)) def testFindByColorSizeAndBelowPrice(self): specs = CompositeSpec() specs.add(ColorSpec('red')) specs.add(SizeSpec('small')) specs.add(BelowPriceSpec(10)) foundProducts = self.repository.selectBy(specs) self.assertEquals(0, len(foundProducts))
def setUp(self): fireTruck = Product('f1234', 'Fire Truck', 'red', 8.95, 'medium') barbieClassic = Product('b7654', 'Barbie Classic', 'yellow', 15.95, 'small') toyConvertible = Product('p1122', 'Toy Porsche Convertible', 'red', 230, 'na') self.repository = ProductRepository() self.repository.add(fireTruck) self.repository.add(barbieClassic) self.repository.add(toyConvertible)