<?php
/*
 * Copyright 2005-2019 Centreon
 * Centreon is developed by : Julien Mathis and Romain Le Merlus under
 * GPL Licence 2.0.
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation ; either version 2 of the License.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, see <http://www.gnu.org/licenses>.
 *
 * Linking this program statically or dynamically with other modules is making a
 * combined work based on this program. Thus, the terms and conditions of the GNU
 * General Public License cover the whole combination.
 *
 * As a special exception, the copyright holders of this program give Centreon
 * permission to link this program with independent modules to produce an executable,
 * regardless of the license terms of these independent modules, and to copy and
 * distribute the resulting executable under terms of Centreon choice, provided that
 * Centreon also meet, for each linked independent module, the terms  and conditions
 * of the license of that module. An independent module is a module which is not
 * derived from this program. If you modify this program, you may extend this
 * exception to your version of the program, but you are not obliged to do so. If you
 * do not wish to do so, delete this exception statement from your version.
 *
 * For more information : contact@centreon.com
 *
 *
 */

namespace Centreon\Tests\Domain\Repository\Traits;

use PHPUnit\Framework\TestCase;
use Centreon\Test\Mock\CentreonDB;
use Centreon\Tests\Resources\Mock\RepositoryMock;
use Centreon\Test\Mock\CentreonDBManagerService;
use Centreon\Domain\Repository\Traits\CheckListOfIdsTrait;

class CheckListOfIdsTraitTest extends TestCase
{
    public function testCheckListOfIdsTrait(): void
    {
        $ids = [1, 3, 4];

        $db = new CentreonDB();
        $manager = new CentreonDBManagerService();
        $trait = new class ($db, $manager) extends RepositoryMock {
            use checkListOfIdsTrait;

            /**
             * @param int[] $ids
             * @param string|null $tableName
             * @param string|null $columnNameOfIdentificator
             * @return boolean
             */
            public function checkListOfIds(
                array $ids,
                string $tableName = null,
                string $columnNameOfIdentificator = null
            ): bool {
                return $this->checkListOfIdsTrait($ids, $tableName, $columnNameOfIdentificator);
            }
        };

        $db->resetResultSet();
        $db
            ->addResultSet(
                "SELECT COUNT(*) AS `total` FROM `mock_table` "
                . "WHERE `id_column` = :id0 OR `id_column` = :id1 LIMIT 0, 1",
                [
                    [
                        'total' => 3,
                    ],
                ]
            )
            ->addResultSet(
                "SELECT COUNT(*) AS `total` FROM `mock_table` "
                . "WHERE `id_column` = :id0 OR `id_column` = :id1 OR `id_column` = :id2 LIMIT 0, 1",
                [
                    [
                        'total' => 3,
                    ],
                ]
            );

        $this->assertTrue($trait->checkListOfIds($ids));

        // remove one ID from the list
        array_pop($ids);

        $this->assertFalse($trait->checkListOfIds($ids));
    }
}
