<?php

namespace Behat\Mink\Tests;

use Behat\Mink\Mink;

class MinkTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @var Mink
     */
    private $mink;

    protected function setUp()
    {
        $this->mink = new Mink();
    }

    public function testRegisterSession()
    {
        $session = $this->getSessionMock();

        $this->assertFalse($this->mink->hasSession('not_registered'));
        $this->assertFalse($this->mink->hasSession('js'));
        $this->assertFalse($this->mink->hasSession('my'));

        $this->mink->registerSession('my', $session);

        $this->assertTrue($this->mink->hasSession('my'));
        $this->assertFalse($this->mink->hasSession('not_registered'));
        $this->assertFalse($this->mink->hasSession('js'));
    }

    public function testRegisterSessionThroughConstructor()
    {
        $mink = new Mink(array('my' => $this->getSessionMock()));

        $this->assertTrue($mink->hasSession('my'));
    }

    public function testSessionAutostop()
    {
        $session1 = $this->getSessionMock();
        $session2 = $this->getSessionMock();
        $this->mink->registerSession('my1', $session1);
        $this->mink->registerSession('my2', $session2);

        $session1
            ->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(true));
        $session1
            ->expects($this->once())
            ->method('stop');
        $session2
            ->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(false));
        $session2
            ->expects($this->never())
            ->method('stop');

        unset($this->mink);
    }

    public function testNotStartedSession()
    {
        $session = $this->getSessionMock();

        $session
            ->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(false));
        $session
            ->expects($this->once())
            ->method('start');

        $this->mink->registerSession('mock_session', $session);
        $this->assertSame($session, $this->mink->getSession('mock_session'));

        $this->setExpectedException('InvalidArgumentException');

        $this->mink->getSession('not_registered');
    }

    public function testGetAlreadyStartedSession()
    {
        $session = $this->getSessionMock();

        $session
            ->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(true));
        $session
            ->expects($this->never())
            ->method('start');

        $this->mink->registerSession('mock_session', $session);
        $this->assertSame($session, $this->mink->getSession('mock_session'));
    }

    public function testSetDefaultSessionName()
    {
        $this->assertNull($this->mink->getDefaultSessionName());

        $session = $this->getSessionMock();
        $this->mink->registerSession('session_name', $session);
        $this->mink->setDefaultSessionName('session_name');

        $this->assertEquals('session_name', $this->mink->getDefaultSessionName());

        $this->setExpectedException('InvalidArgumentException');

        $this->mink->setDefaultSessionName('not_registered');
    }

    public function testGetDefaultSession()
    {
        $session1 = $this->getSessionMock();
        $session2 = $this->getSessionMock();

        $this->assertNotSame($session1, $session2);

        $this->mink->registerSession('session_1', $session1);
        $this->mink->registerSession('session_2', $session2);
        $this->mink->setDefaultSessionName('session_2');

        $this->assertSame($session1, $this->mink->getSession('session_1'));
        $this->assertSame($session2, $this->mink->getSession('session_2'));
        $this->assertSame($session2, $this->mink->getSession());

        $this->mink->setDefaultSessionName('session_1');

        $this->assertSame($session1, $this->mink->getSession());
    }

    public function testGetNoDefaultSession()
    {
        $session1 = $this->getSessionMock();

        $this->mink->registerSession('session_1', $session1);

        $this->setExpectedException('InvalidArgumentException');

        $this->mink->getSession();
    }

    public function testIsSessionStarted()
    {
        $session_1 = $this->getSessionMock();
        $session_2 = $this->getSessionMock();

        $session_1
            ->expects($this->any())
            ->method('isStarted')
            ->will($this->returnValue(false));
        $session_1
            ->expects($this->never())
            ->method('start');

        $session_2
            ->expects($this->any())
            ->method('isStarted')
            ->will($this->returnValue(true));
        $session_2
            ->expects($this->never())
            ->method('start');

        $this->mink->registerSession('not_started', $session_1);
        $this->assertFalse($this->mink->isSessionStarted('not_started'));

        $this->mink->registerSession('started', $session_2);
        $this->assertTrue($this->mink->isSessionStarted('started'));

        $this->setExpectedException('InvalidArgumentException');

        $this->mink->getSession('not_registered');
    }

    public function testAssertSession()
    {
        $session = $this->getSessionMock();

        $this->mink->registerSession('my', $session);

        $this->assertInstanceOf('Behat\Mink\WebAssert', $this->mink->assertSession('my'));
    }

    public function testAssertSessionNotRegistered()
    {
        $session = $this->getSessionMock();

        $this->assertInstanceOf('Behat\Mink\WebAssert', $this->mink->assertSession($session));
    }

    public function testResetSessions()
    {
        $session1 = $this->getSessionMock();
        $session1->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(false));
        $session1->expects($this->never())
            ->method('reset');

        $session2 = $this->getSessionMock();
        $session2->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(true));
        $session2->expects($this->once())
            ->method('reset');

        $this->mink->registerSession('not started', $session1);
        $this->mink->registerSession('started', $session2);

        $this->mink->resetSessions();
    }

    public function testRestartSessions()
    {
        $session1 = $this->getSessionMock();
        $session1->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(false));
        $session1->expects($this->never())
            ->method('restart');

        $session2 = $this->getSessionMock();
        $session2->expects($this->once())
            ->method('isStarted')
            ->will($this->returnValue(true));
        $session2->expects($this->once())
            ->method('restart');

        $this->mink->registerSession('not started', $session1);
        $this->mink->registerSession('started', $session2);

        $this->mink->restartSessions();
    }

    private function getSessionMock()
    {
        return $this->getMockBuilder('Behat\Mink\Session')
            ->disableOriginalConstructor()
            ->getMock();
    }
}