function DrupalTest::setMockContainerService

Sets up a mock expectation for the container get() method.

Parameters

string $service_name: The service name to expect for the get() method.

mixed $return: The value to return from the mocked container get() method.

File

core/tests/Drupal/Tests/Core/DrupalTest.php, line 444

Class

DrupalTest

Tests the Drupal class.

Namespace

Code

protected function setMockContainerService($service_name, $return = NULL) {
    $expects = $this->container
        ->expects($this->once())
        ->method('get')
        ->with($service_name);
    if (isset($return)) {
        $expects->willReturn($return);
    }
    else {
        $expects->willReturn(TRUE);
    }
    \Drupal::setContainer($this->container);
}