<?php

namespace Doctrine\ODM\MongoDB\Tests\Functional\Ticket;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\ODM\MongoDB\DocumentNotFoundException;
use Doctrine\ODM\MongoDB\Mapping\Annotations as ODM;

class GH602Test extends \Doctrine\ODM\MongoDB\Tests\BaseTest
{
    public function testReferenceManyOwningSidePreparesFilterCriteriaForDifferentClass()
    {
        $thingClass = __NAMESPACE__ . '\GH602Thing';
        $userClass = __NAMESPACE__ . '\GH602User';
        $this->enableDeletedFilter($thingClass);

        $user1 = new GH602User();
        $thing1 = new GH602Thing();
        $thing2 = new GH602Thing();

        $user1->likes->add($thing1);
        $user1->likes->add($thing2);
        $thing2->deleted = true;

        $this->dm->persist($user1);
        $this->dm->persist($thing1);
        $this->dm->persist($thing2);
        $this->dm->flush();
        $this->dm->clear();

        /* We cannot simply reinitialize the PersistentCollection, because its
         * $mongoData property has already been cleared and DocumentPersister
         * would be unable to issue a new query for related documents.
         */
        $user1 = $this->dm->find($userClass, $user1->getId());
        $user1likes = iterator_to_array($user1->likes, false);

        /* FilterCollection criteria will only be considered upon initialization
         * of the Proxy object, so expect an exception at that time. This is not
         * ideal, but it is the current behavior for hydrating the owning side
         * of a reference-many collection.
         */
        $this->assertCount(2, $user1likes);

        $this->assertInstanceOf('Doctrine\ODM\MongoDB\Proxy\Proxy', $user1likes[0]);
        $this->assertTrue($user1likes[0]->__isInitialized());
        $this->assertEquals($thing1->getId(), $user1likes[0]->getId());

        $this->assertInstanceOf('Doctrine\ODM\MongoDB\Proxy\Proxy', $user1likes[1]);
        $this->assertFalse($user1likes[1]->__isInitialized());
        $this->assertEquals($thing2->getId(), $user1likes[1]->getId());

        try {
            $user1likes[1]->__load();
            $this->fail('Expected DocumentNotFoundException for filtered Proxy object');
        } catch (DocumentNotFoundException $e) {
        }
    }

    public function testReferenceManyInverseSidePreparesFilterCriteriaForDifferentClass()
    {
        $thingClass = __NAMESPACE__ . '\GH602Thing';
        $userClass = __NAMESPACE__ . '\GH602User';
        $this->enableDeletedFilter($userClass);

        $user1 = new GH602User();
        $user2 = new GH602User();
        $thing1 = new GH602Thing();

        $user1->likes->add($thing1);
        $user2->likes->add($thing1);
        $user2->deleted = true;

        $this->dm->persist($user1);
        $this->dm->persist($user2);
        $this->dm->persist($thing1);
        $this->dm->flush();
        $this->dm->clear();

        /* We cannot simply reinitialize the PersistentCollection, because its
         * $mongoData property has already been cleared and DocumentPersister
         * would be unable to issue a new query for related documents.
         */
        $thing1 = $this->dm->find($thingClass, $thing1->getId());
        $thing1likedBy = iterator_to_array($thing1->likedBy, false);

        $this->assertCount(1, $thing1likedBy);
        $this->assertEquals($user1->getId(), $thing1likedBy[0]->getId());
    }

    private function enableDeletedFilter($class)
    {
        $this->dm->getFilterCollection()->enable('testFilter');
        $filter = $this->dm->getFilterCollection()->getFilter('testFilter');
        $filter->setParameter('class', $class);
        $filter->setParameter('field', 'deleted');
        $filter->setParameter('value', false);
    }
}

/** @ODM\Document */
class GH602User
{
    /** @ODM\Id */
    public $id;

    /** @ODM\Boolean(name="user_deleted") */
    public $deleted = false;

    /** @ODM\ReferenceMany(targetDocument="GH602Thing", inversedBy="likedBy", simple=true) */
    public $likes;

    public function __construct()
    {
        $this->likes = new ArrayCollection();
    }

    // Return the identifier without triggering Proxy initialization
    public function getId()
    {
        return $this->id;
    }
}

/** @ODM\Document */
class GH602Thing
{
    /** @ODM\Id */
    public $id;

    /** @ODM\Boolean(name="thing_deleted") */
    public $deleted = false;

    /** @ODM\ReferenceMany(targetDocument="GH602User", mappedBy="likes") */
    public $likedBy;

    public function __construct()
    {
        $this->likedBy = new ArrayCollection();
    }

    // Return the identifier without triggering Proxy initialization
    public function getId()
    {
        return $this->id;
    }
}