public function testCurtainColoursOnOffer() { $curtainDesign = new CurtainDesign(); $curtainColour1 = new CurtainColour(); $curtainColour1->setName("two"); $curtainColour1->setinStock(1); $curtainColour1->setOnOffer(1); $curtainColour1->setDisplay(1); $curtainDesign->addCurtainColour($curtainColour1); $curtainColour3 = new CurtainColour(); $curtainColour3->setName("three"); $curtainColour3->setinStock(1); $curtainColour3->setOnOffer(0); $curtainDesign->addCurtainColour($curtainColour3); $curtainColour2 = new CurtainColour(); $curtainColour2->setName("one"); $curtainColour2->setinStock(0); $curtainColour2->setOnOffer(1); $curtainColour2->setDisplay(1); $curtainDesign->addCurtainColour($curtainColour2); $curtainColour4 = new CurtainColour(); $curtainColour4->setName("four"); $curtainColour4->setinStock(1); $curtainColour4->setOnOffer(1); $curtainColour4->setDisplay(0); $curtainDesign->addCurtainColour($curtainColour4); $curtainColour4 = new CurtainColour(); $curtainColour4->setName("five"); $curtainColour4->setinStock(1); $curtainColour4->setOnOffer(1); $curtainColour4->setDisplay(1); $curtainDesign->addCurtainColour($curtainColour4); $coloursOnOffer = $curtainDesign->getCurtainColoursOnOffer(); $this->assertTrue(count($coloursOnOffer) == 2); $this->assertEquals("two", current($coloursOnOffer)->getName()); $this->assertEquals("five", next($coloursOnOffer)->getName()); }