Пример #1
0
 /**
  * Test manipulation of owned sites collection.
  *
  * @covers Application\Sonata\UserBundle\Entity\User::addOwnedSite()
  * @covers Application\Sonata\UserBundle\Entity\User::removeOwnedSite()
  * @covers Application\Sonata\UserBundle\Entity\User::getOwnedSites()
  */
 public function testAddOwnedSites()
 {
     $ownedSites[0] = new Site();
     $ownedSites[1] = new Site();
     $ownedSites[2] = new Site();
     $this->assertTrue($this->object->getOwnedSites() instanceof Collection);
     $this->assertEquals(0, $this->object->getOwnedSites()->count());
     $this->object->addOwnedSite($ownedSites[0]);
     $this->assertEquals(1, $this->object->getOwnedSites()->count());
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[0]));
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[1]));
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[2]));
     $this->object->addOwnedSite($ownedSites[1]);
     $this->assertEquals(2, $this->object->getOwnedSites()->count());
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[0]));
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[1]));
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[2]));
     $this->object->removeOwnedSite($ownedSites[2]);
     $this->assertEquals(2, $this->object->getOwnedSites()->count());
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[0]));
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[1]));
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[2]));
     $this->object->removeOwnedSite($ownedSites[0]);
     $this->assertEquals(1, $this->object->getOwnedSites()->count());
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[0]));
     $this->assertTrue($this->object->getOwnedSites()->contains($ownedSites[1]));
     $this->assertFalse($this->object->getOwnedSites()->contains($ownedSites[2]));
 }