/**
  * @covers ::load
  */
 public function testLoad()
 {
     // Make sure that the contexts are passed down (or not).
     $this->pageVariant->getContexts()->willReturn([]);
     $this->panelsDisplay->setContexts([])->shouldBeCalledTimes(1);
     $this->storage->load('id_exists')->willReturn($this->pageVariant->reveal());
     $this->storage->load('doesnt_exist')->willReturn(NULL);
     $this->storage->load('not_a_panel')->willReturn($this->pageVariantNotPanels->reveal());
     $panels_storage = new PageManagerPanelsStorage([], '', [], $this->entityTypeManager->reveal());
     // Test the success condition.
     $this->assertSame($this->panelsDisplay->reveal(), $panels_storage->load('id_exists'));
     // Should be NULL if it doesn't exist.
     $this->assertNull($panels_storage->load('doesnt_exist'));
     // Should also be NULL if it's not a PanelsDisplayVariant.
     $this->assertNull($panels_storage->load('not_a_panel'));
 }