public function testUnitPrices() { $price = $this->getMock('JLM\\ProductBundle\\Model\\ProductPriceInterface'); $this->assertCount(0, $this->entity->getUnitPrices()); $this->assertFalse($this->entity->removeUnitPrice($price)); $this->assertCount(0, $this->entity->getUnitPrices()); $this->assertTrue($this->entity->addUnitPrice($price)); $this->assertCount(1, $this->entity->getUnitPrices()); $this->assertTrue($this->entity->removeUnitPrice($price)); $this->assertCount(0, $this->entity->getUnitPrices()); }