fix: tests

Author: AZabolotnikov

tests/Support/GeneratorMockTrait.php

@@ -63,7 +63,8 @@ public function mockDBTransactionStartRollback(int $count = 1): void
 
     public function mockGettingModelInstance(object $model): void
     {
-        $laravelConnectionMock = Mockery::mock(LaravelConnection::class)
+        $laravelConnectionMock = Mockery::mock(LaravelConnection::class);
+        $laravelConnectionMock
             ->shouldReceive('getConfig')
             ->andReturn([
                 'database' => 'my_db',
@@ -77,20 +78,21 @@ public function mockGettingModelInstance(object $model): void
             ->with('pgsql')
             ->andReturn($laravelConnectionMock);
 
-        $schemaManagerMock = Mockery::mock(AbstractSchemaManager::class)
+        $schemaManagerMock = Mockery::mock(AbstractSchemaManager::class);
+        $schemaManagerMock
             ->shouldReceive('listTableColumns')
             ->andReturn([
                 new Column('id', new IntegerType()),
                 new Column('title', new StringType()),
                 new Column('created_at', new DateTimeType()),
             ]);
 
-        $connectionMock = Mockery::mock(Connection::class)
-            ->makePartial()
+        $connectionMock = Mockery::mock(Connection::class);
+        $connectionMock->makePartial()
             ->expects('createSchemaManager')
             ->andReturn($schemaManagerMock);
 
-        $driverManagerMock = Mockery::mock('alias:' . DriverManager::class)
+        Mockery::mock('alias:' . DriverManager::class)
             ->shouldReceive('getConnection')
             ->with([
                 'dbname' => 'my_db',