znt/test/lib/init.php

<?php
/**
 * 本文件创建一个 app 实例，并且通过执行 $app->loadCommon() 方法创建名为 tester 的commonModel对象。
 * This file build a app instance and provide a instance of commonModel named as tester by $app->loadCommon().
 *
 * All request of entries should be routed by this router.
 *
 * @copyright   Copyright 2009-2017 禅道软件（青岛）有限公司(ZenTao Software (Qingdao) Co., Ltd. www.cnezsoft.com)
 * @license     ZPL(http://zpl.pub/page/zplv12.html) or AGPL(https://www.gnu.org/licenses/agpl-3.0.en.html)
 * @author      Guanxing <guanxiying@easycorp.ltd>
 * @package     ZenTaoPMS
 * @version     $Id: $
 * @link        http://www.zentao.net
 */
/* Set the error reporting. */
error_reporting(E_ALL);
define('RUN_MODE', 'test');
if(!defined('LIB_ROOT')) define('LIB_ROOT', dirname(dirname(__FILE__)) . DIRECTORY_SEPARATOR . 'lib' . DIRECTORY_SEPARATOR);

include_once LIB_ROOT . 'coverage.php';

$codeCoverageConfig = dirname(LIB_ROOT) . DIRECTORY_SEPARATOR . 'config' . DIRECTORY_SEPARATOR . 'my.php';
$codeCoverageConfig = exec("sed -n 's/^\\\$config->codeCoverage *= *\\(.*\\);/\\1/p' $codeCoverageConfig");

if($argc > 1 && $argv[1] == '-extract')
{
    include 'parseztfstep.php';
    printSteps();
    exit;
}

$zentaoRoot    = dirname(__FILE__, 3) . DIRECTORY_SEPARATOR;
$testPath      = $zentaoRoot . 'test' . DIRECTORY_SEPARATOR;
$frameworkRoot = $zentaoRoot . 'framework' . DIRECTORY_SEPARATOR;

if(isset($codeCoverageConfig) and $codeCoverageConfig == 'true')
{
    $coverage = new coverage();
    $coverage->startCodeCoverage();
}

/**
 * Ensure returned variable is an object
 * 确保返回的变量是对象，如果是JSON字符串则转换为对象
 *
 * @param mixed $data   variable to check
 * @return object|mixed object converted from variable or itself (if can't convert）
 */
function ensureObject($data)
{
    if(is_object($data)) return $data;

    if(is_string($data))
    {
        $decoded = json_decode($data);
        if(json_last_error() === JSON_ERROR_NONE) return $decoded;
    }

    return $data;
}

/**
 * Assert status code and set body as $_result.
 *
 * @param  int $code
 * @access public
 * @return bool
 */
function c($code)
{
    global $_result;
    if(empty($_result) or empty($_result->status_code)) return false;

    $codeStr       = (string)$code;
    $statusCodeStr = (string)($_result->status_code);

    if($_result->status_code == $code or ($statusCodeStr[0] === '2' and $codeStr[0] === '2'))
    {
        $_result = ensureObject($_result->body);
        return true;
    }

    echo "\n\n";
    return false;
}

/* Load the framework. */
include $frameworkRoot . 'router.class.php';
include $frameworkRoot . 'control.class.php';
include $frameworkRoot . 'model.class.php';
include $frameworkRoot . 'helper.class.php';

$app    = router::createApp('pms', dirname(__FILE__, 3), 'router');
$poolID = getenv('ZTF_POOL_ID');
$dbPool = empty($config->dbPool) ? array() : $config->dbPool;

/* 根据ztf设置的poolID环境变量设置连接的数据库 */
if(!empty($dbPool) && !empty($poolID))
{
    $selectDB = $dbPool[$poolID%count($dbPool)];

    !empty($selectDB['host'])     && $config->db->host     = $selectDB['host'];
    !empty($selectDB['port'])     && $config->db->port     = $selectDB['port'];
    !empty($selectDB['name'])     && $config->db->name     = $selectDB['name'];
    !empty($selectDB['user'])     && $config->db->user     = $selectDB['user'];
    !empty($selectDB['password']) && $config->db->password = $selectDB['password'];

    $app->connectDB();
}

$tester = $app->loadCommon();

/* Set configs. */
$config->zendataRoot = dirname(dirname(__FILE__)) . '/zendata';
$config->ztfPath     = dirname(dirname(__FILE__)) . '/tools/ztf';
$config->zdPath      = dirname(dirname(__FILE__)) . '/tools/zd';

/* init testDB. */
include $testPath . 'lib/yaml.class.php';
include $testPath . 'lib/rest.php';

if(!empty($config->test->account) and !empty($config->test->password) and !empty($config->test->base))
{
    $rest  = new rest($config->test->base);
    $token = $rest->post('/tokens', array('account' => $config->test->account, 'password' => $config->test->password));
    $token = $token->body;
}

/**
 * Save variable to $_result.
 *
 * @param  mixed    $result
 * @access public
 * @return bool true
 */
function r($result)
{
    global $_result;
    $_result = $result;
    return true;
}

/**
 * Print value or properties.
 *
 * @param  string    $key
 * @param  string    $delimiter
 * @access public
 * @return void
 */
