SchemaFinder to check entity class

mdiyakov · Mar 26, 2018 · bbb5aab · bbb5aab
1 parent 3f376d4
commit bbb5aab
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Finder/SchemaFinder.php b/Finder/SchemaFinder.php
@@ -45,7 +45,7 @@ public function __construct(SelectQueryBuilder $queryBuilder, Schema $schema, Co
         $this->schema = $schema;
 
         foreach ($config->getIndexedEntities() as $entityConfig) {
-            if (!$schema->getName() == $entityConfig['schema']) {
+            if ($schema->getName() != $entityConfig['schema']) {
                 continue;
             }
             $this->entityConfigs[$entityConfig['class']] = $entityConfig;
@@ -58,7 +58,9 @@ public function __construct(SelectQueryBuilder $queryBuilder, Schema $schema, Co
     public function addSelectClass($class)
     {
         if (!array_key_exists($class, $this->entityConfigs)) {
-            throw new \InvalidArgumentException('');
+            throw new \InvalidArgumentException(
+                sprintf('Class "%s" is not configured to be used with schema "%s"', $class, $this->getSchema()->getName())
+            );
         }
 
         if (array_search($class, $this->selectedClasses) === false) {