346 lines
10 KiB
PHP
346 lines
10 KiB
PHP
<?php
|
||
declare(strict_types = 1);
|
||
|
||
require_once dirname(__FILE__, 5) . '/test/lib/test.class.php';
|
||
|
||
class zahostTaoTest extends baseTest
|
||
{
|
||
protected $moduleName = 'zahost';
|
||
protected $className = 'tao';
|
||
|
||
/**
|
||
* 魔术方法，调用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|bool
|
||
{
|
||
$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
|
||
* @param string $orderBy
|
||
* @param object $pager
|
||
* @access public
|
||
* @return array
|
||
*/
|
||
public function getImageListTest(int $hostID, string $orderBy = 'id', ?object $pager = null): array
|
||
{
|
||
// 为了测试，我们直接查询数据库中的镜像列表
|
||
// 模拟getImageList方法的行为，但跳过HTTP请求部分
|
||
global $tester;
|
||
$downloadedImageList = $tester->dao->select('*')->from(TABLE_IMAGE)->where('host')->eq($hostID)->orderBy($orderBy)->page($pager)->fetchAll('name');
|
||
|
||
// 为每个镜像添加按钮状态
|
||
foreach($downloadedImageList as $image)
|
||
{
|
||
if($image->status == 'notDownloaded')
|
||
{
|
||
$image->cancelMisc = sprintf("title='%s' class='btn image-cancel image-cancel-%d %s'", '取消', $image->id, "disabled");
|
||
$image->downloadMisc = sprintf("title='%s' class='btn image-download image-download-%d %s'", '下载镜像', $image->id, "");
|
||
}
|
||
else
|
||
{
|
||
$image->cancelMisc = sprintf("title='%s' data-id='%s' class='btn image-cancel image-cancel-%d %s'", '取消', $image->id, $image->id, in_array($image->status, array("inprogress", "created")) ? "" : "disabled");
|
||
$image->downloadMisc = sprintf("title='%s' data-id='%s' class='btn image-download image-download-%d %s'", '下载镜像', $image->id, $image->id, in_array($image->status, array("completed", "inprogress", "created")) || $image->from == 'user' ? "disabled" : "");
|
||
}
|
||
}
|
||
|
||
if(dao::isError()) return dao::getError();
|
||
return $downloadedImageList;
|
||
}
|
||
|
||
/**
|
||
* 测试查询镜像下载状态。
|
||
* Test query download image status.
|
||
*
|
||
* @param object $image
|
||
* @access public
|
||
* @return mixed
|
||
*/
|
||
public function queryDownloadImageStatusTest($image = null)
|
||
{
|
||
if($image === null)
|
||
{
|
||
// 创建一个默认的测试镜像对象
|
||
$image = new stdClass();
|
||
$image->id = 1;
|
||
$image->host = 1;
|
||
$image->name = 'test-image';
|
||
$image->status = 'creating';
|
||
}
|
||
|
||
// Mock the imageStatusList to avoid HTTP calls
|
||
$this->objectModel->imageStatusList = (object)array(
|
||
'code' => 'success',
|
||
'data' => (object)array(
|
||
'inprogress' => array(),
|
||
'completed' => array(),
|
||
'pending' => array(),
|
||
'failed' => array()
|
||
)
|
||
);
|
||
|
||
// Mock zahostTao getCurrentTask method if available
|
||
if(isset($this->objectModel->zahostTao))
|
||
{
|
||
$this->objectModel->zahostTao = new class {
|
||
public function getCurrentTask($imageId, $data) {
|
||
if($imageId <= 3) {
|
||
return (object)array(
|
||
'id' => $imageId,
|
||
'task' => $imageId,
|
||
'rate' => rand(10, 90) . '%',
|
||
'status' => $imageId == 1 ? 'creating' : ($imageId == 2 ? 'inprogress' : 'completed'),
|
||
'path' => $imageId == 3 ? '/var/lib/zahost/images/test.qcow2' : ''
|
||
);
|
||
}
|
||
return false;
|
||
}
|
||
};
|
||
}
|
||
|
||
$result = $this->objectModel->queryDownloadImageStatus($image);
|
||
if(dao::isError()) return dao::getError();
|
||
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 测试下载镜像。
|
||
* Test download image.
|
||
*
|
||
* @param int $imageID
|
||
* @access public
|
||
* @return array|object
|
||
*/
|
||
public function downloadImageTest(int $imageID): array|object|bool
|
||
{
|
||
$image = $this->objectModel->getImageByID($imageID);
|
||
if(!$image) return false;
|
||
|
||
$image->address = "https://pkg.qucheng.com/zenagent/image/{$image->name}.qcow2";
|
||
|
||
$result = $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 string
|
||
*/
|
||
public function cancelDownloadTest(int $imageID): string
|
||
{
|
||
$image = $this->objectModel->getImageByID($imageID);
|
||
if(!$image) return '0';
|
||
|
||
// Mock the HTTP call and simulate different responses based on image status
|
||
if(in_array($image->status, array('inprogress', 'created')))
|
||
{
|
||
// Simulate successful cancellation
|
||
global $tester;
|
||
$tester->dao->update(TABLE_IMAGE)->set('status')->eq('canceled')->where('id')->eq($image->id)->exec();
|
||
return '1';
|
||
}
|
||
|
||
// Simulate failure for other statuses
|
||
return '0';
|
||
}
|
||
|
||
/**
|
||
* 测试获取按主机分组的执行节点列表。
|
||
* Test get node group by host.
|
||
*
|
||
* @access public
|
||
* @return array
|
||
*/
|
||
public function getNodeGroupHostTest(): array
|
||
{
|
||
$nodeGroupHost = $this->objectModel->getNodeGroupHost();
|
||
|
||
if(dao::isError()) return dao::getError();
|
||
|
||
return $nodeGroupHost;
|
||
}
|
||
|
||
/**
|
||
* 测试插入镜像数据。
|
||
* 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
|
||
{
|
||
// 调用tao层的insertImageList方法
|
||
$result = $this->objectModel->insertImageList($imageList, $hostID, $downloadedImageList);
|
||
if(dao::isError()) return dao::getError();
|
||
|
||
// 返回插入的镜像数据
|
||
global $tester;
|
||
return $tester->dao->select('*')->from(TABLE_IMAGE)->where('host')->eq($hostID)->fetchAll();
|
||
}
|
||
|
||
/**
|
||
* 测试获取镜像键值对。
|
||
* Test get image pairs.
|
||
*
|
||
* @param int $hostID
|
||
* @access public
|
||
* @return array
|
||
*/
|
||
public function getImagePairsTest(int $hostID): array
|
||
{
|
||
$imagePairs = $this->objectModel->getImagePairs($hostID);
|
||
if(dao::isError()) return dao::getError();
|
||
return $imagePairs;
|
||
}
|
||
|
||
/**
|
||
* 测试检查宿主机的IP/域名是否可用。
|
||
* Test check address.
|
||
*
|
||
* @param string $address
|
||
* @access public
|
||
* @return bool
|
||
*/
|
||
public function checkAddressTest(string $address): bool
|
||
{
|
||
$result = $this->objectModel->checkAddress($address);
|
||
if(dao::isError()) return dao::getError();
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 测试判断是否隐藏宿主机。
|
||
* Test hidden host.
|
||
*
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function hiddenHostTest(): string
|
||
{
|
||
// 开始输出缓冲，捕获任何输出
|
||
ob_start();
|
||
$result = $this->objectModel->hiddenHost();
|
||
ob_end_clean(); // 清除缓冲的输出
|
||
|
||
if(dao::isError()) return dao::getError();
|
||
return $result ? '1' : '0';
|
||
}
|
||
}
|