LaravelFreelancerNL /
laravel-arangodb
| 1 | <?php |
||
| 2 | |||
| 3 | declare(strict_types=1); |
||
| 4 | |||
| 5 | namespace LaravelFreelancerNL\Aranguent\Testing; |
||
| 6 | |||
| 7 | use Illuminate\Database\ConnectionInterface; |
||
| 8 | use Illuminate\Foundation\Testing\DatabaseTruncation as IlluminateDatabaseTruncation; |
||
| 9 | use Illuminate\Support\Collection; |
||
| 10 | use LaravelFreelancerNL\Aranguent\Testing\Concerns\CanConfigureMigrationCommands; |
||
| 11 | |||
| 12 | trait DatabaseTruncation |
||
| 13 | { |
||
| 14 | use IlluminateDatabaseTruncation; |
||
| 15 | use CanConfigureMigrationCommands; |
||
| 16 | |||
| 17 | /** |
||
| 18 | * The parameters that should be used when running "migrate:fresh". |
||
| 19 | * |
||
| 20 | * Duplicate code because CanConfigureMigrationCommands has a conflict otherwise. |
||
| 21 | * |
||
| 22 | * @return array |
||
| 23 | */ |
||
| 24 | 1 | protected function migrateFreshUsing() |
|
| 25 | { |
||
| 26 | 1 | $seeder = $this->seeder(); |
|
| 27 | |||
| 28 | 1 | $results = array_merge( |
|
| 29 | 1 | [ |
|
| 30 | 1 | '--drop-analyzers' => $this->shouldDropAnalyzers(), |
|
| 31 | 1 | '--drop-graphs' => $this->shouldDropGraphs(), |
|
| 32 | 1 | '--drop-views' => $this->shouldDropViews(), |
|
| 33 | 1 | '--drop-types' => $this->shouldDropTypes(), |
|
| 34 | 1 | '--drop-all' => $this->shouldDropAll(), |
|
| 35 | 1 | ], |
|
| 36 | 1 | $seeder ? ['--seeder' => $seeder] : ['--seed' => $this->shouldSeed()], |
|
| 37 | 1 | $this->setMigrationPaths(), |
|
| 38 | 1 | ); |
|
| 39 | |||
| 40 | 1 | return $results; |
|
| 41 | } |
||
| 42 | |||
| 43 | /** |
||
| 44 | * Determine if a table exists in the given list, with or without its schema. |
||
| 45 | */ |
||
| 46 | 1 | protected function tableExistsIn(array $table, array $tables): bool |
|
| 47 | { |
||
| 48 | 1 | return isset($table['schema']) |
|
| 49 | ? ! empty(array_intersect([$table['name'], $table['schema'] . '.' . $table['name']], $tables)) |
||
| 50 | 1 | : in_array($table['name'], $tables); |
|
| 51 | } |
||
| 52 | |||
| 53 | /** |
||
| 54 | * Truncate the database tables for the given database connection. |
||
| 55 | * |
||
| 56 | * @param \Illuminate\Database\ConnectionInterface $connection |
||
| 57 | * @param string|null $name |
||
| 58 | * @return void |
||
| 59 | */ |
||
| 60 | 1 | protected function truncateTablesForConnection(ConnectionInterface $connection, ?string $name): void |
|
| 61 | { |
||
| 62 | 1 | $dispatcher = $connection->getEventDispatcher(); |
|
| 63 | |||
| 64 | 1 | $connection->unsetEventDispatcher(); |
|
| 65 | |||
| 66 | 1 | (new Collection($this->getAllTablesForConnection($connection, $name))) |
|
|
0 ignored issues
–
show
Bug
introduced
by
Loading history...
|
|||
| 67 | 1 | ->when( |
|
| 68 | 1 | $this->tablesToTruncate($connection, $name), |
|
| 69 | 1 | function (Collection $tables, array $tablesToTruncate) { |
|
| 70 | return $tables->filter(fn(array $table) => $this->tableExistsIn($table, $tablesToTruncate)); |
||
| 71 | 1 | }, |
|
| 72 | 1 | function (Collection $tables) use ($connection, $name) { |
|
| 73 | 1 | $exceptTables = $this->exceptTables($connection, $name); |
|
| 74 | 1 | return $tables->filter(fn(array $table) => ! $this->tableExistsIn($table, $exceptTables)); |
|
| 75 | 1 | }, |
|
| 76 | 1 | ) |
|
| 77 | 1 | ->each(function (array $table) use ($connection) { |
|
| 78 | 1 | $connection->withoutTablePrefix(function ($connection) use ($table) { |
|
| 79 | 1 | $table = $connection->table( |
|
| 80 | 1 | isset($table['schema']) ? $table['schema'] . '.' . $table['name'] : $table['name'], |
|
| 81 | 1 | ); |
|
| 82 | 1 | if ($table->exists()) { |
|
| 83 | 1 | $table->truncate(); |
|
| 84 | } |
||
| 85 | 1 | }); |
|
| 86 | 1 | }); |
|
| 87 | |||
| 88 | 1 | $connection->setEventDispatcher($dispatcher); |
|
| 89 | } |
||
| 90 | |||
| 91 | } |
||
| 92 |