MockSessionTrait.php

<?php

namespace Drupal\Tests\stream_wrapper_example\Traits;

use Drupal\stream_wrapper_example\SessionHelper;
use Symfony\Component\HttpFoundation\RequestStack;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Session\SessionInterface;
use Prophecy\Argument;
use Prophecy\PhpUnit\ProphecyTrait;

/**
 * A trait to expose a mock session type to PHPUnit tests.
 */
trait MockSessionTrait {
  use ProphecyTrait;
  
  /**
   * We'll use this array to back our mock session.
   *
   * @var array
   */
  protected $sessionStore;
  
  /**
   * A representation of the HTTP request.
   *
   * @var \Symfony\Component\HttpFoundation\RequestStack|\Prophecy\Prophecy\ProphecyInterface
   */
  protected $requestStack;
  
  /**
   * Create a mock session object.
   *
   * @return \Symfony\Component\HttpFoundation\RequestStack|\Prophecy\Prophecy\ProphecyInterface
   *   A test double, or mock, of a RequestStack object
   *   that can be used to return a mock Session object.
   */
  protected function createSessionMock() {
    $this->sessionStore = [];
    $session = $this->prophesize(SessionInterface::class);
    $test = $this;
    $session->get('stream_wrapper_example', [])
      ->will(function ($args) use ($test) {
      return $test->getSessionStore();
    });
    $session->set('stream_wrapper_example', Argument::any())
      ->will(function ($args) use ($test) {
      $test->setSessionStore($args[1]);
    });
    $session->remove('stream_wrapper_example')
      ->will(function ($args) use ($test) {
      $test->resetSessionStore();
    });
    $request = $this->prophesize(Request::class);
    $request->getSession()
      ->willReturn($session->reveal());
    $request_stack = $this->prophesize(RequestStack::class);
    $request_stack->getCurrentRequest()
      ->willReturn($request->reveal());
    return $this->requestStack = $request_stack->reveal();
  }
  
  /**
   * Get a session helper.
   */
  public function getSessionHelper() {
    return new SessionHelper($this->requestStack);
  }
  
  /**
   * Helper for mocks.
   */
  public function getSessionStore() {
    return $this->sessionStore;
  }
  
  /**
   * Helper for our mocks.
   */
  public function setSessionStore($data) {
    $this->sessionStore = $data;
  }
  
  /**
   * Helper for our mocks.
   */
  public function resetSessionStore() {
    $this->sessionStore = [];
  }

}