<?php
/**
 * SeekQuarry/Yioop --
 * Open Source Pure PHP Search Engine, Crawler, and Indexer
 *
 * Copyright (C) 2009 - 2026  Chris Pollett chris@pollett.org
 *
 * LICENSE:
 *
 * 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 3 of the License, or
 * (at your option) any later version.
 *
 * 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 <https://www.gnu.org/licenses/>.
 *
 * END LICENSE
 *
 * This file contains unit tests of the BPlusTree class
 *
 * @author Chris Pollett chris@pollett.org
 * @license https://www.gnu.org/licenses/ GPL3
 * @link https://www.seekquarry.com/
 * @copyright 2009 - 2026
 * @filesource
 */
namespace seekquarry\yioop\tests;

use seekquarry\yioop\configs as C;
use seekquarry\yioop\library as L;
use seekquarry\yioop\models\Model;
use seekquarry\yioop\library\data_structures\BPlusTree;
use seekquarry\yioop\library\data_structures\PackedTableTools;
use seekquarry\yioop\library\UnitTest;
use seekquarry\yioop\library\data_structures as LDS;

/**
 * Yioop B+-tree Unit Class
 *
 * Used to test insert, lookup, and deletion of key-value pairs on the B+-Tree.
 * @author Chris Pollett
 */
class BPlusTreeTest extends UnitTest
{
    /**
     * Test directory to hold btree used for these unit tests
     */
    const TEST_DIR = '/test_files/b_plus_tree_test';
    /**
     * Folder names to use for test b-trees
     * @var array
     */
    public $table_dirs;
    /**
     * Minimum degree is set to 2 and the number of key-value pairs is set to 25
     */
    public function setUp()
    {
        $this->table_dirs = [];
    }
    /**
     * Delete the B+-Tree files created during the test
     */
    public function tearDown()
    {
        $model = new Model();
        foreach ($this->table_dirs as $table_dir) {
            $model->db->unlinkRecursive($table_dir);
        }
        $this->table_dirs = [];
    }
    /**
     * Creates a single B+ tree under a per-degree test directory.
     * Each call tracks the directory so tearDown can clean it up.
     *
     * @param int $max_degree maximum branching factor of the tree
     * @param array $format column definitions for the tree; defaults
     *      to a variable-length string primary key and a BLOB value
     *      column
     * @return LDS\BPlusTree freshly-constructed tree backed
     * by the
     *      per-test directory, using non-compressing leaf and
     *      internal-node serializers
     */
    public function createTree($max_degree, $format =
        ["PRIMARY KEY" => ["KEY", -1], "VALUE" => "BLOB"])
    {
        $table_dir = __DIR__ . self::TEST_DIR . $max_degree;
        $this->table_dirs[] = $table_dir;
        return new LDS\BPlusTree($table_dir, $format, $max_degree,
            C\NS_COMPRESSORS . "NonCompressor",
            C\NS_COMPRESSORS . "NonCompressor");
    }
    /**
     * Test putting items in bplustrees of odd sized nodes between 3 and 13 and
     * then seeing if the items can be retrieved
     */
    public function putGetTextTestCase()
    {
        $format = ["PRIMARY KEY" => ["KEY", -1], "VALUE" => "TEXT"];
        for ($i = 3; $i <= 13; $i += 2) {
            $bptree = $this->createTree($i, $format);
            for ($j = 0; $j < ($i * 40); $j++) {
                for($k = 0; $k < 5; $k++) {
                    $bptree->put(["KEY" => str_pad("$j",4,"0", STR_PAD_LEFT),
                        "VALUE" => "row{$j}_{$k}"],
                        PackedTableTools::APPEND_MODE);
                }
            }
            $bptree->flushLastPutNode();
            for ($j = 0; $j < ($i * 40); $j++) {
                $rows = $bptree->get(str_pad("$j",4,"0", STR_PAD_LEFT));
                for($k = 0; $k < 5; $k++) {
                    $this->assertEqual("row{$j}_{$k}", $rows[$k]["VALUE"],
                        "{$j}th insert into tree of size $i was retrieved okay".
                        " {$k}th case");
                }
            }
        }
    }
    /**
     * Test putting items in bplustrees of odd sized nodes between 3 and 13 and
     * then seeing if the items can be retrieved
     */
    public function putGetBlobTestCase()
    {
        for ($i = 3; $i <= 13; $i += 2) {
            $bptree = $this->createTree($i);
            for ($j = 0; $j < ($i * 40); $j++) {
                for($k = 0; $k < 5; $k++) {
                    $bptree->put(["KEY" => str_pad("$j", 4, "0", STR_PAD_LEFT),
                        "VALUE" => "row{$j}_{$k}"],
                        PackedTableTools::APPEND_MODE);
                }
            }
            $bptree->flushLastPutNode();
            for ($j = 0; $j < ($i * 40); $j++) {
                $rows = $bptree->get(str_pad("$j",4,"0",
                    STR_PAD_LEFT));
                for($k = 0; $k < 5; $k++) {
                    $this->assertEqual("row{$j}_{$k}", $rows[$k]["VALUE"],
                        "{$j}th insert into tree of size $i was retrieved okay".
                        " {$k}th case");
                }
            }
        }
    }
}
