EasySoft-ZenTaoPMS/module/zahost/test/zahost.class.php

<?php
declare(strict_types=1);

use function zin\on;

class zahostTest
{
    public function __construct()
    {
        global $tester;
        $this->objectModel = $tester->loadModel('zahost');

        $tester->loadModel('host');
    }

    /**
     * 魔术方法，调用zahost模型中的方法。
     * Magic method, call the method in the zahost model.
     *
     * @param  string $name
     * @param  array  $arguments
     * @access public
     * @return mixed
     */
    public function __call($name, $arguments)
    {
        return $this->objectModel->$name(...$arguments);
    }

    /**
     * 测试根据编号获取主机信息。
     * Get by id test.
     *
     * @param  int          $zahostID
     * @access public
     * @return array|object
     */
    public function getByIDTest(int $zahostID): array|object
    {
        $zahost = $this->objectModel->getByID($zahostID);
        if(dao::isError()) return dao::getError();
        return $zahost;
    }

    /**
     * 测试获取主机键值对。
     * Get host pairs test.
     *
     * @access public
     * @return array
     */
    public function getPairsTest(): array
    {
        $hostPairs = $this->objectModel->getPairs();

        if(dao::isError()) return dao::getError();

        return $hostPairs;
    }

    /**
     * 测试获取主机列表。
     * Get host list test.
     *
     * @param  string $browseType
     * @param  int    $param
     * @access public
     * @return array
     */
    public function getListTest(string $browseType, int $param): array
    {
        $hosts = $this->objectModel->getList($browseType, $param);

        if(dao::isError()) return dao::getError();

        return $hosts;
    }

    /**
     * 测试创建主机。
     * Test create host.
     *
     * @param  object $hostInfo
     * @access public
     * @return array|object
     */
    public function createTest(object $hostInfo): array|object
    {
        $return = $this->objectModel->create($hostInfo);
        if(dao::isError()) return dao::getError();

        $hostID = $return;
        return $this->objectModel->getByID($hostID);
    }

    /**
     * 测试更新主机。
     * Test update host.
     *
     * @param  object $hostInfo
     * @access public
     * @return array
     */
    public function updateTest(object $hostInfo): array
    {
        $return = $this->objectModel->update($hostInfo);
        if(dao::isError()) return dao::getError();

        return $return;
    }

    /**
     * 测试检查宿主机的IP/域名是否能 ping 通。
     * Test ping address.
     *
     * @param  string $address
     * @access public
     * @return string
     */
    public function pingTest(string $address): string
    {
        $ping = $this->objectModel->ping($address);
        if($ping) return 'yes';
        return 'no';
    }

    /**
     * 测试获取主机的镜像列表。
     * Test get image list.
     *
     * @param  int    $hostID
     * @access public
     * @return array
     */
    public function getImageListTest(int $hostID): array
    {
        $imageList = $this->objectModel->getImageList($hostID);
        if(dao::isError()) return dao::getError();
        return $imageList;
    }

    /**
     * 测试查询镜像的状态。
     * Test download image.
     *
     * @param  int          $imageID
     * @access public
     * @return array|bool
     */
    public function queryDownloadImageStatusTest(int $imageID): array|bool
    {
        $image = $this->objectModel->getImageByID($imageID);
        $image->address = "https://pkg.qucheng.com/zenagent/image/{$image->name}.qcow2";

        $this->imageStatusList = array();

        $this->objectModel->queryDownloadImageStatus($image);
        if(dao::isError()) return dao::getError();

        $updatedImage = $this->objectModel->getImageByID($imageID);

        return $updatedImage->status != 'creating';
    }

    /**
     * 测试下载镜像。
     * Test download image.
     *
     * @param  int          $imageID
     * @access public
     * @return array|object
     */
    public function downloadImageTest(int $imageID): array|object
    {
        $image = $this->objectModel->getImageByID($imageID);
        $image->address = "https://pkg.qucheng.com/zenagent/image/{$image->name}.qcow2";

        $this->objectModel->downloadImage($image);
        if(dao::isError()) return dao::getError();

        return $this->objectModel->getImageByID($imageID);
    }

    /**
     * 测试取消下载镜像。
     * Test cancel download image.
     *
     * @param  int          $imageID
     * @access public
     * @return array|object
     */
    public function cancelDownloadTest(int $imageID): array|object
    {
        $image = $this->objectModel->getImageByID($imageID);
        $image->address = "https://pkg.qucheng.com/zenagent/image/{$image->name}.qcow2";

        $this->objectModel->cancelDownload($image);
        if(dao::isError()) return dao::getError();

        return $this->objectModel->getImageByID($imageID);
    }

    /**
     * 测试获取按主机分组的执行节点列表。
     * Test get node group by host.
     *
     * @param  int    $hostID
     * @access public
     * @return array
     */
    public function getNodeGroupHostTest(int $hostID): array
    {
        $nodeGroupHost = $this->objectModel->getNodeGroupHost();

        if(dao::isError()) return dao::getError();

        return zget($nodeGroupHost, $hostID, array());
    }

    /**
     * 测试插入镜像数据。
     * Test insert image list.
     *
     * @param  array $imageList
     * @param  int   $hostID
     * @param  array $downloadedImageList
     * @access public
     * @return array
     */
    public function insertImageListTest(array $imageList, int $hostID, array $downloadedImageList): array
    {
        $this->objectModel->insertImageList($imageList, $hostID, $downloadedImageList);

        global $tester;
        return $tester->dao->select('*')->from(TABLE_IMAGE)->where('host')->eq($hostID)->fetchAll();
    }
}