thecodingmachine / DBFaker

tests/DBFakerTest.php

Spacing +4 added lines, -4 removed lines

@@ -63,7 +63,7 @@ 
             ->column("population")->bigInt()
             ->column("birthrate")->float()
             ->column("president_id")->references("persons")
-            ->column("population_density")->decimal(10,2)
+            ->column("population_density")->decimal(10, 2)
             ->column("summary")->text();
 
         $users->column("country_id")->references("countries");
@@ -110,7 +110,7 @@ 
 
         // address.postal_code column is a varchar, so default generated data will be text. Here we want a postal code :
         $generatorFinderBuilder->addGenerator(
-            new \DBFaker\Generators\Conditions\CallBackCondition(function(\Doctrine\DBAL\Schema\Table $table,  \Doctrine\DBAL\Schema\Column $column){
+            new \DBFaker\Generators\Conditions\CallBackCondition(function(\Doctrine\DBAL\Schema\Table $table, \Doctrine\DBAL\Schema\Column $column) {
                 return $table->getName() == "address" && $column->getName() == "postal_code";
             }),
             new SimpleGeneratorFactory("postcode")
@@ -118,7 +118,7 @@ 
 
         // all columns that end with "_email" or are named exactly "email" should be emails
         $generatorFinderBuilder->addGenerator(
-            new CallBackCondition(function(\Doctrine\DBAL\Schema\Table $table,  \Doctrine\DBAL\Schema\Column $column){
+            new CallBackCondition(function(\Doctrine\DBAL\Schema\Table $table, \Doctrine\DBAL\Schema\Column $column) {
                 return preg_match("/([(.*_)|_|]|^)email$/", $column->getName()) === 1;
             }),
             new SimpleGeneratorFactory("email")
@@ -146,7 +146,7 @@ 
         $faker->setFakeTableRowNumbers($tableRowNumbers);
         $faker->fakeDB();
 
-        foreach ($tableRowNumbers as $tableName => $expectedCount){
+        foreach ($tableRowNumbers as $tableName => $expectedCount) {
             $count = $conn->fetchColumn('SELECT count(*) from ' . $tableName);
             $this->assertEquals($expectedCount, $count);
         }

src/DBFaker.php

Spacing +45 added lines, -45 removed lines

@@ -112,7 +112,7 @@ 
      */
     public function fakeDB() : void
     {
-        set_time_limit(0);//Import may take a looooooong time :)
+        set_time_limit(0); //Import may take a looooooong time :)
         $data = $this->generateFakeData();
         $extensionContraints = $this->getExtensionConstraints();
         $foreignKeys = $this->dropForeignKeys();
@@ -165,9 +165,9 @@ 
                 }
                 //Other data will be Faked depending of column's type and attributes. FKs to, but their values wil be overridden.
                 else {
-                    if (!$column->getNotnull() && $this->nullProbabilityOccured()){
+                    if (!$column->getNotnull() && $this->nullProbabilityOccured()) {
                         $value = null;
-                    }else{
+                    }else {
                         $generator = $this->getSimpleColumnGenerator($table, $column);
                         $value = $generator();
                     }
@@ -217,14 +217,14 @@ 
     private function insertWithoutFksAndUniqueIndexes($data): void
     {
         $plateform = $this->connection->getDatabasePlatform();
-        foreach ($data as $tableName => $rows){
+        foreach ($data as $tableName => $rows) {
             $table = $this->schemaManager->listTableDetails($tableName);
 
             //initiate column types for insert : only get the first array to retrieve column names
             $types = [];
             $first = reset($rows);
-            if ($first){
-                foreach ($first as $columnName => $value){
+            if ($first) {
+                foreach ($first as $columnName => $value) {
                     /** @var Column $column */
                     $column = $table->getColumn($columnName);
                     $types[] = $column->getType()->getBindingType();
@@ -233,9 +233,9 @@ 
 
             //insert faked data
             $cnt = count($rows);
-            foreach ($rows as $index => $row){
+            foreach ($rows as $index => $row) {
                 $dbRow = [];
-                foreach ($row as $columnName => $value){
+                foreach ($row as $columnName => $value) {
                     $column = $table->getColumn($columnName);
                     $newVal = $column->getType()->convertToDatabaseValue($value, $plateform);
                     $dbRow[$column->getQuotedName($this->connection->getDatabasePlatform())] = $newVal;
@@ -244,10 +244,10 @@ 
                 $this->connection->insert($table->getName(), $dbRow, $types);
             }
             //if autoincrement, add the new ID to the PKRegistry
-            if ($table->hasPrimaryKey()){
+            if ($table->hasPrimaryKey()) {
                 $pkColumnName = $table->getPrimaryKeyColumns()[0];
                 $pkColumn = $table->getColumn($pkColumnName);
-                if ($pkColumn->getAutoincrement() && $this->schemaHelper->isPrimaryKeyColumn($table, $pkColumn)){
+                if ($pkColumn->getAutoincrement() && $this->schemaHelper->isPrimaryKeyColumn($table, $pkColumn)) {
                     $this->getPkRegistry($table)->addValue([$pkColumnName => $this->connection->lastInsertId()]);
                 }
             }
@@ -262,7 +262,7 @@ 
      */
     public function getPkRegistry(Table $table, $isSelfReferencing = false) : PrimaryKeyRegistry
     {
-        $index = $table->getName().($isSelfReferencing ? 'dbfacker_self_referencing' :'');
+        $index = $table->getName() . ($isSelfReferencing ? 'dbfacker_self_referencing' : '');
         if (!isset($this->primaryKeyRegistries[$index])) {
             $this->primaryKeyRegistries[$index] = new PrimaryKeyRegistry($this->connection, $table, $this->schemaHelper, $isSelfReferencing);
         }
@@ -309,10 +309,10 @@ 
         $this->log->info('Step 2.1 : Drop FKs ...');
         $foreignKeys = [];
         $tables = $this->schemaManager->listTables();
-        foreach ($tables as $table){
-            foreach ($table->getForeignKeys() as $fk){
+        foreach ($tables as $table) {
+            foreach ($table->getForeignKeys() as $fk) {
                 $foreignTable = $this->schemaManager->listTableDetails($fk->getForeignTableName());
-                foreach ($fk->getColumns() as $localColumnName){
+                foreach ($fk->getColumns() as $localColumnName) {
                     $localColumn = $table->getColumn($localColumnName);
                     $selfReferencing = $fk->getForeignTableName() === $table->getName();
                     $fkValueGenerator = new ForeignKeyColumnGenerator($table, $localColumn, $this->getPkRegistry($foreignTable, $selfReferencing), $fk, $this->fakerManagerHelper, $this->schemaHelper);
@@ -332,8 +332,8 @@ 
     private function restoreForeignKeys($foreignKeys) : void
     {
         $this->log->info('Step 4 : restore foreign keys');
-        foreach ($foreignKeys as $tableName => $fks){
-            foreach ($fks as $fk){
+        foreach ($foreignKeys as $tableName => $fks) {
+            foreach ($fks as $fk) {
                 $this->schemaManager->createForeignKey($fk, $tableName);
             }
         }
@@ -348,9 +348,9 @@ 
         $this->log->info('Step 2.2 : Drop Multiple indexes ...');
         $multipleUniqueContraints = [];
         $tables = $this->schemaManager->listTables();
-        foreach ($tables as $table){
-            foreach ($table->getIndexes() as $index){
-                if ($index->isUnique() && count($index->getColumns()) > 1){
+        foreach ($tables as $table) {
+            foreach ($table->getIndexes() as $index) {
+                if ($index->isUnique() && count($index->getColumns()) > 1) {
                     $multipleUniqueContraints[$table->getName()][] = $index;
                     $this->schemaManager->dropIndex($index->getQuotedName($this->connection->getDatabasePlatform()), $table->getName());
                 }
@@ -365,8 +365,8 @@ 
     private function restoreMultipleUniqueContraints($multipleUniqueContraints): void
     {
         $this->log->info('Step 5 : restore multiple unique indexes keys');
-        foreach ($multipleUniqueContraints as $tableName => $indexes){
-            foreach ($indexes as $index){
+        foreach ($multipleUniqueContraints as $tableName => $indexes) {
+            foreach ($indexes as $index) {
                 $this->schemaManager->createIndex($index, $tableName);
             }
         }
@@ -384,24 +384,24 @@ 
     private function updateMultipleUniqueIndexedColumns($multipleUniqueContraints, $handledFKColumns) : array
     {
 
-        foreach ($multipleUniqueContraints as $tableName => $indexes){
+        foreach ($multipleUniqueContraints as $tableName => $indexes) {
             $table = $this->schemaManager->listTableDetails($tableName);
 
-            foreach ($indexes as $index){
-                foreach ($index->getColumns() as $columnName){
-                    $fullColumnName = $tableName. '.' .$columnName;
-                    if (!\in_array($fullColumnName, $handledFKColumns, true)){
+            foreach ($indexes as $index) {
+                foreach ($index->getColumns() as $columnName) {
+                    $fullColumnName = $tableName . '.' . $columnName;
+                    if (!\in_array($fullColumnName, $handledFKColumns, true)) {
                         $handledFKColumns[] = $fullColumnName;
                     }
                 }
             }
 
-            $stmt = $this->connection->query('SELECT * FROM ' .$tableName);
-            $count = $this->connection->fetchColumn('SELECT count(*) FROM ' .$tableName);
+            $stmt = $this->connection->query('SELECT * FROM ' . $tableName);
+            $count = $this->connection->fetchColumn('SELECT count(*) FROM ' . $tableName);
             $i = 1;
             while ($row = $stmt->fetch()) {
                 $newValues = [];
-                foreach ($indexes as $index){
+                foreach ($indexes as $index) {
                     /** @var Index $index */
                     $compoundColumnGenerator = $this->getCompoundColumnGenerator($table, $index, $count);
                     $newValues = array_merge($newValues, $compoundColumnGenerator());
@@ -423,23 +423,23 @@ 
      */
     private function updateRemainingForeignKeys($foreignKeys, $handledFKColumns): void
     {
-        foreach ($foreignKeys as $tableName => $fks){
-            if (!array_key_exists($tableName, $this->fakeTableRowNumbers)){
+        foreach ($foreignKeys as $tableName => $fks) {
+            if (!array_key_exists($tableName, $this->fakeTableRowNumbers)) {
                 //only update tables where data has been inserted
                 continue;
             }
 
             $table = $this->schemaManager->listTableDetails($tableName);
 
-            $stmt = $this->connection->query('SELECT * FROM ' .$tableName);
-            $count = $this->connection->fetchColumn("SELECT count(*) FROM ".$tableName);
+            $stmt = $this->connection->query('SELECT * FROM ' . $tableName);
+            $count = $this->connection->fetchColumn("SELECT count(*) FROM " . $tableName);
             $i = 1;
             while ($row = $stmt->fetch()) {
                 $newValues = [];
                 foreach ($fks as $fk) {
                     $localColumns = $fk->getLocalColumns();
                     foreach ($localColumns as $index => $localColumn) {
-                        if (\in_array($tableName . '.' . $localColumn, $handledFKColumns)){
+                        if (\in_array($tableName . '.' . $localColumn, $handledFKColumns)) {
                             continue;
                         }
                         $column = $table->getColumn($localColumn);
@@ -448,8 +448,8 @@ 
                     }
                 }
                 $row = $this->stripUnselectableColumns($table, $row);
-                if (count($newValues) && $this->connection->update($tableName, $newValues, $row) === 0){
-                    throw new DBFakerException("Row has not been updated $tableName - ". var_export($newValues,true) . ' - ' . var_export($row, true));
+                if (count($newValues) && $this->connection->update($tableName, $newValues, $row) === 0) {
+                    throw new DBFakerException("Row has not been updated $tableName - " . var_export($newValues, true) . ' - ' . var_export($row, true));
                 };
                 $this->log->info("Step 3.3 : updated $i of $count for $tableName");
                 $i++;
@@ -468,10 +468,10 @@ 
     private function updateExtensionContraints($extensionConstraints) : array
     {
         $handledFKColumns = [];
-        foreach ($extensionConstraints as $fk){
+        foreach ($extensionConstraints as $fk) {
             $localTableName = $fk->getLocalTable()->getQuotedName($this->connection->getDatabasePlatform());
             $stmt = $this->connection->query('SELECT * FROM ' . $localTableName);
-            $count = $this->connection->fetchColumn('SELECT count(*) FROM ' .$localTableName);
+            $count = $this->connection->fetchColumn('SELECT count(*) FROM ' . $localTableName);
             $i = 1;
             while ($row = $stmt->fetch()) {
                 $newValues = [];
@@ -483,8 +483,8 @@ 
                     $newValues[$localColumn] = $fkValueGenerator();
                 }
                 $row = $this->stripUnselectableColumns($fk->getLocalTable(), $row);
-                if ($this->connection->update($localTableName, $newValues, $row) === 0){
-                    throw new DBFakerException("Row has not been updated $localTableName - ". var_export($newValues,true) . ' - ' . var_export($row, true));
+                if ($this->connection->update($localTableName, $newValues, $row) === 0) {
+                    throw new DBFakerException("Row has not been updated $localTableName - " . var_export($newValues, true) . ' - ' . var_export($row, true));
                 };
                 $this->log->info("Updated $i of $count for $localTableName");
                 $i++;
@@ -503,7 +503,7 @@ 
      */
     private function getCompoundColumnGenerator(Table $table, Index $index, int $count): CompoundColumnGenerator
     {
-        if (!isset($this->compoundColumnGenerators[$table->getName() . "." . $index->getName()])){
+        if (!isset($this->compoundColumnGenerators[$table->getName() . "." . $index->getName()])) {
             $compoundGenerator = new CompoundColumnGenerator($table, $index, $this->schemaHelper, $this, $this->schemaManager, $this->fakerManagerHelper, $count);
             $this->compoundColumnGenerators[$table->getName() . '.' . $index->getName()] = $compoundGenerator;
         }
@@ -544,7 +544,7 @@ 
         return array_filter($row, function(string $columnName) use ($table) {
             return !\in_array($table->getColumn($columnName)->getType()->getName(), [
                 Type::BINARY, Type::JSON, Type::JSON_ARRAY, Type::SIMPLE_ARRAY, Type::TARRAY, Type::BLOB, Type::JSON, Type::OBJECT
-            ],true);
+            ], true);
         }, ARRAY_FILTER_USE_KEY);
     }
 
@@ -557,9 +557,9 @@ 
         $this->log->info('Step 2.1 : store extension constraints ...');
         $extensionConstraints = [];
         $tables = $this->schemaManager->listTables();
-        foreach ($tables as $table){
-            foreach ($table->getForeignKeys() as $fk){
-                if ($this->schemaHelper->isExtendingKey($fk)){
+        foreach ($tables as $table) {
+            foreach ($table->getForeignKeys() as $fk) {
+                if ($this->schemaHelper->isExtendingKey($fk)) {
                     $extensionConstraints[] = $fk;
                 }
             }

src/Generators/ForeignKeyColumnGenerator.php

Spacing +2 added lines, -2 removed lines

@@ -45,7 +45,7 @@ 
      * @throws \Doctrine\DBAL\DBALException
      * @throws \Doctrine\DBAL\Schema\SchemaException
      */
-    public function __construct(Table $table, Column $column, PrimaryKeyRegistry $foreignPkRegistry,  ForeignKeyConstraint $fk, DBFakerSchemaManager $schemaManager, SchemaHelper $schemaHelper)
+    public function __construct(Table $table, Column $column, PrimaryKeyRegistry $foreignPkRegistry, ForeignKeyConstraint $fk, DBFakerSchemaManager $schemaManager, SchemaHelper $schemaHelper)
     {
         $this->foreignColumn = $schemaManager->getForeignColumn($table, $column);
         $this->foreignPkRegistry = $foreignPkRegistry;
@@ -61,7 +61,7 @@ 
     {
         $randomPk = $this->foreignPkRegistry->loadValuesFromTable()->getRandomValue($this->alreadyGeneratedValues);
         $value = $randomPk[$this->foreignColumn->getName()];
-        if ($this->generateUniqueValues){
+        if ($this->generateUniqueValues) {
             $this->alreadyGeneratedValues[] = $randomPk;
         }
         return $value;

src/Generators/CompoundColumnGenerator.php

Spacing +8 added lines, -8 removed lines

@@ -40,12 +40,12 @@ 
      */
     public function __construct(Table $table, Index $index, SchemaHelper $schemaHelper, DBFaker $dbFaker, AbstractSchemaManager $schemaManager, DBFakerSchemaManager $fakerManagerHelper, int $valuesCount)
     {
-        foreach ($index->getColumns() as  $columnName){
+        foreach ($index->getColumns() as  $columnName) {
             //FK or normal column ?
             $column = $table->getColumn($columnName);
-            if ($schemaHelper->isColumnPartOfForeignKeyConstraint($table, $column)){
+            if ($schemaHelper->isColumnPartOfForeignKeyConstraint($table, $column)) {
                 $fkConstraint = $schemaHelper->getForeignKeyConstraintByLocal($table, $column);
-                if ($fkConstraint === null){
+                if ($fkConstraint === null) {
                     throw new SchemaLogicException($column->getName() . ' was detected as foreign key but could not get it');
                 }
                 $foreignTableName = $fkConstraint->getForeignTableName();
@@ -53,12 +53,12 @@ 
                 $foreignTable = $schemaManager->listTableDetails($foreignTableName);
                 $pkRegistry = $dbFaker->getPkRegistry($foreignTable);
                 $values = $pkRegistry->loadValuesFromTable()->getAllValues();
-                $this->possibleValues[$columnName] = array_map(function ($value) use ($foreignColumn) {
+                $this->possibleValues[$columnName] = array_map(function($value) use ($foreignColumn) {
                     return $value[$foreignColumn->getName()];
                 }, $values);
-            }else{
+            }else {
                 $generator = $dbFaker->getSimpleColumnGenerator($table, $column);
-                for($i = 0; $i < $valuesCount; $i++) {
+                for ($i = 0; $i < $valuesCount; $i++) {
                     $this->possibleValues[$columnName][] = $generator();
                 }
             }
@@ -70,10 +70,10 @@ 
     public function __invoke() : array
     {
         $returnVal = [];
-        foreach ($this->possibleValues as $columnName => $values){
+        foreach ($this->possibleValues as $columnName => $values) {
             $returnVal[$columnName] = $values[array_rand($values)];
         }
-        if (!\in_array($returnVal, $this->generatedValues, true)){
+        if (!\in_array($returnVal, $this->generatedValues, true)) {
             $this->generatedValues[] = $returnVal;
             return $returnVal;
         }

src/Helpers/SchemaHelper.php

Spacing +5 added lines, -5 removed lines

@@ -74,8 +74,8 @@ 
     public function getForeignKeyConstraintByLocal(Table $table, Column $column) : ?ForeignKeyConstraint
     {
         $table = $this->schema->getTable($table->getName());
-        foreach ($table->getForeignKeys() as $foreignKeyConstraint){
-            if (\in_array($column->getName(), $foreignKeyConstraint->getLocalColumns(), true)){
+        foreach ($table->getForeignKeys() as $foreignKeyConstraint) {
+            if (\in_array($column->getName(), $foreignKeyConstraint->getLocalColumns(), true)) {
                 return $foreignKeyConstraint;
             }
         }
@@ -112,13 +112,13 @@ 
     public function isForeignKetAlsoUniqueIndex($fk) : bool
     {
         $table = $fk->getLocalTable();
-        foreach ($table->getIndexes() as $index){
-            if ($index->isUnique() && count($index->getColumns()) === count($fk->getLocalColumns())){
+        foreach ($table->getIndexes() as $index) {
+            if ($index->isUnique() && count($index->getColumns()) === count($fk->getLocalColumns())) {
                 $indexCols = $index->getColumns();
                 $fkCols = $fk->getColumns();
                 sort($indexCols);
                 sort($fkCols);
-                if ($indexCols == $fkCols){
+                if ($indexCols == $fkCols) {
                     return true;
                 }
             }

src/Helpers/PrimaryKeyRegistry.php

Spacing +8 added lines, -8 removed lines

@@ -49,15 +49,15 @@ 
         $this->connection = $connection;
         $refTable = null;
         $refCols = [];
-        foreach ($table->getForeignKeys() as $fk){
-            if ($helper->isExtendingKey($fk) && $table->getName() !== $fk->getForeignTableName()){
+        foreach ($table->getForeignKeys() as $fk) {
+            if ($helper->isExtendingKey($fk) && $table->getName() !== $fk->getForeignTableName()) {
                 $refTable = $fk->getForeignTableName();
                 $refCols = $fk->getForeignColumns();
             }
         }
-        if ($isSelfReferencing || !$refTable){
+        if ($isSelfReferencing || !$refTable) {
             $pk = $table->getPrimaryKey();
-            if ($pk === null){
+            if ($pk === null) {
                 throw new SchemaLogicException('No PK on table ' . $table->getName());
             }
             $refTable = $table->getName();
@@ -74,13 +74,13 @@ 
      */
     public function loadValuesFromTable() : PrimaryKeyRegistry
     {
-        if (!$this->valuesLoaded){
+        if (!$this->valuesLoaded) {
             $this->values = [];
             $colNames = implode(",", $this->columnNames);
             $rows = $this->connection->query("SELECT $colNames FROM " . $this->tableName)->fetchAll();
-            foreach ($rows as $row){
+            foreach ($rows as $row) {
                 $pk = [];
-                foreach ($this->columnNames as $column){
+                foreach ($this->columnNames as $column) {
                     $pk[$column] = $row[$column];
                 }
                 $this->values[] = $pk;
@@ -98,7 +98,7 @@ 
     {
         $keys = array_keys($value);
         sort($keys);
-        if ($this->columnNames != $keys){
+        if ($this->columnNames != $keys) {
             throw new PrimaryKeyColumnMismatchException('PrimaryKeys do not match between PKStore and addValue');
         }
         $this->values[] = $value;