function p($keys = '', $delimiter = ',')
{
    global $_result;

    if(empty($_result)) return print(implode("\n", array_fill(0, substr_count($keys, $delimiter) + 1, 0)) . "\n");

    if(is_array($_result) && isset($_result['code']) && $_result['code'] == 'fail') return print((string) $_result['message'] . "\n");

    /* Print $_result. */
    if($keys === '' && is_array($_result)) return print_r($_result) . "\n";
    if($keys === '' || !is_array($_result) && !is_object($_result)) return print((string) $_result . "\n");

    $parts  = explode(';', $keys);
    foreach($parts as $part)
    {
        $values = getValues($_result, $part, $delimiter);
        if(!is_array($values)) continue;

        foreach($values as $value) echo $value . "\n";
    }

    return true;
}

/**
 * Get values
 *
 * @param mixed  $value
 * @param string $keys
 * @param string $delimiter
 * @access public
 * @return void
 */
function getValues($value, $keys, $delimiter)
{
    $object = '';
    $index  = -1;
    $pos    = strpos($keys, ':');
    if($pos)
    {
        $arrKey = substr($keys, 0, $pos);
        $keys   = substr($keys, $pos + 1);

        $pos = strpos($arrKey, '[');
        if($pos)
        {
            $object = substr($arrKey, 0, $pos);
            $index  = trim(substr($arrKey, $pos + 1), ']');
        }
        else
        {
            $index = $arrKey;
        }
    }
    $keys = explode($delimiter, $keys);

    if($object !== '')
    {
        if(is_array($value))
        {
            $value = $value[$object];
        }
        else if(is_object($value))
        {
            $value = $value->$object;
        }
        else
        {
            return print("Error: No object name '$object'.\n");
        }
    }

    if($index != -1)
    {
        if(is_array($value))
        {
            if(!isset($value[$index])) return print("Error: Cannot get index $index.\n");
            $value = $value[$index];
        }
        else if(is_object($value))
        {
            if(!isset($value->$index)) return print("Error: Cannot get index $index.\n");
            $value = $value->$index;
        }
        else
        {
            return print("Error: Not array, cannot get index $index.\n");
        }
    }

    $values = array();
    foreach($keys as $key) $values[] = zget($value, $key, '');

    return $values;
}

/**
 * Expect values, ztf will put params to step.
 *
 * @param  string    $exepect
 * @access public
 * @return void
 */
function e($expect)
{
    global $codeCoverageConfig;
    global $coverage;
    if(isset($codeCoverageConfig) and $codeCoverageConfig == 'true') $coverage->saveAndRestartCodeCoverage();
}

/**
 * Check order
 *
 * @param array  $objs
 * @param string $orderBy
 * @access public
 * @return bool
 */
function checkOrder($objs, $orderBy)
{
    if(empty($objs)) return true;

    list($field, $sort) = explode('_', $orderBy);
    $last = current($objs)->$field;
    foreach($objs as $obj)
    {
        if($sort == 'desc')
        {
            if($obj->$field > $last) return false;
        }
        else
        {
            if($obj->$field < $last) return false;
        }
    }

    return true;
}

/**
 * Import data create by zendata to one table.
 *
 * @param  string    $table
 * @param  string    $yaml
 * @param  int       $count
 * @access public
 * @return void
 */
function zdImport($table, $yaml, $count = 10)
{
    chdir(dirname(__FILE__));

    global $app, $config;
    $dns   = "mysql://{$config->db->user}:{$config->db->password}@{$config->db->host}:{$config->db->port}/{$config->db->name}#utf8";
    $table = trim($table, '`');

    $command = "$config->zdPath -c $yaml -t $table -T -dns $dns --clear -n $count";
    system($command);
}

/**
 * Switch user.
 *
 * @param  string $account
 * @param  bool   $initRights  If true, will init user rights, groups, view.
 * @access public
 * @return bool
 */
function su($account, $initRights = true)
{
    $userModel = new userModel();
    $user      = $userModel->getByID($account);
    if(!$user) return false;

    common::$userPrivs = array();
    if($initRights) return $userModel->login($user, false);

    global $app;
    $user = $userModel->identify($account, $user->password);
    $user->admin = strpos($app->company->admins, ",{$user->account},") !== false;

    $app->session->set('user', $user);
    $app->user = $user;

    return true;
}

/**
 * 初始化反射类。
 * Initialize reflection class.
 *
 * @param  int    $moduleName
 * @param  string $type
 * @access public
 * @return bool|ReflectionClass
 */
function initReference($moduleName, $type = 'zen')
{
    global $tester;
    helper::import($tester->app->getModulePath('', $moduleName) . 'control.php');
    $tester->app->control = new $moduleName();
    $class = '';
    if($type == 'zen')
    {
        helper::import($tester->app->getModulePath('', $moduleName) . 'zen.php');
        $class = $moduleName . 'Zen';
    }
    else
    {
        helper::import($tester->app->getModulePath('', $moduleName) . 'model.php');
        $class = $moduleName . 'Model';
    }

    if(!$class || !class_exists($class)) return false;
    return new ReflectionClass($class);
}

function callZenMethod($moduleName, $name, $args, $property = '')
{
    global $app;
    $zenTest = $app->loadTarget($moduleName, '', 'zen');
    $reflection = new ReflectionClass($zenTest);

    $method = $reflection->getMethod($name);
    $method->setAccessible(true);
    $result = $method->invokeArgs($zenTest, $args);
    if(empty($property)) return $result; // 如果不获取属性值，则返回被调用方法的返回值。

    $zenProperty = $reflection->getProperty($property);
    $zenProperty->setAccessible(true);
    return $zenProperty->getValue($zenTest); // 返回zen的属性值。
}