Added timeout setting for drush, since default timeout of 60 seconds might not be enough for some time-consuming commands. Also added new parameter $return_process.

Author: Rade333

src/Codeception/Module/DrupalDrush.php

@@ -13,6 +13,7 @@
  *     modules:
  *        - DrupalDrush:
  *          working_directory: './web'
+ *          timeout: 120
  *          drush: './vendor/bin/drush'
  *          alias: '@mysite.com'
  *          options:
@@ -42,11 +43,14 @@ class DrupalDrush extends Module {
    *   e.g. "en devel -y".
    * @param array $options
    *   Associative array of options.
+   * @param bool $return_process
+   *   If TRUE, the Process object will be returned. If false, the output of
+   *   Process::getOutput() will be returned. Defaults to FALSE.
    *
-   * @return string
-   *   The process output.
+   * @return string|\Symfony\Component\Process\Process
+   *   The process output, or the process itself.
    */
-  public function runDrush($command, array $options = []) {
+  public function runDrush($command, array $options = [], $return_process = FALSE) {
     if ($alias = $this->_getConfig('alias')) {
       $command = $alias . ' ' . $command;
     }
@@ -56,7 +60,7 @@ public function runDrush($command, array $options = []) {
     elseif ($this->_getConfig('options')) {
       $command = $this->normalizeOptions($this->_getConfig('options')) . $command;
     }
-    return Drush::runDrush($command, $this->_getConfig('drush'), $this->_getConfig('working_directory'));
+    return Drush::runDrush($command, $this->_getConfig('drush'), $this->_getConfig('working_directory'), $this->_getConfig('timeout'), $return_process);
   }
 
   /**

src/Codeception/Module/DrupalUser.php

@@ -142,7 +142,7 @@ public function createUserWithRoles(array $roles = [], $password = FALSE) {
    */
   public function logInAs($username) {
     $alias = $this->_getConfig('alias') ? $this->_getConfig('alias') . ' ' : '';
-    $output = Drush::runDrush($alias. 'uli --name=' . $username, $this->_getConfig('drush'), $this->_getConfig('working_directory'));
+    $output = Drush::runDrush($alias. 'uli --name=' . $username, $this->_getConfig('drush'), $this->_getConfig('working_directory'), $this->_getConfig('timeout'));
     $gen_url = str_replace(PHP_EOL, '', $output);
     $url = substr($gen_url, strpos($gen_url, '/user/reset'));
     $this->driver->amOnPage($url);

src/Codeception/Util/Drush.php

@@ -21,11 +21,16 @@ class Drush {
    *   The drush command to use.
    * @param string $pwd
    *   Working directory.
+   * @param int|float $timeout.
+   *   Drush execution timeout.
+   * @param bool $return_process
+   *   If TRUE, the Process object will be returned. If false, the output of
+   *   Process::getOutput() will be returned. Defaults to FALSE.
    *
-   * @return string
-   *   The process output.
+   * @return string|\Symfony\Component\Process\Process
+   *   The process output, or the process object itself.
    */
-  public static function runDrush($command, $drush = 'drush', $pwd = NULL) {
+  public static function runDrush($command, $drush = 'drush', $pwd = NULL, $timeout = NULL, $return_process = FALSE) {
     $command_args = array_merge([$drush], explode(' ', $command));
     $process = new Process($command_args);
 
@@ -34,7 +39,14 @@ public static function runDrush($command, $drush = 'drush', $pwd = NULL) {
       $process->setWorkingDirectory($pwd);
     }
 
-    return $process->mustRun()->getOutput();
+    // Set timeout if configured.
+    if (isset($timeout)) {
+      $process->setTimeout($timeout);
+    }
+
+    $process->mustRun();
+
+    return $return_process ? $process : $process->getOutput();
   }
 
 }