Mocking DB object can't use getDouble? about ci-phpunit-test HOT 1 CLOSED

<?php

class TestCase extends CIPHPUnitTestCase {
  /**
   * Override to fix stub bug
   */
  public function getDouble($classname, $params, $constructor_params = false) {
    $mockBuilder = $this->getMockBuilder($classname);
    if ($constructor_params === false) {
      $mockBuilder->disableOriginalConstructor();
    }
    elseif (is_array($constructor_params)) {
      $mockBuilder->setConstructorArgs($constructor_params);
    }

    $methods = [];
    $onConsecutiveCalls = [];
    $otherCalls = [];

    foreach ($params as $key => $val) {
      if (is_int($key)) {
        $onConsecutiveCalls = array_merge($onConsecutiveCalls, $val);
        $methods[] = array_keys($val)[0];
      } else {
        $otherCalls[$key] = $val;
        $methods[] = $key;
      }
    }

    $mock = $mockBuilder->setMethods($methods)->getMock();

    foreach ($onConsecutiveCalls as $method => $returns) {
      $mock->expects($this->any())->method($method)
        ->will(
          call_user_func_array(
            [$this->testCase, 'onConsecutiveCalls'],
            $returns
          )
        );
    }

    foreach ($otherCalls as $method => $return) {
      if (is_object($return) && ($return instanceof PHPUnit_Framework_MockObject_Stub || $return instanceof PHPUnit\Framework\MockObject\Stub\Stub)) {
        $mock->expects($this->any())->method($method)
          ->will($return);
      }
      elseif (is_object($return) && $return instanceof Closure) {
        $mock->expects($this->any())->method($method)
          ->willReturnCallback($return);
      }
      elseif ($return === ':void') {
        $mock->expects($this->any())->method($method);
      }
      else {
        $mock->expects($this->any())->method($method)
          ->willReturn($return);
      }
    }

    return $mock;
  }
}