class ToolbarCacheContextsTest

class ToolbarCacheContextsTest extends BrowserTestBase {
  use AssertPageCacheContextsAndTagsTrait;
  
  /**
   * Modules to enable.
   *
   * @var array
   */
  protected static $modules = [
    'toolbar',
    'test_page_test',
  ];
  
  /**
   * {@inheritdoc}
   */
  protected $defaultTheme = 'stark';
  
  /**
   * An authenticated user to use for testing.
   *
   * @var \Drupal\user\UserInterface
   */
  protected $adminUser;
  
  /**
   * An authenticated user to use for testing.
   *
   * @var \Drupal\user\UserInterface
   */
  protected $adminUser2;
  
  /**
   * A list of default permissions for test users.
   *
   * @var array
   */
  protected $perms = [
    'access toolbar',
    'access administration pages',
    'administer site configuration',
  ];
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->adminUser = $this->drupalCreateUser($this->perms);
    $this->adminUser2 = $this->drupalCreateUser($this->perms);
  }
  
  /**
   * Tests toolbar cache integration.
   *
   * @group legacy
   */
  public function testCacheIntegration() {
    $this->expectDeprecation('Route requirement _access_rest_csrf is deprecated in drupal:9.2.0 and is removed in drupal:10.0.0. Use _csrf_request_header_token instead. See https://www.drupal.org/node/2772399');
    $this->installExtraModules([
      'csrf_test',
      'dynamic_page_cache',
    ]);
    $this->drupalLogin($this->adminUser);
    $this->drupalGet('test-page');
    $this->assertSession()
      ->responseHeaderEquals('X-Drupal-Dynamic-Cache', 'MISS');
    $this->assertCacheContexts([
      'session',
      'user',
      'url.query_args:' . MainContentViewSubscriber::WRAPPER_FORMAT,
    ], 'Expected cache contexts found with CSRF token link.');
    $this->drupalGet('test-page');
    $this->assertSession()
      ->responseHeaderEquals('X-Drupal-Dynamic-Cache', 'HIT');
    $this->assertCacheContexts([
      'session',
      'user',
      'url.query_args:' . MainContentViewSubscriber::WRAPPER_FORMAT,
    ], 'Expected cache contexts found with CSRF token link.');
  }
  
  /**
   * Tests toolbar cache contexts.
   */
  public function testToolbarCacheContextsCaller() {
    // Test with default combination and permission to see toolbar.
    $this->assertToolbarCacheContexts([
      'user',
    ], 'Expected cache contexts found for default combination and permission to see toolbar.');
    // Test without user toolbar tab. User module is a required module so we have to
    // manually remove the user toolbar tab.
    $this->installExtraModules([
      'toolbar_disable_user_toolbar',
    ]);
    $this->assertToolbarCacheContexts([
      'user.permissions',
    ], 'Expected cache contexts found without user toolbar tab.');
    // Test with the toolbar and contextual enabled.
    $this->installExtraModules([
      'contextual',
    ]);
    $this->adminUser2 = $this->drupalCreateUser(array_merge($this->perms, [
      'access contextual links',
    ]));
    $this->assertToolbarCacheContexts([
      'user.permissions',
    ], 'Expected cache contexts found with contextual module enabled.');
    \Drupal::service('module_installer')->uninstall([
      'contextual',
    ]);
    // Test with the tour module enabled.
    $this->installExtraModules([
      'tour',
    ]);
    $this->adminUser2 = $this->drupalCreateUser(array_merge($this->perms, [
      'access tour',
    ]));
    $this->assertToolbarCacheContexts([
      'user.permissions',
    ], 'Expected cache contexts found with tour module enabled.');
    \Drupal::service('module_installer')->uninstall([
      'tour',
    ]);
  }
  
  /**
   * Tests that cache contexts are applied for both users.
   *
   * @param string[] $cache_contexts
   *   Expected cache contexts for both users.
   * @param string $message
   *   (optional) A verbose message to output.
   *
   * @internal
   */
  protected function assertToolbarCacheContexts(array $cache_contexts, string $message = NULL) : void {
    // Default cache contexts that should exist on all test cases.
    $default_cache_contexts = [
      'languages:language_interface',
      'theme',
      'url.query_args:' . MainContentViewSubscriber::WRAPPER_FORMAT,
    ];
    $cache_contexts = Cache::mergeContexts($default_cache_contexts, $cache_contexts);
    // Assert contexts for user1 which has only default permissions.
    $this->drupalLogin($this->adminUser);
    $this->drupalGet('test-page');
    $this->assertCacheContexts($cache_contexts, $message);
    $this->drupalLogout();
    // Assert contexts for user2 which has some additional permissions.
    $this->drupalLogin($this->adminUser2);
    $this->drupalGet('test-page');
    $this->assertCacheContexts($cache_contexts, $message);
  }
  
  /**
   * Installs a given list of modules and rebuilds the cache.
   *
   * @param string[] $module_list
   *   An array of module names.
   */
  protected function installExtraModules(array $module_list) {
    \Drupal::service('module_installer')->install($module_list);
    // Installing modules updates the container and needs a router rebuild.
    $this->container = \Drupal::getContainer();
    $this->container
      ->get('router.builder')
      ->rebuildIfNeeded();
  }

}