此脚本可以为已有的代码快速生成PHP的单元测试代码,用于辅助测试,以减少开发量。同时,也可以使用PHPUnit官方提供的phpunit-skelgen来达到同样的效果。只是自定义的生成脚本,更多地关注了用户的需求,以及方便修改扩展自己的需求。并且,新版中可以根据注
此脚本可以为已有的代码快速生成PHP的单元测试代码,用于辅助测试,以减少开发量。同时,也可以使用PHPUnit官方提供的phpunit-skelgen来达到同样的效果。只是自定义的生成脚本,更多地关注了用户的需求,以及方便修改扩展自己的需求。并且,新版中可以根据注释@testcase来自动生成简单的测试用例!欢迎试用,谢谢!
//$ cat ./build_phpunit_test_tpl.php #!/usr/bin/env php <?php /** * 单元测试骨架代码自动生成脚本 * 主要是针对当前项目系列生成相应的单元测试代码,提高开发效率 * * 用法: * Usage: php ./build_phpunit_test_tpl.php <file_path> <class_name> [bootstrap] [author = dogstar] * * 1、针对全部public的函数进行单元测试 * 2、可根据@testcase注释自动生成测试用例 * * 备注:另可使用phpunit-skelgen进行骨架代码生成 * * @author: dogstar 20150108 * @version: 4.0.0 */ if ($argc < 3) { echo " Usage: php $argv[0] <file_path> <class_name> [bootstrap] [author = dogstar] Demo: php ./build_phpunit_test_tpl.php ./Demo.php Demo > Demo_Test.php "; die(); } $filePath = $argv[1]; $className = $argv[2]; $bootstrap = isset($argv[3]) ? $argv[3] : null; $author = isset($argv[4]) ? $argv[4] : 'dogstar'; if (!empty($bootstrap)) { require $bootstrap; } require $filePath; if (!class_exists($className)) { die("Error: cannot find class($className). \n"); } $reflector = new ReflectionClass($className); $methods = $reflector->getMethods(ReflectionMethod::IS_PUBLIC); date_default_timezone_set('Asia/Shanghai'); $objName = lcfirst(str_replace('_', '', $className)); $code = "<?php /** * PhpUnderControl_" . str_replace('_', '', $className) . "_Test * * 针对 $filePath $className 类的PHPUnit单元测试 * * @author: $author " . date('Ymd') . " */ "; if (file_exists(dirname(__FILE__) . '/test_env.php')) { $code .= "require_once dirname(__FILE__) . '/test_env.php'; "; } else { $code .= "//require_once dirname(__FILE__) . '/test_env.php'; "; } $initWay = "new $className()"; if (method_exists($className, '__construct')) { $constructMethod = new ReflectionMethod($className, '__construct'); if (!$constructMethod->isPublic()) { if (is_callable(array($className, 'getInstance'))) { $initWay = "$className::getInstance()"; } else if(is_callable(array($className, 'newInstance'))) { $initWay = "$className::newInstance()"; } else { $initWay = 'NULL'; } } } $code .= " if (!class_exists('$className')) { require dirname(__FILE__) . '/$filePath'; } class PhpUnderControl_" . str_replace('_', '', $className) . "_Test extends PHPUnit_Framework_TestCase { public \$$objName; protected function setUp() { parent::setUp(); \$this->$objName = $initWay; } protected function tearDown() { } "; foreach ($methods as $method) { if($method->class != $className) continue; $fun = $method->name; $Fun = ucfirst($fun); if (strlen($Fun) > 2 && substr($Fun, 0, 2) == '__') continue; $rMethod = new ReflectionMethod($className, $method->name); $params = $rMethod->getParameters(); $isStatic = $rMethod->isStatic(); $isConstructor = $rMethod->isConstructor(); if($isConstructor) continue; $initParamStr = ''; $callParamStr = ''; foreach ($params as $param) { $default = ''; $rp = new ReflectionParameter(array($className, $fun), $param->name); if ($rp->isOptional()) { $default = $rp->getDefaultValue(); } if (is_string($default)) { $default = "'$default'"; } else if (is_array($default)) { $default = var_export($default, true); } else if (is_bool($default)) { $default = $default ? 'true' : 'false'; } else if ($default === null) { $default = 'null'; } else { $default = "''"; } $initParamStr .= " \$" . $param->name . " = $default;"; $callParamStr .= '$' . $param->name . ', '; } $callParamStr = empty($callParamStr) ? $callParamStr : substr($callParamStr, 0, -2); /** ------------------- 根据@return对结果类型的简单断言 ------------------ **/ $returnAssert = ''; $docComment = $rMethod->getDocComment(); $docCommentArr = explode("\n", $docComment); foreach ($docCommentArr as $comment) { if (strpos($comment, '@return') == false) { continue; } $returnCommentArr = explode(' ', strrchr($comment, '@return')); if (count($returnCommentArr) >= 2) { switch (strtolower($returnCommentArr[1])) { case 'bool': case 'boolean': $returnAssert = '$this->assertTrue(is_bool($rs));'; break; case 'int': $returnAssert = '$this->assertTrue(is_int($rs));'; break; case 'integer': $returnAssert = '$this->assertTrue(is_integer($rs));'; break; case 'string': $returnAssert = '$this->assertTrue(is_string($rs));'; break; case 'object': $returnAssert = '$this->assertTrue(is_object($rs));'; break; case 'array': $returnAssert = '$this->assertTrue(is_array($rs));'; break; case 'float': $returnAssert = '$this->assertTrue(is_float($rs));'; break; } break; } } /** ------------------- 基本的单元测试代码生成 ------------------ **/ $code .= " /** * @group test$Fun */ public function test$Fun() {" . (empty($initParamStr) ? '' : "$initParamStr\n") . "\n " . ($isStatic ? "\$rs = $className::$fun($callParamStr);" : "\$rs = \$this->$objName->$fun($callParamStr);") . (empty($returnAssert) ? '' : "\n\n " . $returnAssert . "\n") . " } "; /** ------------------- 根据@testcase 生成测试代码 ------------------ **/ $caseNum = 0; foreach ($docCommentArr as $comment) { if (strpos($comment, '@testcase') == false) { continue; } $returnCommentArr = explode(' ', strrchr($comment, '@testcase')); if (count($returnCommentArr) > 1) { $expRs = $returnCommentArr[1]; $callParamStrInCase = isset($returnCommentArr[2]) ? $returnCommentArr[2] : ''; $code .= " /** * @group test$Fun */ public function test{$Fun}Case{$caseNum}() {" . "\n " . ($isStatic ? "\$rs = $className::$fun($callParamStrInCase);" : "\$rs = \$this->$objName->$fun($callParamStrInCase);") . "\n\n \$this->assertEquals({$expRs}, \$rs);" . " } "; $caseNum ++; } } } $code .= " }"; echo $code; echo "\n";