refactor: code

refs: #50

Author: pirs1337

src/Generators/AbstractTestsGenerator.php

@@ -218,7 +218,7 @@ protected function generateFixture($fixtureName, $data): void
     protected function buildRelationsTree($models): array
     {
         foreach ($models as $model) {
-            $relations = $this->getRelatedModels($model, 'Test');
+            $relations = $this->getRelatedModels($model, $this->getTestClassName());
             $relationsWithFactories = $this->getModelsWithFactories($relations);
 
             if (empty($relationsWithFactories)) {

src/Generators/EntityGenerator.php

@@ -164,23 +164,23 @@ protected function throwFailureException($exceptionClass, $failureMessage, $reco
         throw new $exceptionClass("{$failureMessage} {$recommendedMessage}");
     }
 
-    protected function getRelatedModels(string $model, string $classType): array
+    protected function getRelatedModels(string $model, string $creatableClass): array
     {
-        $class = $this->getModelClass($model);
+        $modelClass = $this->getModelClass($model);
 
-        if (!class_exists($class)) {
+        if (!class_exists($modelClass)) {
             $this->throwFailureException(
                 ClassNotExistsException::class,
-                "Cannot create {$model}{$classType} cause {$model} Model does not exists.",
+                "Cannot create {$creatableClass} cause {$model} Model does not exists.",
                 "Create a {$model} Model by himself or run command 'php artisan make:entity {$model} --only-model'."
             );
         }
 
-        $instance = new $class();
+        $instance = new $modelClass();
 
-        $publicMethods = (new ReflectionClass($class))->getMethods(ReflectionMethod::IS_PUBLIC);
+        $publicMethods = (new ReflectionClass($modelClass))->getMethods(ReflectionMethod::IS_PUBLIC);
 
-        $methods = array_filter($publicMethods, fn ($method) => $method->class === $class && !$method->getParameters());
+        $methods = array_filter($publicMethods, fn ($method) => $method->class === $modelClass && !$method->getParameters());
 
         $relatedModels = [];
 

src/Generators/FactoryGenerator.php

@@ -112,7 +112,7 @@ protected function prepareEmptyFactory(): void
     protected function checkExistRelatedModelsFactories(): bool
     {
         $modelFactoryContent = file_get_contents($this->paths['factory']);
-        $relatedModels = $this->getRelatedModels($this->model, 'Factory');
+        $relatedModels = $this->getRelatedModels($this->model, "{$this->model}Factory");
         $modelNamespace = $this->getOrCreateNamespace('models');
 
         foreach ($relatedModels as $relatedModel) {