Followup to: Issue #3052875 by Pancho: Skip storing entity context for samples

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']);
     }
   }