Kunstmaan / KunstmaanBundlesCMS

        $this->assertEquals($handler, current($handlers));

    }

    public function testPrepareMediaWithDefaultHandler()

    {

        $media = new Media();

        $this->defaultHandler

            ->expects($this->any())

            ->method('prepareMedia')

            ->with($this->equalTo($media));

        $this->object->prepareMedia($media);

    }

    public function testPrepareMediaWithCustomHandler()

    {

        $this->object->saveMedia($media);

    }

    public function testRemoveMediaWithDefaultHandler()

    {

        $media = new Media();

        $this->defaultHandler

            ->expects($this->once())

            ->method('removeMedia')

            ->with($this->equalTo($media));

        $this->object->removeMedia($media);

    }

    public function testRemoveMediaWithCustomHandler()

    {