From a03cd7619678e6dd109db11a265ddd208d135f49 Mon Sep 17 00:00:00 2001
From: Bernd Oliver Suenderhauf <bos@suenderhauf.de>
Date: Tue, 7 May 2019 21:07:56 +0200
Subject: [PATCH] Followup to: Issue #3052875 by Pancho: Skip storing entity
 context for samples

---
 .../src/Unit/LinkManipulator/ParseLinkSampleTest.php  | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php b/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
index eb96742..39d1266 100644
--- a/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
+++ b/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
@@ -28,11 +28,14 @@ class ParseLinkSampleTest extends UnitTestCase {
     if ($request_context['sample']) {
       $this->assertEmpty($request_context['entity_ids']);
     }
-    elseif (count($entity_ids) == 2) {
-      $this->assertEquals(['node' => ['1' => '1', '2' => '2']], $request_context['entity_ids']);
-    }
     else {
-      $this->assertEquals(['node' => ['1' => '1']], $request_context['entity_ids']);
+      $expected = [
+        'node' => ['1' => '1'],
+      ];
+      if (is_array($entity_ids) && count($entity_ids) == 2) {
+        $expected['node']['2'] = '2';
+      }
+      $this->assertEquals($expected, $request_context['entity_ids']);
     }
   }
 
-- 
GitLab