<?php

namespace Doctrine\MongoDB\Tests\Event;

use Doctrine\MongoDB\Event\UpdateEventArgs;
use Doctrine\MongoDB\Tests\TestCase;

class UpdateEventArgsTest extends TestCase
{
    public function testUpdateEventArgs()
    {
        $invoker = new \stdClass();
        $query = ['x' => 1];
        $newObj = ['$set' => ['x' => 2]];
        $options = ['upsert' => true];

        $updateEventArgs = new UpdateEventArgs($invoker, $query, $newObj, $options);

        $this->assertSame($invoker, $updateEventArgs->getInvoker());
        $this->assertSame($query, $updateEventArgs->getQuery());
        $this->assertSame($newObj, $updateEventArgs->getNewObj());
        $this->assertSame($options, $updateEventArgs->getOptions());

        // Ensure the setters work.
        $query2 = ['x' => 2];
        $newObj2 = ['$set' => ['x' => 2]];
        $options2 = ['upsert' => false];

        $updateEventArgs->setQuery($query2);
        $updateEventArgs->setNewObj($newObj2);
        $updateEventArgs->setOptions($options2);

        $this->assertSame($query2, $updateEventArgs->getQuery());
        $this->assertSame($newObj2, $updateEventArgs->getNewObj());
        $this->assertSame($options2, $updateEventArgs->getOptions());
    }
}