iTop/test/core/ormLinkSetTest.php

<?php
// Copyright (c) 2010-2017 Combodo SARL
//
//   This file is part of iTop.
//
//   iTop is free software; you can redistribute it and/or modify
//   it under the terms of the GNU Affero General Public License as published by
//   the Free Software Foundation, either version 3 of the License, or
//   (at your option) any later version.
//
//   iTop is distributed in the hope that it will be useful,
//   but WITHOUT ANY WARRANTY; without even the implied warranty of
//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//   GNU Affero General Public License for more details.
//
//   You should have received a copy of the GNU Affero General Public License
//   along with iTop. If not, see <http://www.gnu.org/licenses/>
//


/**
 * Created by PhpStorm.
 * User: Eric
 * Date: 20/12/2017
 * Time: 11:56
 */


namespace Combodo\iTop\Test\UnitTest\Core;

use Combodo\iTop\Test\UnitTest\ItopDataTestCase;
use Exception;
use ormLinkSet;
use PHPUnit\Framework\TestCase;


/**
 * Tests of the ormLinkSet class using N-N links between FunctionalCI and Contact
 *
 * @runTestsInSeparateProcesses
 * @preserveGlobalState disabled
 * @backupGlobals disabled
 */
class ormLinkSetTest extends ItopDataTestCase
{

    /**
     * @throws Exception
     */
    protected function setUp()
    {
        parent::setUp();
    }

    /**
     *
     */
    public function testConstruct()
    {
        $oOrmLink = new ormLinkSet('UserRequest', 'contacts_list');
        $this->assertNotNull($oOrmLink);
    }

    /**
     *
     */
    public function testConstruct2()
    {
        $this->expectException('Exception');

        new ormLinkSet('UserRequest', 'ref');
    }

    /**
     * @throws Exception
     */
    public function testBasic()
    {
        $oServer = $this->CreateServer(1);
        $aPersons = array();
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $aPersons[] = $oPerson;
            $this->AddContactToCI($oPerson, $oServer);
        }
        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());
    }

    /**
     * @throws Exception
     */
    public function testSuccesiveAdds()
    {
        $oServer = $this->CreateServer(1);
        $aPersons = array();
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $aPersons[] = $oPerson;
            $this->AddContactToCI($oPerson, $oServer);
        }
        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());

        $this->AddContactToCI($this->CreatePerson($i), $oServer);
        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(4, $oContactsSet->Count());

        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(4, $oContactsSet->Count());
    }

    /**
     * @throws Exception
     */
    public function testRemove()
    {
        $oServer = $this->CreateServer(1);
        $aPersons = array();
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $aPersons[] = $oPerson;
            $this->AddContactToCI($oPerson, $oServer);
        }
        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());

        for ($i = 0; $i < 3; $i++)
        {
            $this->RemoveContactFromCI($aPersons[$i], $oServer);
        }
        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(0, $oContactsSet->Count());

        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(0, $oContactsSet->Count());
    }

    /**
     * @throws Exception
     */
    public function testAddThenRemove()
    {
        $oServer = $this->CreateServer(1);
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $this->AddContactToCI($oPerson, $oServer);
            $this->RemoveContactFromCI($oPerson, $oServer);
        }
        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(0, $oContactsSet->Count());

        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(0, $oContactsSet->Count());
    }

    /**
     * @throws Exception
     */
    public function testRemoveThenAdd()
    {
        $oServer = $this->CreateServer(1);
        $aPersons = array();
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $aPersons[] = $oPerson;
            $this->AddContactToCI($oPerson, $oServer);
        }
        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());

        for ($i = 0; $i < 3; $i++)
        {
            $this->RemoveContactFromCI($aPersons[$i], $oServer);
            $this->AddContactToCI($aPersons[$i], $oServer);
        }
        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());

        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());
    }

    /**
     * @throws Exception
     */
    public function testAddDuplicate()
    {
        $oServer = $this->CreateServer(1);
        $aPersons = array();
        for ($i = 0; $i < 3; $i++)
        {
            $oPerson = $this->CreatePerson($i);
            $aPersons[] = $oPerson;
            $this->AddContactToCI($oPerson, $oServer);
        }
        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());

        for ($i = 0; $i < 3; $i++)
        {
            $this->AddContactToCI($aPersons[$i], $oServer);
        }
        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(6, $oContactsSet->Count());

        $oServer->DBUpdate();
        $this->ReloadObject($oServer);

        $oContactsSet = $oServer->Get('contacts_list');
        $this->assertEquals(3, $oContactsSet->Count());
    }
}