Repository navigation
Expand file tree
/
Copy pathModelGenerateTest.php
More file actions
120 lines (107 loc) · 3.64 KB
/
Copy pathModelGenerateTest.php
File metadata and controls
120 lines (107 loc) · 3.64 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
<?php
declare(strict_types=1);
/**
* This file is part of Hyperf.
*
* @link https://www.hyperf.io
* @document https://hyperf.wiki
* @contact [email protected]
* @license https://github.com/hyperf/hyperf/blob/master/LICENSE
*/
namespace HyperfTest\Database;
use Hyperf\Database\Commands\Ast\GenerateModelIDEVisitor;
use Hyperf\Database\Commands\ModelData;
use Hyperf\Database\Commands\ModelOption;
use HyperfTest\Database\Stubs\Model\TestGenerateIdeModel;
use PhpParser\Lexer\Emulative;
use PhpParser\NodeTraverser;
use PhpParser\ParserFactory;
use PhpParser\PrettyPrinter\Standard;
use PHPUnit\Framework\TestCase;
/**
* @internal
* @coversNothing
*/
class ModelGenerateTest extends TestCase
{
public function setUp(): void
{
$this->lexer = new Emulative([
'usedAttributes' => [
'comments',
'startLine', 'endLine',
'startTokenPos', 'endTokenPos',
],
]);
$this->astParser = (new ParserFactory())->create(ParserFactory::ONLY_PHP7, $this->lexer);
$this->printer = new Standard();
}
public function testGenerateScope()
{
$code = file_get_contents(__DIR__ . '/Stubs/Model/TestGenerateIdeModel.php');
$option = new ModelOption();
$data = new ModelData(TestGenerateIdeModel::class, [
['column_name' => 'name'],
['column_name' => 'age'],
]);
$stmts = $this->astParser->parse($code);
$traverser = new NodeTraverser();
$traverser->addVisitor(new GenerateModelIDEVisitor($option, $data));
$stmts = $traverser->traverse($stmts);
$code = $this->printer->prettyPrintFile($stmts);
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeOptionNull
*/
public static function optionNull(?string $test)
{
return static::$builder;
}'));
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeOptionNull
*/
public static function optionNull(?string $test)
{
return static::$builder;
}'));
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeString
*/
public static function string(string $test)
{
return static::$builder;
}'));
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeUnion
*/
public static function union(int $appId, string|int $uid)
{
return static::$builder;
}'));
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeUnionOrNull
*/
public static function unionOrNull(int $appId, string|int|null $uid)
{
return static::$builder;
}'));
$this->assertNotFalse(strpos($code, '
/**
* @return \Hyperf\Database\Model\Builder|static
* @see HyperfTest\Database\Stubs\Model\TestGenerateIdeModel::scopeSingleOrNull
*/
public static function singleOrNull(?string $test)
{
return static::$builder;
}'));
}
}